def test_can_add_list(self):
node0 = lib.NodeAddresses("node0")
node1 = lib.NodeAddresses("node1")
node2 = lib.NodeAddresses("node2")
self.assertEqual(
lib.NodeAddressesList([node0, node1, node2])._list,
(lib.NodeAddressesList([node0, node1]) + [node2])._list)
def test_create_from_node_list(self):
node1 = lib.NodeAddresses("node1")
node2 = lib.NodeAddresses("node2")
node3 = lib.NodeAddresses("node3")
nodes_source = lib.NodeAddressesList([node1, node2])
nodes = lib.NodeAddressesList(nodes_source)
nodes_source.append(node3)
self.assertEqual(2, len(nodes))
self.assertEqual([node1, node2], list(nodes))
self.assertEqual(node1, nodes[0])
self.assertEqual(node2, nodes[1])
def test_create_from_list(self):
node1 = lib.NodeAddresses("node1")
node2 = lib.NodeAddresses("node2")
nodes = lib.NodeAddressesList([node1, node2])
self.assertEqual(2, len(nodes))
self.assertEqual([node1, node2], list(nodes))
self.assertEqual(node1, nodes[0])
self.assertEqual(node2, nodes[1])
def test_find_by_label(self):
node0 = lib.NodeAddresses("node0")
node1 = lib.NodeAddresses("node1")
node_list = lib.NodeAddressesList([node0, node1])
self.assertEqual(node1, node_list.find_by_label("node1"))
self.assertEqual(node0, node_list.find_by_label("node0"))
self.assertRaises(lib.NodeNotFound,
lambda: node_list.find_by_label("node2"))
def test_append(self):
node1 = lib.NodeAddresses("node1")
node2 = lib.NodeAddresses("node2")
nodes = lib.NodeAddressesList()
nodes.append(node1)
self.assertEqual(1, len(nodes))
self.assertEqual([node1], list(nodes))
self.assertEqual(node1, nodes[0])
nodes.append(node2)
self.assertEqual(2, len(nodes))
self.assertEqual([node1, node2], list(nodes))
self.assertEqual(node1, nodes[0])
self.assertEqual(node2, nodes[1])
def test_create_from_empty_list(self):
nodes = lib.NodeAddressesList([])
self.assertEqual(0, len(nodes))
self.assertEqual([], list(nodes))
self.assertRaises(IndexError, lambda: nodes[0])
