def test_get_url_simple(action_probe_http):
mock_resp = MagicMock()
schema = dict(
target=dict(
url="http://10.10.10.10:80"
)
)
url = action_probe_http.get_url(schema)
assert url == "http://10.10.10.10:80/"
def test_get_url_service(action_probe_http):
mock_resp = MagicMock()
mock_resp.spec.cluster_ip = "10.10.10.2"
action_probe_http.k8s_inventory.get_service = MagicMock(
return_value=mock_resp)
schema = dict(target=dict(service=dict(
name="somename",
namespace="somenamespace",
)))
url = action_probe_http.get_url(schema)
assert url == "http://10.10.10.2:80/"
assert action_probe_http.k8s_inventory.get_service.call_count == 1
assert action_probe_http.k8s_inventory.get_service.call_args == (dict(
name=schema.get("target").get("service").get("name"),
namespace=schema.get("target").get("service").get("namespace"),
), )