From 5c41910d408d8992bfe6ff66d187bf2a60be3ca2 Mon Sep 17 00:00:00 2001
From: Maryam Mokhtarifar <maryam.mokhtarifar@geant.org>
Date: Thu, 20 Mar 2025 09:31:15 +0100
Subject: [PATCH] Mock whois method in tests.
---
test/test_msr_routes.py | 15 +++++++++++++--
1 file changed, 13 insertions(+), 2 deletions(-)
diff --git a/test/test_msr_routes.py b/test/test_msr_routes.py
index 7fd78dfb..15eaa488 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,
--
GitLab