Python generate_topology_from_cdp 예제들

예제 #1

파일: test_task_17_2a.py 프로젝트: sie8410/pyneng-online-jan-apr-2019

def test_function_return_value(list_of_cdp_files, sh_cdp_topology_dicts):
    correct_return_value = sh_cdp_topology_dicts

    return_value = task_17_2a.generate_topology_from_cdp(list_of_cdp_files)
    assert return_value != None, "Функция ничего не возвращает"
    assert type(return_value) == dict, "Функция должна возвращать словарь"
    assert return_value == correct_return_value, "Функция возвращает неправильное значение"

예제 #2

파일: test_task_17_2a.py 프로젝트: sie8410/pyneng-online-jan-apr-2019

def test_writing_to_yaml_file(list_of_cdp_files, sh_cdp_topology_dicts, tmpdir):
    dest_filename = tmpdir.mkdir("test_tasks").join("topology.yaml")
    return_value = task_17_2a.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.load(f)
    assert yaml_file_content == sh_cdp_topology_dicts, "Топология не записана в YAML файл"

예제 #3

파일: test_task_17_2a.py 프로젝트: WitalyK/Python_Pyneng

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_2a.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 файл"

예제 #4

파일: test_task_17_2a.py 프로젝트: WitalyK/Python_Pyneng

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_2a.generate_topology_from_cdp(list_of_cdp_files)
    assert return_value != None, "Функция ничего не возвращает"
    assert type(return_value) == dict, "Функция должна возвращать словарь"
    assert return_value == correct_return_value, "Функция возвращает неправильное значение"