airbyte-cdk 0.72.1__py3-none-any.whl → 6.13.1.dev4106__py3-none-any.whl

unit_tests/sources/streams/http/requests_native_auth/test_requests_native_auth.py

@@ -1,423 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-
-import json
-import logging
-from typing import Optional, Union
-from unittest.mock import Mock
-
-import freezegun
-import pendulum
-import pytest
-import requests
-from airbyte_cdk.models import FailureType, OrchestratorType, Type
-from airbyte_cdk.sources.streams.http.requests_native_auth import (
-    BasicHttpAuthenticator,
-    MultipleTokenAuthenticator,
-    Oauth2Authenticator,
-    SingleUseRefreshTokenOauth2Authenticator,
-    TokenAuthenticator,
-)
-from airbyte_cdk.utils import AirbyteTracedException
-from requests import Response
-from requests.exceptions import RequestException
-
-LOGGER = logging.getLogger(__name__)
-
-resp = Response()
-
-
-def test_token_authenticator():
-    """
-    Should match passed in token, no matter how many times token is retrieved.
-    """
-    token_auth = TokenAuthenticator(token="test-token")
-    header1 = token_auth.get_auth_header()
-    header2 = token_auth.get_auth_header()
-
-    prepared_request = requests.PreparedRequest()
-    prepared_request.headers = {}
-    token_auth(prepared_request)
-
-    assert {"Authorization": "Bearer test-token"} == prepared_request.headers
-    assert {"Authorization": "Bearer test-token"} == header1
-    assert {"Authorization": "Bearer test-token"} == header2
-
-
-def test_basic_http_authenticator():
-    """
-    Should match passed in token, no matter how many times token is retrieved.
-    """
-    token_auth = BasicHttpAuthenticator(username="user", password="password")
-    header1 = token_auth.get_auth_header()
-    header2 = token_auth.get_auth_header()
-
-    prepared_request = requests.PreparedRequest()
-    prepared_request.headers = {}
-    token_auth(prepared_request)
-
-    assert {"Authorization": "Basic dXNlcjpwYXNzd29yZA=="} == prepared_request.headers
-    assert {"Authorization": "Basic dXNlcjpwYXNzd29yZA=="} == header1
-    assert {"Authorization": "Basic dXNlcjpwYXNzd29yZA=="} == header2
-
-
-def test_multiple_token_authenticator():
-    multiple_token_auth = MultipleTokenAuthenticator(tokens=["token1", "token2"])
-    header1 = multiple_token_auth.get_auth_header()
-    header2 = multiple_token_auth.get_auth_header()
-    header3 = multiple_token_auth.get_auth_header()
-
-    prepared_request = requests.PreparedRequest()
-    prepared_request.headers = {}
-    multiple_token_auth(prepared_request)
-
-    assert {"Authorization": "Bearer token2"} == prepared_request.headers
-    assert {"Authorization": "Bearer token1"} == header1
-    assert {"Authorization": "Bearer token2"} == header2
-    assert {"Authorization": "Bearer token1"} == header3
-
-
-class TestOauth2Authenticator:
-    """
-    Test class for OAuth2Authenticator.
-    """
-
-    refresh_endpoint = "refresh_end"
-    client_id = "client_id"
-    client_secret = "client_secret"
-    refresh_token = "refresh_token"
-
-    def test_get_auth_header_fresh(self, mocker):
-        """
-        Should not retrieve new token if current token is valid.
-        """
-        oauth = Oauth2Authenticator(
-            token_refresh_endpoint=TestOauth2Authenticator.refresh_endpoint,
-            client_id=TestOauth2Authenticator.client_id,
-            client_secret=TestOauth2Authenticator.client_secret,
-            refresh_token=TestOauth2Authenticator.refresh_token,
-        )
-
-        mocker.patch.object(Oauth2Authenticator, "refresh_access_token", return_value=("access_token", 1000))
-        header = oauth.get_auth_header()
-        assert {"Authorization": "Bearer access_token"} == header
-
-    def test_get_auth_header_expired(self, mocker):
-        """
-        Should retrieve new token if current token is expired.
-        """
-        oauth = Oauth2Authenticator(
-            token_refresh_endpoint=TestOauth2Authenticator.refresh_endpoint,
-            client_id=TestOauth2Authenticator.client_id,
-            client_secret=TestOauth2Authenticator.client_secret,
-            refresh_token=TestOauth2Authenticator.refresh_token,
-        )
-
-        expire_immediately = 0
-        mocker.patch.object(Oauth2Authenticator, "refresh_access_token", return_value=("access_token_1", expire_immediately))
-        oauth.get_auth_header()  # Set the first expired token.
-
-        valid_100_secs = 100
-        mocker.patch.object(Oauth2Authenticator, "refresh_access_token", return_value=("access_token_2", valid_100_secs))
-        header = oauth.get_auth_header()
-        assert {"Authorization": "Bearer access_token_2"} == header
-
-    def test_refresh_request_body(self):
-        """
-        Request body should match given configuration.
-        """
-        scopes = ["scope1", "scope2"]
-        oauth = Oauth2Authenticator(
-            token_refresh_endpoint="refresh_end",
-            client_id="some_client_id",
-            client_secret="some_client_secret",
-            refresh_token="some_refresh_token",
-            scopes=["scope1", "scope2"],
-            token_expiry_date=pendulum.now().add(days=3),
-            grant_type="some_grant_type",
-            refresh_request_body={"custom_field": "in_outbound_request", "another_field": "exists_in_body", "scopes": ["no_override"]},
-        )
-        body = oauth.build_refresh_request_body()
-        expected = {
-            "grant_type": "some_grant_type",
-            "client_id": "some_client_id",
-            "client_secret": "some_client_secret",
-            "refresh_token": "some_refresh_token",
-            "scopes": scopes,
-            "custom_field": "in_outbound_request",
-            "another_field": "exists_in_body",
-        }
-        assert body == expected
-
-    def test_refresh_access_token(self, mocker):
-        oauth = Oauth2Authenticator(
-            token_refresh_endpoint="refresh_end",
-            client_id="some_client_id",
-            client_secret="some_client_secret",
-            refresh_token="some_refresh_token",
-            scopes=["scope1", "scope2"],
-            token_expiry_date=pendulum.now().add(days=3),
-            refresh_request_body={"custom_field": "in_outbound_request", "another_field": "exists_in_body", "scopes": ["no_override"]},
-        )
-
-        resp.status_code = 200
-        mocker.patch.object(resp, "json", return_value={"access_token": "access_token", "expires_in": 1000})
-        mocker.patch.object(requests, "request", side_effect=mock_request, autospec=True)
-        token, expires_in = oauth.refresh_access_token()
-
-        assert isinstance(expires_in, int)
-        assert ("access_token", 1000) == (token, expires_in)
-
-        # Test with expires_in as str
-        mocker.patch.object(resp, "json", return_value={"access_token": "access_token", "expires_in": "2000"})
-        token, expires_in = oauth.refresh_access_token()
-
-        assert isinstance(expires_in, str)
-        assert ("access_token", "2000") == (token, expires_in)
-
-        # Test with expires_in as str
-        mocker.patch.object(resp, "json", return_value={"access_token": "access_token", "expires_in": "2022-04-24T00:00:00Z"})
-        token, expires_in = oauth.refresh_access_token()
-
-        assert isinstance(expires_in, str)
-        assert ("access_token", "2022-04-24T00:00:00Z") == (token, expires_in)
-
-    @pytest.mark.parametrize(
-        "expires_in_response, token_expiry_date_format, expected_token_expiry_date",
-        [
-            (3600, None, pendulum.datetime(year=2022, month=1, day=1, hour=1)),
-            ("90012", None, pendulum.datetime(year=2022, month=1, day=2, hour=1, second=12)),
-            ("2024-02-28", "YYYY-MM-DD", pendulum.datetime(year=2024, month=2, day=28)),
-            ("2022-02-12T00:00:00.000000+00:00", "YYYY-MM-DDTHH:mm:ss.SSSSSSZ", pendulum.datetime(year=2022, month=2, day=12)),
-        ],
-        ids=["seconds", "string_of_seconds", "simple_date", "simple_datetime"],
-    )
-    @freezegun.freeze_time("2022-01-01")
-    def test_parse_refresh_token_lifespan(
-        self,
-        mocker,
-        expires_in_response: Union[str, int],
-        token_expiry_date_format: Optional[str],
-        expected_token_expiry_date: pendulum.DateTime,
-    ):
-        oauth = Oauth2Authenticator(
-            token_refresh_endpoint="refresh_end",
-            client_id="some_client_id",
-            client_secret="some_client_secret",
-            refresh_token="some_refresh_token",
-            scopes=["scope1", "scope2"],
-            token_expiry_date=pendulum.now().subtract(days=3),
-            token_expiry_date_format=token_expiry_date_format,
-            token_expiry_is_time_of_expiration=bool(token_expiry_date_format),
-            refresh_request_body={"custom_field": "in_outbound_request", "another_field": "exists_in_body", "scopes": ["no_override"]},
-        )
-
-        resp.status_code = 200
-        mocker.patch.object(resp, "json", return_value={"access_token": "access_token", "expires_in": expires_in_response})
-        mocker.patch.object(requests, "request", side_effect=mock_request, autospec=True)
-        token, expire_in = oauth.refresh_access_token()
-        expires_datetime = oauth._parse_token_expiration_date(expire_in)
-
-        assert isinstance(expires_datetime, pendulum.DateTime)
-        assert ("access_token", expected_token_expiry_date) == (token, expires_datetime)
-
-    @pytest.mark.parametrize("error_code", (429, 500, 502, 504))
-    def test_refresh_access_token_retry(self, error_code, requests_mock):
-        oauth = Oauth2Authenticator(
-            f"https://{TestOauth2Authenticator.refresh_endpoint}",
-            TestOauth2Authenticator.client_id,
-            TestOauth2Authenticator.client_secret,
-            TestOauth2Authenticator.refresh_token,
-        )
-        requests_mock.post(
-            f"https://{TestOauth2Authenticator.refresh_endpoint}",
-            [{"status_code": error_code}, {"status_code": error_code}, {"json": {"access_token": "token", "expires_in": 10}}],
-        )
-        token, expires_in = oauth.refresh_access_token()
-        assert isinstance(expires_in, int)
-        assert (token, expires_in) == ("token", 10)
-        assert requests_mock.call_count == 3
-
-    def test_auth_call_method(self, mocker):
-        oauth = Oauth2Authenticator(
-            token_refresh_endpoint=TestOauth2Authenticator.refresh_endpoint,
-            client_id=TestOauth2Authenticator.client_id,
-            client_secret=TestOauth2Authenticator.client_secret,
-            refresh_token=TestOauth2Authenticator.refresh_token,
-        )
-
-        mocker.patch.object(Oauth2Authenticator, "refresh_access_token", return_value=("access_token", 1000))
-        prepared_request = requests.PreparedRequest()
-        prepared_request.headers = {}
-        oauth(prepared_request)
-
-        assert {"Authorization": "Bearer access_token"} == prepared_request.headers
-
-    @pytest.mark.parametrize(
-        ("config_codes", "response_code", "config_key", "response_key", "config_values", "response_value", "wrapped"),
-        (
-            ((400,), 400, "error", "error", ("invalid_grant",), "invalid_grant", True),
-            ((401,), 400, "error", "error", ("invalid_grant",), "invalid_grant", False),
-            ((400,), 400, "error_key", "error", ("invalid_grant",), "invalid_grant", False),
-            ((400,), 400, "error", "error", ("invalid_grant",), "valid_grant", False),
-            ((), 400, "", "error", (), "valid_grant", False),
-        ),
-    )
-    def test_refresh_access_token_wrapped(
-        self, requests_mock, config_codes, response_code, config_key, response_key, config_values, response_value, wrapped
-    ):
-        oauth = Oauth2Authenticator(
-            f"https://{TestOauth2Authenticator.refresh_endpoint}",
-            TestOauth2Authenticator.client_id,
-            TestOauth2Authenticator.client_secret,
-            TestOauth2Authenticator.refresh_token,
-            refresh_token_error_status_codes=config_codes,
-            refresh_token_error_key=config_key,
-            refresh_token_error_values=config_values,
-        )
-        error_content = {response_key: response_value}
-        requests_mock.post(f"https://{TestOauth2Authenticator.refresh_endpoint}", status_code=response_code, json=error_content)
-
-        exception_to_raise = AirbyteTracedException if wrapped else RequestException
-        with pytest.raises(exception_to_raise) as exc_info:
-            oauth.refresh_access_token()
-
-        if wrapped:
-            error_message = "Refresh token is invalid or expired. Please re-authenticate from Sources/<your source>/Settings."
-            assert exc_info.value.internal_message == error_message
-            assert exc_info.value.message == error_message
-            assert exc_info.value.failure_type == FailureType.config_error
-
-
-class TestSingleUseRefreshTokenOauth2Authenticator:
-    @pytest.fixture
-    def connector_config(self):
-        return {
-            "credentials": {
-                "access_token": "my_access_token",
-                "refresh_token": "my_refresh_token",
-                "client_id": "my_client_id",
-                "client_secret": "my_client_secret",
-                "token_expiry_date": "2022-12-31T00:00:00+00:00",
-            }
-        }
-
-    @pytest.fixture
-    def invalid_connector_config(self):
-        return {"no_credentials_key": "foo"}
-
-    def test_init(self, connector_config):
-        authenticator = SingleUseRefreshTokenOauth2Authenticator(
-            connector_config,
-            token_refresh_endpoint="foobar",
-            client_id=connector_config["credentials"]["client_id"],
-            client_secret=connector_config["credentials"]["client_secret"],
-        )
-        assert authenticator.access_token == connector_config["credentials"]["access_token"]
-        assert authenticator.get_refresh_token() == connector_config["credentials"]["refresh_token"]
-        assert authenticator.get_token_expiry_date() == pendulum.parse(connector_config["credentials"]["token_expiry_date"])
-
-    @freezegun.freeze_time("2022-12-31")
-    @pytest.mark.parametrize(
-        "test_name, expires_in_value, expiry_date_format, expected_expiry_date",
-        [
-            ("number_of_seconds", 42, None, "2022-12-31T00:00:42+00:00"),
-            ("string_of_seconds", "42", None, "2022-12-31T00:00:42+00:00"),
-            ("date_format", "2023-04-04", "YYYY-MM-DD", "2023-04-04T00:00:00+00:00"),
-        ],
-    )
-    def test_given_no_message_repository_get_access_token(
-        self, test_name, expires_in_value, expiry_date_format, expected_expiry_date, capsys, mocker, connector_config
-    ):
-        authenticator = SingleUseRefreshTokenOauth2Authenticator(
-            connector_config,
-            token_refresh_endpoint="foobar",
-            client_id=connector_config["credentials"]["client_id"],
-            client_secret=connector_config["credentials"]["client_secret"],
-            token_expiry_date_format=expiry_date_format,
-        )
-        authenticator.refresh_access_token = mocker.Mock(return_value=("new_access_token", expires_in_value, "new_refresh_token"))
-        authenticator.token_has_expired = mocker.Mock(return_value=True)
-        access_token = authenticator.get_access_token()
-        captured = capsys.readouterr()
-        airbyte_message = json.loads(captured.out)
-        expected_new_config = connector_config.copy()
-        expected_new_config["credentials"]["access_token"] = "new_access_token"
-        expected_new_config["credentials"]["refresh_token"] = "new_refresh_token"
-        expected_new_config["credentials"]["token_expiry_date"] = expected_expiry_date
-        assert airbyte_message["control"]["connectorConfig"]["config"] == expected_new_config
-        assert authenticator.access_token == access_token == "new_access_token"
-        assert authenticator.get_refresh_token() == "new_refresh_token"
-        assert authenticator.get_token_expiry_date() > pendulum.now()
-        authenticator.token_has_expired = mocker.Mock(return_value=False)
-        access_token = authenticator.get_access_token()
-        captured = capsys.readouterr()
-        assert not captured.out
-        assert authenticator.access_token == access_token == "new_access_token"
-
-    def test_given_message_repository_when_get_access_token_then_emit_message(self, mocker, connector_config):
-        message_repository = Mock()
-        authenticator = SingleUseRefreshTokenOauth2Authenticator(
-            connector_config,
-            token_refresh_endpoint="foobar",
-            client_id=connector_config["credentials"]["client_id"],
-            client_secret=connector_config["credentials"]["client_secret"],
-            token_expiry_date_format="YYYY-MM-DD",
-            message_repository=message_repository,
-        )
-        authenticator.refresh_access_token = mocker.Mock(return_value=("new_access_token", "2023-04-04", "new_refresh_token"))
-        authenticator.token_has_expired = mocker.Mock(return_value=True)
-
-        authenticator.get_access_token()
-
-        emitted_message = message_repository.emit_message.call_args_list[0].args[0]
-        assert emitted_message.type == Type.CONTROL
-        assert emitted_message.control.type == OrchestratorType.CONNECTOR_CONFIG
-        assert emitted_message.control.connectorConfig.config["credentials"]["access_token"] == "new_access_token"
-        assert emitted_message.control.connectorConfig.config["credentials"]["refresh_token"] == "new_refresh_token"
-        assert emitted_message.control.connectorConfig.config["credentials"]["token_expiry_date"] == "2023-04-04T00:00:00+00:00"
-        assert emitted_message.control.connectorConfig.config["credentials"]["client_id"] == "my_client_id"
-        assert emitted_message.control.connectorConfig.config["credentials"]["client_secret"] == "my_client_secret"
-
-    def test_given_message_repository_when_get_access_token_then_log_request(self, mocker, connector_config):
-        message_repository = Mock()
-        authenticator = SingleUseRefreshTokenOauth2Authenticator(
-            connector_config,
-            token_refresh_endpoint="foobar",
-            client_id=connector_config["credentials"]["client_id"],
-            client_secret=connector_config["credentials"]["client_secret"],
-            message_repository=message_repository,
-        )
-        mocker.patch("airbyte_cdk.sources.streams.http.requests_native_auth.abstract_oauth.requests.request")
-        mocker.patch(
-            "airbyte_cdk.sources.streams.http.requests_native_auth.abstract_oauth.format_http_message", return_value="formatted json"
-        )
-        authenticator.token_has_expired = mocker.Mock(return_value=True)
-
-        authenticator.get_access_token()
-
-        assert message_repository.log_message.call_count == 1
-
-    def test_refresh_access_token(self, mocker, connector_config):
-        authenticator = SingleUseRefreshTokenOauth2Authenticator(
-            connector_config,
-            token_refresh_endpoint="foobar",
-            client_id=connector_config["credentials"]["client_id"],
-            client_secret=connector_config["credentials"]["client_secret"],
-        )
-
-        authenticator._get_refresh_access_token_response = mocker.Mock(
-            return_value={
-                authenticator.get_access_token_name(): "new_access_token",
-                authenticator.get_expires_in_name(): "42",
-                authenticator.get_refresh_token_name(): "new_refresh_token",
-            }
-        )
-        assert authenticator.refresh_access_token() == ("new_access_token", "42", "new_refresh_token")
-
-
-def mock_request(method, url, data):
-    if url == "refresh_end":
-        return resp
-    raise Exception(f"Error while refreshing access token with request: {method}, {url}, {data}")

unit_tests/sources/streams/http/test_availability_strategy.py

@@ -1,180 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-
-import logging
-from typing import Any, Iterable, List, Mapping, Optional, Tuple
-
-import pytest
-import requests
-from airbyte_cdk.sources import AbstractSource
-from airbyte_cdk.sources.streams import Stream
-from airbyte_cdk.sources.streams.http.availability_strategy import HttpAvailabilityStrategy
-from airbyte_cdk.sources.streams.http.http import HttpStream
-from requests import HTTPError
-
-logger = logging.getLogger("airbyte")
-
-
-class MockHttpStream(HttpStream):
-    url_base = "https://test_base_url.com"
-    primary_key = ""
-
-    def __init__(self, **kwargs):
-        super().__init__(**kwargs)
-        self.resp_counter = 1
-
-    def next_page_token(self, response: requests.Response) -> Optional[Mapping[str, Any]]:
-        return None
-
-    def path(self, **kwargs) -> str:
-        return ""
-
-    def parse_response(self, response: requests.Response, **kwargs) -> Iterable[Mapping]:
-        stub_resp = {"data": self.resp_counter}
-        self.resp_counter += 1
-        yield stub_resp
-
-    pass
-
-    def retry_factor(self) -> float:
-        return 0.01
-
-
-@pytest.mark.parametrize(
-    ("status_code", "json_contents", "expected_is_available", "expected_messages"),
-    [
-        (
-            403,
-            {"error": "Something went wrong"},
-            False,
-            [
-                "This is most likely due to insufficient permissions on the credentials in use.",
-                "Something went wrong",
-            ],
-        ),
-        (200, {}, True, []),
-    ],
-)
-@pytest.mark.parametrize(
-    ("include_source", "expected_docs_url_messages"),
-    [
-        (True, ["Please visit https://docs.airbyte.com/integrations/sources/MockSource to learn more."]),
-        (False, ["Please visit the connector's documentation to learn more."]),
-    ],
-)
-@pytest.mark.parametrize("records_as_list", [True, False])
-def test_default_http_availability_strategy(
-    mocker,
-    status_code,
-    json_contents,
-    expected_is_available,
-    expected_messages,
-    include_source,
-    expected_docs_url_messages,
-    records_as_list,
-):
-    class MockListHttpStream(MockHttpStream):
-        def read_records(self, *args, **kvargs):
-            if records_as_list:
-                return list(super().read_records(*args, **kvargs))
-            else:
-                return super().read_records(*args, **kvargs)
-
-    http_stream = MockListHttpStream()
-    assert isinstance(http_stream.availability_strategy, HttpAvailabilityStrategy)
-
-    class MockResponseWithJsonContents(requests.Response, mocker.MagicMock):
-        def __init__(self, *args, **kvargs):
-            mocker.MagicMock.__init__(self)
-            requests.Response.__init__(self, **kvargs)
-            self.json = mocker.MagicMock()
-
-    class MockSource(AbstractSource):
-        def __init__(self, streams: List[Stream] = None):
-            self._streams = streams
-
-        def check_connection(self, logger: logging.Logger, config: Mapping[str, Any]) -> Tuple[bool, Optional[Any]]:
-            return True, ""
-
-        def streams(self, config: Mapping[str, Any]) -> List[Stream]:
-            if not self._streams:
-                raise Exception("Stream is not set")
-            return self._streams
-
-    response = MockResponseWithJsonContents()
-    response.status_code = status_code
-    response.json.return_value = json_contents
-    mocker.patch.object(requests.Session, "send", return_value=response)
-
-    if include_source:
-        source = MockSource(streams=[http_stream])
-        actual_is_available, reason = http_stream.check_availability(logger, source)
-    else:
-        actual_is_available, reason = http_stream.check_availability(logger)
-
-    assert expected_is_available == actual_is_available
-    if expected_is_available:
-        assert reason is None
-    else:
-        all_expected_messages = expected_messages + expected_docs_url_messages
-        for message in all_expected_messages:
-            assert message in reason
-
-
-def test_http_availability_raises_unhandled_error(mocker):
-    http_stream = MockHttpStream()
-    assert isinstance(http_stream.availability_strategy, HttpAvailabilityStrategy)
-
-    req = requests.Response()
-    req.status_code = 404
-    mocker.patch.object(requests.Session, "send", return_value=req)
-
-    with pytest.raises(HTTPError):
-        http_stream.check_availability(logger)
-
-
-def test_send_handles_retries_when_checking_availability(mocker, caplog):
-    http_stream = MockHttpStream()
-    assert isinstance(http_stream.availability_strategy, HttpAvailabilityStrategy)
-
-    req_1 = requests.Response()
-    req_1.status_code = 429
-    req_2 = requests.Response()
-    req_2.status_code = 503
-    req_3 = requests.Response()
-    req_3.status_code = 200
-    mock_send = mocker.patch.object(requests.Session, "send", side_effect=[req_1, req_2, req_3])
-
-    with caplog.at_level(logging.INFO):
-        stream_is_available, _ = http_stream.check_availability(logger)
-
-    assert stream_is_available
-    assert mock_send.call_count == 3
-    for message in ["Caught retryable error", "Response Code: 429", "Response Code: 503"]:
-        assert message in caplog.text
-
-
-@pytest.mark.parametrize("records_as_list", [True, False])
-def test_http_availability_strategy_on_empty_stream(mocker, records_as_list):
-    class MockEmptyHttpStream(mocker.MagicMock, MockHttpStream):
-        def __init__(self, *args, **kvargs):
-            mocker.MagicMock.__init__(self)
-            self.read_records = mocker.MagicMock()
-
-    empty_stream = MockEmptyHttpStream()
-    assert isinstance(empty_stream, HttpStream)
-
-    assert isinstance(empty_stream.availability_strategy, HttpAvailabilityStrategy)
-
-    # Generator should have no values to generate
-    if records_as_list:
-        empty_stream.read_records.return_value = []
-    else:
-        empty_stream.read_records.return_value = iter([])
-
-    logger = logging.getLogger("airbyte.test-source")
-    stream_is_available, _ = empty_stream.check_availability(logger)
-
-    assert stream_is_available
-    assert empty_stream.read_records.called