Python get_urlの例

コード例 #1

ファイル: test_action_probe_http.py プロジェクト: powerfulseal/powerfulseal

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/"

コード例 #2

ファイル: test_action_probe_http.py プロジェクト: seeker89/powerfulseal

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"),
    ), )