def test_add_node(self, mknode): # pylint: disable=no-self-use # when full, return false bucket = KBucket(0, 10, 2) assert bucket.add_node(mknode()) is True assert bucket.add_node(mknode()) is True assert bucket.add_node(mknode()) is False assert len(bucket) == 2 # make sure when a node is double added it's put at the end bucket = KBucket(0, 10, 3) nodes = [mknode(), mknode(), mknode()] for node in nodes: bucket.add_node(node) for index, node in enumerate(bucket.get_nodes()): assert node == nodes[index]
def test_add_node(self): # when full, return false bucket = KBucket(0, 10, 2) self.assertTrue(bucket.add_node(mknode())) self.assertTrue(bucket.add_node(mknode())) self.assertFalse(bucket.add_node(mknode())) self.assertEqual(len(bucket), 2) # make sure when a node is double added it's put at the end bucket = KBucket(0, 10, 3) nodes = [mknode(), mknode(), mknode()] for node in nodes: bucket.add_node(node) for index, node in enumerate(bucket.get_nodes()): self.assertEqual(node, nodes[index])