def test_link_get_bad_parent(self): pod = Link session = Session(URL, LOGIN, PASSWORD) self.assertRaises(TypeError, Link.get, session, pod) pod = Pod('1') self.assertRaises(TypeError, Link.get, 'bad_session', pod)
def test_node_equal(self): """ Test node equal :return: None """ node_name = 'Leaf1' node_pod = '1' node_node = '3' node_role = 'leaf' node1 = Node(node_pod, node_node, node_name, node_role) node_name = 'Leaf2' node2 = Node(node_pod, node_node, node_name, node_role) self.assertNotEqual(node1, node2) node_name = 'Leaf1' node_pod = '2' node2 = Node(node_pod, node_node, node_name, node_role) self.assertNotEqual(node1, node2) node_pod = '1' node_node = '4' node2 = Node(node_pod, node_node, node_name, node_role) self.assertNotEqual(node1, node2) node_node = '3' node_role = 'controller' node2 = Node(node_pod, node_node, node_name, node_role) self.assertNotEqual(node1, node2) node_role = 'leaf' node2 = Node(node_pod, node_node, node_name, node_role) self.assertEqual(node1, node2) node2 = Pod(node_pod) self.assertNotEqual(node1, node2)
def test_parameters(self): pod = Pod('1') node1 = '2' node2 = '3' slot1 = '4' slot2 = '5' port1 = '6' port2 = '7' link = '101' link1 = Link(pod) link1.pod = pod.pod link1.link = link link1.node1 = node1 link1.slot1 = slot1 link1.port1 = port1 link1.node2 = node2 link1.slot2 = slot2 link1.port2 = port2 self.assertEqual(link1.pod, pod.pod) self.assertEqual(link1.link, link) self.assertEqual(link1.node1, node1) self.assertEqual(link1.slot1, slot1) self.assertEqual(link1.port1, port1) self.assertEqual(link1.node2, node2) self.assertEqual(link1.slot2, slot2) self.assertEqual(link1.port2, port2) self.assertEqual(link1.get_parent(), pod) self.assertEqual(link1.get_port_id1(), '1/2/4/6') self.assertEqual(link1.get_port_id2(), '1/3/5/7')
def test_pod_type(self): """ Test the pod type :return: None """ pod = Pod('1') self.assertEqual(pod.get_type(), 'pod')
def test_pod_id(self): """ Test the pod id :return: None """ pod = Pod('1') self.assertEqual(pod.pod, '1')
def test_get_systemcontroller_parent_exception(self): """ Checks that an exception is raised when a Linecard.get is called with the wrong parent type :return: """ session = self.login_to_apic() node = Pod('1') self.assertRaises(TypeError, Systemcontroller.get, session, node)
def test_invalid_session_populate_children(self): """ Test node invalid populate_children :return: None """ pod1 = Pod('1') node = Node('1', '2', 'Spine1', 'spine', pod1) self.assertRaises(TypeError, node.populate_children)
def test_pod_get_json(self): """ Test pod get_json :return: None """ pod_id = '3' pod = Pod(pod_id) self.assertEqual(pod.get_json(), None)
def test_pod_str_name(self): """ Test pod string :return: None """ pod_id = '2' pod = Pod(pod_id) self.assertEqual(str(pod), 'pod-' + pod_id)
def test_pod_name(self): """ Test the pod name :return: None """ pod_id = '1' pod = Pod(pod_id) self.assertEqual(pod.get_name(), 'pod-' + pod_id) self.assertEqual(pod.get_pod(), '1')
def test_node_parent(self): """ Test node parent :return: None """ pod_id = '1' pod1 = Pod(pod_id) node = Node('1', '2', 'Spine1', role='leaf', parent=pod1) self.assertEqual(pod1, node.get_parent())
def test_pod_equal(self): """ Test pods are equal :return: None """ pod_id = '1' pod1 = Pod(pod_id) # check different IDs pod_id = '2' pod2 = Pod(pod_id) self.assertNotEqual(pod1, pod2) # check same pod_id = '1' pod2 = Pod(pod_id) self.assertEqual(pod1, pod2) # check different types pod2 = Node(pod_id, '2', 'Leaf1', role='leaf') self.assertNotEqual(pod1, pod2)
def test_eNode_parent(self): pod = Pod('1') node = ExternalSwitch(parent=pod) self.assertEqual(node.role, 'external_switch') self.assertEqual(node._parent, pod) children = pod.get_children() self.assertEqual(len(children), 1) for child in children: self.assertEqual(child, node) pod = Node('1', '101', 'Spine2', 'leaf') self.assertRaises(TypeError, ExternalSwitch, pod)
def test_get_endpoint_objects(self): pod_id = '1' node1_id = '2' node2_id = '3' slot1_id = '4' slot2_id = '5' port1_id = '6' port2_id = '7' link_id = '101' pod = Pod(pod_id) node1 = Node(pod_id, node1_id, 'Spine', 'spine', pod) node2 = Node(pod_id, node2_id, 'Leaf', 'leaf', pod) linecard1 = Linecard(slot1_id, node1) linecard2 = Linecard(slot2_id, node2) interface1 = Interface(interface_type='eth', pod=pod_id, node=node1_id, module=slot1_id, port=port1_id, parent=linecard1) interface2 = Interface(interface_type='eth', pod=pod_id, node=node2_id, module=slot2_id, port=port2_id, parent=linecard2) link1 = Link(pod) link1.pod = pod_id link1.link = link_id link1.node1 = node1_id link1.slot1 = slot1_id link1.port1 = port1_id link1.node2 = node2_id link1.slot2 = slot2_id link1.port2 = port2_id self.assertEqual(node1, link1.get_node1()) self.assertEqual(node2, link1.get_node2()) self.assertEqual(linecard1, link1.get_slot1()) self.assertEqual(linecard2, link1.get_slot2()) self.assertEqual(interface1, link1.get_port1()) self.assertEqual(interface2, link1.get_port2())
def test_find(self): pod_id = '1' node1_id = '2' node2_id = '3' slot1_id = '4' slot2_id = '5' port2_id = '7' pod = Pod(pod_id) node1 = Node(pod_id, node1_id, 'Spine', 'spine', pod) node2 = Node(pod_id, node2_id, 'Leaf', 'leaf', pod) linecard1 = Linecard(slot1_id, node1) linecard2 = Linecard(slot2_id, node2) linecard1.serial = 'SerialNumber1' linecard2.serial = 'SerialNumber2' interface2 = Interface(interface_type='eth', pod=pod_id, node=node2_id, module=slot2_id, port=port2_id, parent=linecard2) so = Search() so.node = node2_id results = pod.find(so) self.assertIn(node2, results) self.assertIn(linecard2, results) self.assertIn(interface2, results) self.assertEqual(len(results), 3) for result in results: self.assertEqual(result.node, node2_id) so = Search() so.serial = 'SerialNumber1' results = pod.find(so) self.assertIn(linecard1, results) self.assertEqual(len(results), 1) self.assertEqual(results[0].serial, 'SerialNumber1')