diff options
-rw-r--r-- | combox/events.py | 2 | ||||
-rw-r--r-- | tests/events_test.py | 9 |
2 files changed, 11 insertions, 0 deletions
diff --git a/combox/events.py b/combox/events.py index 91be1a4..359aa03 100644 --- a/combox/events.py +++ b/combox/events.py @@ -188,6 +188,8 @@ class NodeDirMonitor(LoggingEventHandler): self.config = config self.silo = ComboxSilo(self.config, dblock) + self.num_nodes = len(get_nodedirs(self.config)) + def shardp(self, path): """Returns True if `path' is a shard diff --git a/tests/events_test.py b/tests/events_test.py index d754d44..bef6423 100644 --- a/tests/events_test.py +++ b/tests/events_test.py @@ -228,6 +228,15 @@ class TestEvents(object): assert not silo.stale(self.lorem_ipsum) + def test_NDM_numnodes(self): + """Tests whether the NodeDirMonitor's num_nodes variable has the + right value. + + """ + nmonitor = NodeDirMonitor(self.config, self.silo_lock) + assert_equal(2, nmonitor.num_nodes) + + def test_NDM(self): """ Tests the NodeDirMonitor class. |