From 0332e94656d34b132366c92008df62540656d120 Mon Sep 17 00:00:00 2001
From: Brian Warner <warner@lothar.com>
Date: Sun, 12 Aug 2007 16:29:34 -0700
Subject: [PATCH] test_system.py: add coverage for get_permuted_peers()

---
 src/allmydata/test/test_system.py | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

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")
-- 
2.45.2