def test_login(): def dummy_create_cert_session(): obj = DummyClass() obj.dn = 'dummy-dn' return (obj, 'dummy-key') def json(): return { 'imdata': [{ 'aaaLogin': { 'attributes': { 'token': 'dummy-token' } } }] } def dummy_post(self, url, data, headers, timeout, verify): obj = DummyClass() obj.status_code = 200 obj.json = json return obj Session.get = dummy_post apic_utils.create_cert_session = dummy_create_cert_session obj = AciUtils() response = obj.login() assert response == 'dummy-token'
def test_aci_get(status, expected): """Test aci get function""" # Mock session's get def json(): return {'dummy_key': 'dummy_val'} def dummy_get(self, url, cookies, timeout, verify): obj = DummyClass() obj.status_code = status if status == 200: obj.json = json return obj Session.get = dummy_get # Mock apic_util login def dummy_login(self): return "dummy-token" AciUtils.login = dummy_login # Test the function obj = AciUtils() response = obj.aci_get("http://dummy.apic.url") assert response == expected
def test_parse_ep_data(data, get_ep_data, expected): def dummy_parse_and_return_ep_data(self, ep_data): return get_data(get_ep_data) AciUtils.parse_and_return_ep_data = dummy_parse_and_return_ep_data obj = AciUtils() response = obj.parse_ep_data(get_data(data)) assert response == get_data(expected)
def test_get_controller_and_hosting_server(data, mo_instance_data, expected): def dummy_get_all_mo_instances(self, mo_class, query_string=""): return get_data(mo_instance_data) AciUtils.get_all_mo_instances = dummy_get_all_mo_instances # Test the function obj = AciUtils() response = obj.get_controller_and_hosting_server(get_data(data)) assert response == expected
def test_get_all_mo_instances(data, expected): def dummy_aci_get(self, url): return get_data(data) AciUtils.aci_get = dummy_aci_get # Test the function obj = AciUtils() response = obj.get_all_mo_instances("dummy-class", "dummy-query") assert response == get_data(expected)
def test_get_vm_domain_and_name(aci_get_data, data): """Test the VM Domain and Name""" def dummy_aci_get(self, url): return get_data(aci_get_data) AciUtils.aci_get = dummy_aci_get # Test the function obj = AciUtils() response = obj.get_vm_domain_and_name(get_data(data)) assert response == ('DUMMY0-leaf000', 'dummy-vm-name')
def test_parse_epg_data(): # Mock AciUtils methods def dummy_parse_and_return_epg_data(self, url): return {"dummy-key": "dummy-val"} AciUtils.parse_and_return_epg_data = dummy_parse_and_return_epg_data obj = AciUtils() response = obj.parse_and_return_epg_data([{"fvAEPg": {}}]) assert response == {"dummy-key": "dummy-val"}
def test_get_epg_health(data, dn, expected): # Mock AciUtils methods def dummy_aci_get(self, url): return get_data(data) AciUtils.aci_get = dummy_aci_get AciUtils.proto = "http://" AciUtils.apic_ip = "dummy-apic-ip" obj = AciUtils() response = obj.get_epg_health(dn) assert response == get_data(expected)
def test_get_ap_epg_audit_logs(): # Mock AciUtils methods def dummy_get_mo_related_item(self, mo_dn, item_query_string, item_type): return ['data1', 'data2'] AciUtils.get_mo_related_item = dummy_get_mo_related_item AciUtils.proto = "http://" AciUtils.apic_ip = "dummy-apic-ip" obj = AciUtils() response = obj.get_ap_epg_audit_logs('dummy-dn') assert response == {'auditLogRecords': ['data1', 'data2']}
def test_get_interface(data, node, expected): # Mock apic_util get_node_from_interface def dummy_get_node_from_interface(self, interface): return get_data(node) AciUtils.get_node_from_interface = dummy_get_node_from_interface # Test the function obj = AciUtils() response = obj.get_interface(get_data(data)) assert response == get_data(expected)
def test_apic_fetch_vrf_tenant(data, tn, expected): # Mock AciUtils methods def dummy_aci_get(self, url): return get_data(data) AciUtils.aci_get = dummy_aci_get AciUtils.proto = "http://" AciUtils.apic_ip = "dummy-apic-ip" obj = AciUtils() response = obj.apic_fetch_vrf_tenant(tn) assert response == get_data(expected)
def test_get_mo_related_item(data, mo_dn, item_query_string, item_type, expected): # Mock AciUtils methods def dummy_aci_get(self, url): return data AciUtils.aci_get = dummy_aci_get AciUtils.proto = "http://" AciUtils.apic_ip = "dummy-apic-ip" obj = AciUtils() response = obj.get_mo_related_item(mo_dn, item_query_string, item_type) assert response == expected
def test_parse_and_return_ep_data(data, ep_data, expected): # Mock AciUtils functions def dummy_get_ep_info(self, ep_attr): return get_data(ep_data) def dummy_get_ip_mac_list(ep_child): return [[get_data("data/ip_2.json"), True], [get_data("data/ip_1.json"), False]] AciUtils.get_ep_info = dummy_get_ep_info # Test the function obj = AciUtils() response = obj.parse_and_return_ep_data(get_data(data)) assert response == get_data(expected)
def test_apic_fetch_epg_data(): resp_data = [{"key": "val"}, {"key": "val"}] # Mock AciUtils methods def dummy_aci_get(self, url): return {"imdata": [{}]} def dummy_parse_epg_data(self, data): return resp_data AciUtils.aci_get = dummy_aci_get AciUtils.parse_epg_data = dummy_parse_epg_data AciUtils.epg_url = "dummy-epg-url" obj = AciUtils() response = obj.apic_fetch_epg_data('DummyTn') assert response == resp_data
def test_get_ep_info(data, expected): # Mock AciUtils functions def dummy_get_controller_and_hosting_server(self, ep_child): return ("hyper000", get_data("data/ip_1.json")) def dummy_get_interface(self, ep_child): return "Pod-0/Node-111/eth0/0" def dummy_get_vm_domain_and_name(self, ep_child): return ('DUMMY0-leaf000', 'dummy-vm-name') AciUtils.get_controller_and_hosting_server = dummy_get_controller_and_hosting_server AciUtils.get_interface = dummy_get_interface AciUtils.get_vm_domain_and_name = dummy_get_vm_domain_and_name # Test the function obj = AciUtils() response = obj.get_ep_info(get_data(data)) assert response == get_data(expected)
def test_get_filter_list(data, expected): def dummy_get_mo_related_item(self, mo_dn, item_query_string, item_type): return get_data_json(data) def dummy_login(self): return "dummy-token" AciUtils.login = dummy_login AciUtils.get_mo_related_item = dummy_get_mo_related_item obj = AciUtils() response = plugin_server.get_filter_list( "uni/tn-DummyTn/ap-DummyAp/epg-DummyEpg", obj) assert response == get_data_json(expected)
def test_parse_and_return_epg_data(data, contract, expected): # Mock AciUtils methods def dummy_apic_fetch_bd(self, url): return "Dummy-BD0" def dummy_apic_fetch_vrf(self, url): return "Dummy-VRF" def dummy_apic_fetch_contract(self, url): return get_data(contract) def dummy_get_epg_health(self, url): return "96" AciUtils.apic_fetch_bd = dummy_apic_fetch_bd AciUtils.apic_fetch_vrf = dummy_apic_fetch_vrf AciUtils.apic_fetch_contract = dummy_apic_fetch_contract AciUtils.get_epg_health = dummy_get_epg_health obj = AciUtils() response = obj.parse_and_return_epg_data(get_data(data)) assert response == get_data(expected)
def test_get_ingress_egress(data, expected): def dummy_get_mo_related_item(self, mo_dn, item_query_string, item_type): return get_data_json(data) def dummy_login(self): return "dummy-token" AciUtils.login = dummy_login AciUtils.get_mo_related_item = dummy_get_mo_related_item obj = AciUtils() response = plugin_server.get_ingress_egress( "uni/tn-DummyTn/ap-DummyAp/epg-DummyToEpg", "uni/tn-DummyTn/ap-DummyAp/epg-DummyFromEpg", "subj", "flt", obj) assert response == expected
def test_get_node_from_interface(interface, expected): # Test the function response = AciUtils.get_node_from_interface(interface) assert response == expected
def test_get_dict_records(): response = AciUtils.get_dict_records([1, 2, 3], 'key') assert response == {'key': [1, 2, 3]}
def test_get_ip_mac_list(data, expected): response = AciUtils.get_ip_mac_list(get_data(data)) assert response == get_data(expected)