diff --git a/tests/base_tests/test_napalm_bgp_neighbors.py b/tests/base_tests/test_napalm_bgp_neighbors.py index 40bd1e4b..fc906f09 100644 --- a/tests/base_tests/test_napalm_bgp_neighbors.py +++ b/tests/base_tests/test_napalm_bgp_neighbors.py @@ -204,7 +204,6 @@ def test_contains_peers_at_second_level(transformed_result, host, neighbors): def test_contains_information_about_neighbor(transformed_result, testdata): - print(testdata) host_result = transformed_result[testdata["host"]] host_result.validate() neighbor_details = host_result.result[testdata["peer"]] @@ -228,6 +227,13 @@ def test_contains_information_about_neighbor(transformed_result, testdata): assert neighbor_details == expected +@pytest.mark.parametrize("host, neighbor_amount", [("R1", 2), ("R2", 1)]) +def test_contains_correct_amount_of_hosts(transformed_result, host, neighbor_amount): + host_result = transformed_result[host] + host_result.validate() + assert len(host_result.result) == neighbor_amount + + def test_marks_as_failed_if_task_failed(transformed_result): assert transformed_result["R3"].failed assert transformed_result["R3"].exception is not None diff --git a/tests/base_tests/test_napalm_interfaces.py b/tests/base_tests/test_napalm_interfaces.py index db088511..6a67a76d 100644 --- a/tests/base_tests/test_napalm_interfaces.py +++ b/tests/base_tests/test_napalm_interfaces.py @@ -167,7 +167,6 @@ def test_contains_interface_names_at_second_level(single_result, testdata): def test_contains_information_about_interface(single_result, testdata): - print(testdata) expected = { "is_enabled": testdata["is_enabled"], "is_up": testdata["is_up"], diff --git a/tests/base_tests/test_napalm_lldp_neighbors.py b/tests/base_tests/test_napalm_lldp_neighbors.py index 8ad8d5fa..bb01fd5f 100644 --- a/tests/base_tests/test_napalm_lldp_neighbors.py +++ b/tests/base_tests/test_napalm_lldp_neighbors.py @@ -180,7 +180,6 @@ def test_contains_failed_result_at_second_level_if_task_failed(transformed_resul def test_contains_information_about_neighbor(interface_result, testdata, selftestdata): - print(testdata) expected = { "remote_system_description": REMOTE_SYSTEM_DESCRIPTION, "remote_system_capab": REMOTE_SYSTEM_CAPAB, diff --git a/tests/base_tests/test_netmiko_cdp_neighbors.py b/tests/base_tests/test_netmiko_cdp_neighbors.py index 85bd8b59..db33d0ab 100644 --- a/tests/base_tests/test_netmiko_cdp_neighbors.py +++ b/tests/base_tests/test_netmiko_cdp_neighbors.py @@ -200,7 +200,6 @@ def test_contains_neighbors_at_second_level(transformed_result, host, neighbors) def test_contains_information_about_neighbor(transformed_result, testdata): - print(testdata) host = testdata["host"] remote_host = testdata["remote_host"] diff --git a/tests/base_tests/test_netmiko_ospf_neighbors.py b/tests/base_tests/test_netmiko_ospf_neighbors.py index a7314e77..b096d732 100644 --- a/tests/base_tests/test_netmiko_ospf_neighbors.py +++ b/tests/base_tests/test_netmiko_ospf_neighbors.py @@ -235,7 +235,6 @@ def test_contains_neighbors_at_second_level(transformed_result, host, neighbors) def test_contains_information_about_neighbor(transformed_result, testdata): host_result = transformed_result[testdata["host"]] host_result.validate() - print(testdata) neighbor_details = host_result.result[testdata["neighbor_id"]] expected = { "neighbor_id": testdata["neighbor_id"], @@ -248,6 +247,13 @@ def test_contains_information_about_neighbor(transformed_result, testdata): assert neighbor_details == expected +@pytest.mark.parametrize("host, neighbor_amount", [("R1", 3), ("R2", 3)]) +def test_contains_correct_amount_of_hosts(transformed_result, host, neighbor_amount): + host_result = transformed_result[host] + host_result.validate() + assert len(host_result.result) == neighbor_amount + + def test_marks_as_failed_if_task_failed(transformed_result): assert transformed_result["R3"].failed assert transformed_result["R3"].exception is not None