diff --git a/test/test_msr_routes.py b/test/test_msr_routes.py index 7fd78dfb966443c3775436999ec441dbbc27e487..15eaa48819568d92f58e0682e0fe4d999c3fc24a 100644 --- a/test/test_msr_routes.py +++ b/test/test_msr_routes.py @@ -338,6 +338,17 @@ def test_get_mdvpn_peerings(client, mocked_redis): assert response_data # test data is non-empty +@pytest.fixture +def whois(mocker): + check_output = mocker.patch( + "inventory_provider.routes.msr.subprocess.check_output" + ) + check_output.return_value = """AS Name +RCCN Fundacao para a Ciencia e a Tecnologia, I.P., PT""".encode( + "utf-8" + ) + + @pytest.mark.parametrize('endpoint_variant', [ "", # default, no filter "/1853", @@ -348,7 +359,7 @@ def test_get_mdvpn_peerings(client, mocked_redis): "/1853?instance=IAS", "/1853?group=IAS-NRENS&instance=IAS" ]) -def test_get_asn_peers_get(endpoint_variant, client, mocked_redis): +def test_get_asn_peers_get(endpoint_variant, client, mocked_redis, whois): rv = client.get( f'/msr/asn-peers{endpoint_variant}', headers=DEFAULT_REQUEST_HEADERS @@ -370,7 +381,7 @@ def test_get_asn_peers_get(endpoint_variant, client, mocked_redis): ("/1853", '{"instance": "IAS"}'), ("/1853", '{"group": "IAS-NRENS", "instance": "IAS"}') ]) -def test_get_asn_peers_post(endpoint_variant, post_body, client, mocked_redis): +def test_get_asn_peers_post(endpoint_variant, post_body, client, mocked_redis, whois): rv = client.post( f'/msr/asn-peers{endpoint_variant}', headers=DEFAULT_REQUEST_HEADERS,