new variable at NodeDirMonitor -- `num_nodes'

`num_nodes' a value that represents the number of node directories.

  modified:   combox/events.py
  modified:   tests/events_test.py

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.