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"
