def test_parse_cli_args_basic_none(): args = ["somethingelse", "notrelevant"] args = parse_cli_args(args) assert args["limit"] is False assert args["groups"] is False assert args["skip_tags"] == [] assert args["run_tags"] == [] assert args["workers"] is False
def test_patch_inventory_limit_host_ignore_case(): args = ["-l", "UPPER-HOST"] args = parse_cli_args(args) nr = InitNornir( inventory={ "plugin": "nornir.plugins.inventory.simple.SimpleInventory", "options": { "host_file": f"{TEST_DIR}_test_nornir_inventory/hosts.yaml", "group_file": f"{TEST_DIR}_test_nornir_inventory/groups.yaml", }, }, logging={"enabled": False}, ) nr.inventory = patch_inventory(args, nr.inventory) assert set(nr.inventory.hosts.keys()) == {"UPPER-HOST"}
def test_patch_inventory_basic_limit_group_invalid(): args = ["-g", "eos1234"] args = parse_cli_args(args) nr = InitNornir( inventory={ "plugin": "nornir.plugins.inventory.simple.SimpleInventory", "options": { "host_file": f"{TEST_DIR}_test_nornir_inventory/hosts.yaml", "group_file": f"{TEST_DIR}_test_nornir_inventory/groups.yaml", }, }, logging={"enabled": False}, ) nr.inventory = patch_inventory(args, nr.inventory) assert set(nr.inventory.hosts.keys()) == set()
def test_patch_config_basic_limit_workers(): args = ["-w", "10"] args = parse_cli_args(args) nr = InitNornir( inventory={ "plugin": "nornir.plugins.inventory.simple.SimpleInventory", "options": { "host_file": f"{TEST_DIR}_test_nornir_inventory/hosts.yaml", "group_file": f"{TEST_DIR}_test_nornir_inventory/groups.yaml", }, }, logging={"enabled": False}, ) nr.config = patch_config(args, nr.config) assert nr.config.core.num_workers == 10
def test_parse_cli_args_basic_short(): args = [ "-l", "sea-eos-1", "-g", "sea", "-s", "deploy_configs", "-t", "render_configs", "-w", "10", ] args = parse_cli_args(args) assert args["limit"] == {"sea-eos-1"} assert args["groups"] == {"sea"} assert args["skip_tags"] == {"deploy_configs"} assert args["run_tags"] == {"render_configs"} assert args["workers"] == 10
def test_parse_cli_args_basic_long(): args = [ "--limit", "sea-eos-1", "--groups", "sea", "--skip", "deploy_configs", "--tags", "render_configs", "--workers", "10", "--disable-delegate", ] args = parse_cli_args(args) assert args["limit"] == {"sea-eos-1"} assert args["groups"] == {"sea"} assert args["skip_tags"] == {"deploy_configs"} assert args["run_tags"] == {"render_configs"} assert args["workers"] == 10