airbyte_cdk.test.utils.http_mocking

 1# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
 2
 3import re
 4from typing import Any, Mapping
 5
 6from requests_mock import Mocker
 7
 8
 9def register_mock_responses(
10    mocker: Mocker, http_calls: list[Mapping[str, Mapping[str, Any]]]
11) -> None:
12    """Register a list of HTTP request-response pairs."""
13    for call in http_calls:
14        request, response = call["request"], call["response"]
15        matcher = re.compile(request["url"]) if request["is_regex"] else request["url"]
16        mocker.register_uri(request["method"], matcher, **response)
def register_mock_responses( mocker: requests_mock.mocker.Mocker, http_calls: list[typing.Mapping[str, typing.Mapping[str, typing.Any]]]) -> None:
10def register_mock_responses(
11    mocker: Mocker, http_calls: list[Mapping[str, Mapping[str, Any]]]
12) -> None:
13    """Register a list of HTTP request-response pairs."""
14    for call in http_calls:
15        request, response = call["request"], call["response"]
16        matcher = re.compile(request["url"]) if request["is_regex"] else request["url"]
17        mocker.register_uri(request["method"], matcher, **response)

Register a list of HTTP request-response pairs.