Exemplo n.º 1
0
def test_find_cfg_file():
    """
    Search for netmiko_tools config file in the following order:

    NETMIKO_TOOLS_CFG environment variable
    Current directory
    Home directory

    Look for file named: .netmiko.yml or netmiko.yml
    """
    # Search using environment variable (point directly at end file)
    os.environ["NETMIKO_TOOLS_CFG"] = join(RESOURCE_FOLDER, ".netmiko.yml")
    assert utilities.find_cfg_file() == CONFIG_FILENAME

    # Search using environment variable (pointing at directory)
    os.environ["NETMIKO_TOOLS_CFG"] = RESOURCE_FOLDER
    assert utilities.find_cfg_file() == CONFIG_FILENAME

    try:
        cwd = os.getcwd()
        os.chdir(dirname(__file__))

        # Environment var should be preferred over current dir
        assert utilities.find_cfg_file() == CONFIG_FILENAME

        #  Delete env var and verify current dir is returned
        del os.environ["NETMIKO_TOOLS_CFG"]
        assert utilities.find_cfg_file() == "./.netmiko.yml"
    finally:
        # Change directory back to previous state
        os.chdir(cwd)

    # Verify explicit call using full filename
    assert utilities.find_cfg_file(CONFIG_FILENAME) == CONFIG_FILENAME
Exemplo n.º 2
0
def test_find_cfg_file():
    """Try to find a configuration file"""
    assert utilities.find_cfg_file(CONFIG_FILENAME) == CONFIG_FILENAME
Exemplo n.º 3
0
def test_find_cfg_file():
    """Try to find a configuration file"""
    assert utilities.find_cfg_file(CONFIG_FILENAME) == CONFIG_FILENAME
Exemplo n.º 4
0
def save_devices():
    device_file = find_cfg_file()
    print("Saving devices {} ".format(device_file))
    with open(device_file, 'w') as file:
        data = yaml.dump(device_inventory, file)