From: Brian Warner <warner@lothar.com>
Date: Sun, 12 Aug 2007 23:29:34 +0000 (-0700)
Subject: test_system.py: add coverage for get_permuted_peers()
X-Git-Url: https://git.rkrishnan.org/rgr-080307.php?a=commitdiff_plain;h=0332e94656d34b132366c92008df62540656d120;p=tahoe-lafs%2Ftahoe-lafs.git

test_system.py: add coverage for get_permuted_peers()
---

diff --git a/src/allmydata/test/test_system.py b/src/allmydata/test/test_system.py
index eb5eaf60..b66ed063 100644
--- a/src/allmydata/test/test_system.py
+++ b/src/allmydata/test/test_system.py
@@ -119,7 +119,13 @@ class SystemTest(testutil.SignalMixin, unittest.TestCase):
         def _check(extra_node):
             self.extra_node = extra_node
             for c in self.clients:
-                self.failUnlessEqual(len(list(c.get_all_peerids())), 6)
+                all_peerids = list(c.get_all_peerids())
+                self.failUnlessEqual(len(all_peerids), 6)
+                permuted_peers = list(c.get_permuted_peers("a", True))
+                self.failUnlessEqual(len(permuted_peers), 6)
+                permuted_other_peers = list(c.get_permuted_peers("a", False))
+                self.failUnlessEqual(len(permuted_other_peers), 5)
+
         d.addCallback(_check)
         def _shutdown_extra_node(res):
             if self.extra_node:
@@ -141,6 +147,15 @@ class SystemTest(testutil.SignalMixin, unittest.TestCase):
         # to disk among all our simulated nodes
         DATA = "Some data to upload\n" * 200
         d = self.set_up_nodes()
+        def _check_connections(res):
+            for c in self.clients:
+                all_peerids = list(c.get_all_peerids())
+                self.failUnlessEqual(len(all_peerids), 5)
+                permuted_peers = list(c.get_permuted_peers("a", True))
+                self.failUnlessEqual(len(permuted_peers), 5)
+                permuted_other_peers = list(c.get_permuted_peers("a", False))
+                self.failUnlessEqual(len(permuted_other_peers), 4)
+        d.addCallback(_check_connections)
         def _do_upload(res):
             log.msg("UPLOADING")
             u = self.clients[0].getServiceNamed("uploader")