def test_function_return_value(): """ Проверка работы функции """ list_of_cdp_files = [ "sh_cdp_n_r2.txt", "sh_cdp_n_r5.txt", "sh_cdp_n_r1.txt", "sh_cdp_n_sw1.txt", "sh_cdp_n_r3.txt", "sh_cdp_n_r4.txt", "sh_cdp_n_r6.txt", ] correct_return_value = { "R2": { "Eth 0/0": {"SW1": "Eth 0/2"}, "Eth 0/1": {"R5": "Eth 0/0"}, "Eth 0/2": {"R6": "Eth 0/1"}, }, "R5": {"Eth 0/0": {"R2": "Eth 0/1"}, "Eth 0/1": {"R4": "Eth 0/1"}}, "R1": {"Eth 0/0": {"SW1": "Eth 0/1"}}, "SW1": { "Eth 0/1": {"R1": "Eth 0/0"}, "Eth 0/2": {"R2": "Eth 0/0"}, "Eth 0/3": {"R3": "Eth 0/0"}, "Eth 0/4": {"R4": "Eth 0/0"}, }, "R3": {"Eth 0/0": {"SW1": "Eth 0/3"}}, "R4": {"Eth 0/0": {"SW1": "Eth 0/4"}, "Eth 0/1": {"R5": "Eth 0/1"}}, "R6": {"Eth 0/1": {"R2": "Eth 0/2"}}, } return_value = task_17_3a.generate_topology_from_cdp(list_of_cdp_files) assert return_value != None, "Функция ничего не возвращает" assert ( type(return_value) == dict ), f"По заданию функция должна возвращать словарь, а возвращает {type(return_value).__name__}" assert ( return_value == correct_return_value ), "Функция возвращает неправильное значение"
def test_writing_to_yaml_file(tmpdir): list_of_cdp_files = [ "sh_cdp_n_r2.txt", "sh_cdp_n_r5.txt", "sh_cdp_n_r1.txt", "sh_cdp_n_sw1.txt", "sh_cdp_n_r3.txt", "sh_cdp_n_r4.txt", "sh_cdp_n_r6.txt", ] correct_return_value = { "R2": { "Eth 0/0": {"SW1": "Eth 0/2"}, "Eth 0/1": {"R5": "Eth 0/0"}, "Eth 0/2": {"R6": "Eth 0/1"}, }, "R5": {"Eth 0/0": {"R2": "Eth 0/1"}, "Eth 0/1": {"R4": "Eth 0/1"}}, "R1": {"Eth 0/0": {"SW1": "Eth 0/1"}}, "SW1": { "Eth 0/1": {"R1": "Eth 0/0"}, "Eth 0/2": {"R2": "Eth 0/0"}, "Eth 0/3": {"R3": "Eth 0/0"}, "Eth 0/4": {"R4": "Eth 0/0"}, }, "R3": {"Eth 0/0": {"SW1": "Eth 0/3"}}, "R4": {"Eth 0/0": {"SW1": "Eth 0/4"}, "Eth 0/1": {"R5": "Eth 0/1"}}, "R6": {"Eth 0/1": {"R2": "Eth 0/2"}}, } dest_filename = tmpdir.mkdir("test_tasks").join("topology.yaml") return_value = task_17_3a.generate_topology_from_cdp( list_of_cdp_files, save_to_filename=dest_filename ) assert os.path.exists(dest_filename), "YAML файл не создан" with open(dest_filename) as f: yaml_file_content = yaml.safe_load(f) assert ( yaml_file_content == correct_return_value ), "Топология не записана в YAML файл"
def test_writing_to_yaml_file(tmpdir): list_of_cdp_files = [ 'sh_cdp_n_r2.txt', 'sh_cdp_n_r5.txt', 'sh_cdp_n_r1.txt', 'sh_cdp_n_sw1.txt', 'sh_cdp_n_r3.txt', 'sh_cdp_n_r4.txt', 'sh_cdp_n_r6.txt' ] correct_return_value = { 'R2': { 'Eth 0/0': { 'SW1': 'Eth 0/2' }, 'Eth 0/1': { 'R5': 'Eth 0/0' }, 'Eth 0/2': { 'R6': 'Eth 0/1' } }, 'R5': { 'Eth 0/0': { 'R2': 'Eth 0/1' }, 'Eth 0/1': { 'R4': 'Eth 0/1' } }, 'R1': { 'Eth 0/0': { 'SW1': 'Eth 0/1' } }, 'SW1': { 'Eth 0/1': { 'R1': 'Eth 0/0' }, 'Eth 0/2': { 'R2': 'Eth 0/0' }, 'Eth 0/3': { 'R3': 'Eth 0/0' }, 'Eth 0/4': { 'R4': 'Eth 0/0' } }, 'R3': { 'Eth 0/0': { 'SW1': 'Eth 0/3' } }, 'R4': { 'Eth 0/0': { 'SW1': 'Eth 0/4' }, 'Eth 0/1': { 'R5': 'Eth 0/1' } }, 'R6': { 'Eth 0/1': { 'R2': 'Eth 0/2' } } } dest_filename = tmpdir.mkdir("test_tasks").join("topology.yaml") return_value = task_17_3a.generate_topology_from_cdp( list_of_cdp_files, save_to_filename=dest_filename) assert os.path.exists(dest_filename), "YAML файл не создан" with open(dest_filename) as f: yaml_file_content = yaml.safe_load(f) assert yaml_file_content == correct_return_value, "Топология не записана в YAML файл"
def test_function_return_value(): list_of_cdp_files = [ 'sh_cdp_n_r2.txt', 'sh_cdp_n_r5.txt', 'sh_cdp_n_r1.txt', 'sh_cdp_n_sw1.txt', 'sh_cdp_n_r3.txt', 'sh_cdp_n_r4.txt', 'sh_cdp_n_r6.txt' ] correct_return_value = { 'R2': { 'Eth 0/0': { 'SW1': 'Eth 0/2' }, 'Eth 0/1': { 'R5': 'Eth 0/0' }, 'Eth 0/2': { 'R6': 'Eth 0/1' } }, 'R5': { 'Eth 0/0': { 'R2': 'Eth 0/1' }, 'Eth 0/1': { 'R4': 'Eth 0/1' } }, 'R1': { 'Eth 0/0': { 'SW1': 'Eth 0/1' } }, 'SW1': { 'Eth 0/1': { 'R1': 'Eth 0/0' }, 'Eth 0/2': { 'R2': 'Eth 0/0' }, 'Eth 0/3': { 'R3': 'Eth 0/0' }, 'Eth 0/4': { 'R4': 'Eth 0/0' } }, 'R3': { 'Eth 0/0': { 'SW1': 'Eth 0/3' } }, 'R4': { 'Eth 0/0': { 'SW1': 'Eth 0/4' }, 'Eth 0/1': { 'R5': 'Eth 0/1' } }, 'R6': { 'Eth 0/1': { 'R2': 'Eth 0/2' } } } return_value = task_17_3a.generate_topology_from_cdp(list_of_cdp_files) assert return_value != None, "Функция ничего не возвращает" assert type(return_value) == dict, "Функция должна возвращать словарь" assert return_value == correct_return_value, "Функция возвращает неправильное значение"
def test_function_return_value(): """ Function check """ list_of_cdp_files = [ "sh_cdp_n_r2.txt", "sh_cdp_n_r5.txt", "sh_cdp_n_r1.txt", "sh_cdp_n_sw1.txt", "sh_cdp_n_r3.txt", "sh_cdp_n_r4.txt", "sh_cdp_n_r6.txt", ] correct_return_value = { "R2": { "Eth 0/0": { "SW1": "Eth 0/2" }, "Eth 0/1": { "R5": "Eth 0/0" }, "Eth 0/2": { "R6": "Eth 0/1" }, }, "R5": { "Eth 0/0": { "R2": "Eth 0/1" }, "Eth 0/1": { "R4": "Eth 0/1" } }, "R1": { "Eth 0/0": { "SW1": "Eth 0/1" } }, "SW1": { "Eth 0/1": { "R1": "Eth 0/0" }, "Eth 0/2": { "R2": "Eth 0/0" }, "Eth 0/3": { "R3": "Eth 0/0" }, "Eth 0/4": { "R4": "Eth 0/0" }, }, "R3": { "Eth 0/0": { "SW1": "Eth 0/3" } }, "R4": { "Eth 0/0": { "SW1": "Eth 0/4" }, "Eth 0/1": { "R5": "Eth 0/1" } }, "R6": { "Eth 0/1": { "R2": "Eth 0/2" } }, } return_value = task_17_3a.generate_topology_from_cdp(list_of_cdp_files) assert return_value != None, "The function returns None" assert ( type(return_value) == dict ), f"The function should return a dict, instead it returns a {type(return_value).__name__}" assert ( return_value == correct_return_value), "Function returns wrong value"