From 0332e94656d34b132366c92008df62540656d120 Mon Sep 17 00:00:00 2001 From: Brian Warner 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