summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--combox/events.py2
-rw-r--r--tests/events_test.py9
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.