airbyte-cdk 0.72.1__py3-none-any.whl → 6.17.1.dev0__py3-none-any.whl

unit_tests/sources/declarative/requesters/test_interpolated_request_input_provider.py

@@ -1,32 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-
-import pytest as pytest
-from airbyte_cdk.sources.declarative.interpolation.interpolated_mapping import InterpolatedMapping
-from airbyte_cdk.sources.declarative.requesters.request_options.interpolated_request_input_provider import InterpolatedRequestInputProvider
-
-
-@pytest.mark.parametrize(
-    "test_name, input_request_data, expected_request_data",
-    [
-        ("test_static_map_data", {"a_static_request_param": "a_static_value"}, {"a_static_request_param": "a_static_value"}),
-        ("test_map_depends_on_stream_slice", {"read_from_slice": "{{ stream_slice['slice_key'] }}"}, {"read_from_slice": "slice_value"}),
-        ("test_map_depends_on_config", {"read_from_config": "{{ config['config_key'] }}"}, {"read_from_config": "value_of_config"}),
-        (
-            "test_map_depends_on_parameters",
-            {"read_from_parameters": "{{ parameters['read_from_parameters'] }}"},
-            {"read_from_parameters": "value_of_parameters"},
-        ),
-        ("test_defaults_to_empty_dictionary", None, {}),
-    ],
-)
-def test_initialize_interpolated_mapping_request_input_provider(test_name, input_request_data, expected_request_data):
-    config = {"config_key": "value_of_config"}
-    stream_slice = {"slice_key": "slice_value"}
-    parameters = {"read_from_parameters": "value_of_parameters"}
-    provider = InterpolatedRequestInputProvider(request_inputs=input_request_data, config=config, parameters=parameters)
-    actual_request_data = provider.eval_request_inputs(stream_state={}, stream_slice=stream_slice)
-
-    assert isinstance(provider._interpolator, InterpolatedMapping)
-    assert actual_request_data == expected_request_data

unit_tests/sources/declarative/retrievers/__init__.py

@@ -1,3 +0,0 @@
-#
-# Copyright (c) 2021 Airbyte, Inc., all rights reserved.
-#

unit_tests/sources/declarative/retrievers/test_simple_retriever.py

@@ -1,542 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-from unittest.mock import MagicMock, Mock, patch
-
-import pytest
-import requests
-from airbyte_cdk.models import AirbyteLogMessage, AirbyteMessage, Level, SyncMode, Type
-from airbyte_cdk.sources.declarative.auth.declarative_authenticator import NoAuth
-from airbyte_cdk.sources.declarative.incremental import Cursor, DatetimeBasedCursor
-from airbyte_cdk.sources.declarative.partition_routers import SinglePartitionRouter
-from airbyte_cdk.sources.declarative.requesters.error_handlers.response_status import ResponseStatus
-from airbyte_cdk.sources.declarative.requesters.request_option import RequestOptionType
-from airbyte_cdk.sources.declarative.requesters.requester import HttpMethod
-from airbyte_cdk.sources.declarative.retrievers.simple_retriever import SimpleRetriever, SimpleRetrieverTestReadDecorator
-from airbyte_cdk.sources.declarative.types import Record
-
-A_SLICE_STATE = {"slice_state": "slice state value"}
-A_STREAM_SLICE = {"stream slice": "slice value"}
-A_STREAM_STATE = {"stream state": "state value"}
-
-primary_key = "pk"
-records = [{"id": 1}, {"id": 2}]
-request_response_logs = [
-    AirbyteLogMessage(level=Level.INFO, message="request:{}"),
-    AirbyteLogMessage(level=Level.INFO, message="response{}"),
-]
-config = {}
-
-
-@patch.object(SimpleRetriever, "_read_pages", return_value=iter([]))
-def test_simple_retriever_full(mock_http_stream):
-    requester = MagicMock()
-    request_params = {"param": "value"}
-    requester.get_request_params.return_value = request_params
-
-    paginator = MagicMock()
-    next_page_token = {"cursor": "cursor_value"}
-    paginator.path.return_value = None
-    paginator.next_page_token.return_value = next_page_token
-    paginator.get_requesyyt_headers.return_value = {}
-
-    record_selector = MagicMock()
-    record_selector.select_records.return_value = records
-
-    cursor = MagicMock(spec=Cursor)
-    stream_slices = [{"date": "2022-01-01"}, {"date": "2022-01-02"}]
-    cursor.stream_slices.return_value = stream_slices
-
-    response = requests.Response()
-    response.status_code = 200
-
-    underlying_state = {"date": "2021-01-01"}
-    cursor.get_stream_state.return_value = underlying_state
-
-    requester.get_authenticator.return_value = NoAuth({})
-    url_base = "https://airbyte.io"
-    requester.get_url_base.return_value = url_base
-    path = "/v1"
-    requester.get_path.return_value = path
-    http_method = HttpMethod.GET
-    requester.get_method.return_value = http_method
-    backoff_time = 60
-    should_retry = ResponseStatus.retry(backoff_time)
-    requester.interpret_response_status.return_value = should_retry
-    request_body_json = {"body": "json"}
-    requester.request_body_json.return_value = request_body_json
-
-    request_body_data = {"body": "data"}
-    requester.get_request_body_data.return_value = request_body_data
-    request_body_json = {"body": "json"}
-    requester.get_request_body_json.return_value = request_body_json
-    request_kwargs = {"kwarg": "value"}
-    requester.request_kwargs.return_value = request_kwargs
-
-    retriever = SimpleRetriever(
-        name="stream_name",
-        primary_key=primary_key,
-        requester=requester,
-        paginator=paginator,
-        record_selector=record_selector,
-        stream_slicer=cursor,
-        cursor=cursor,
-        parameters={},
-        config={},
-    )
-
-    assert retriever.primary_key == primary_key
-    assert retriever.state == underlying_state
-    assert retriever._next_page_token(response) == next_page_token
-    assert retriever._request_params(None, None, None) == {}
-    assert retriever.stream_slices() == stream_slices
-
-    assert retriever._last_response is None
-    assert retriever._records_from_last_response == []
-    assert retriever._parse_response(response, stream_state={}, records_schema={}) == records
-    assert retriever._last_response == response
-    assert retriever._records_from_last_response == records
-
-    [r for r in retriever.read_records(SyncMode.full_refresh)]
-    paginator.reset.assert_called()
-
-
-@patch.object(SimpleRetriever, "_read_pages", return_value=iter([*request_response_logs, *records]))
-def test_simple_retriever_with_request_response_logs(mock_http_stream):
-    requester = MagicMock()
-    paginator = MagicMock()
-    record_selector = MagicMock()
-    stream_slicer = DatetimeBasedCursor(
-        start_datetime="",
-        end_datetime="",
-        step="P1D",
-        cursor_field="id",
-        datetime_format="",
-        cursor_granularity="P1D",
-        config={},
-        parameters={},
-    )
-
-    retriever = SimpleRetriever(
-        name="stream_name",
-        primary_key=primary_key,
-        requester=requester,
-        paginator=paginator,
-        record_selector=record_selector,
-        stream_slicer=stream_slicer,
-        parameters={},
-        config={},
-    )
-
-    actual_messages = [r for r in retriever.read_records(SyncMode.full_refresh)]
-    paginator.reset.assert_called()
-
-    assert isinstance(actual_messages[0], AirbyteLogMessage)
-    assert isinstance(actual_messages[1], AirbyteLogMessage)
-    assert actual_messages[2] == records[0]
-    assert actual_messages[3] == records[1]
-
-
-@patch.object(SimpleRetriever, "_read_pages", return_value=iter([]))
-def test_simple_retriever_with_request_response_log_last_records(mock_http_stream):
-    requester = MagicMock()
-    paginator = MagicMock()
-    record_selector = MagicMock()
-    record_selector.select_records.return_value = request_response_logs
-    response = requests.Response()
-    response.status_code = 200
-    stream_slicer = DatetimeBasedCursor(
-        start_datetime="",
-        end_datetime="",
-        step="P1D",
-        cursor_field="id",
-        datetime_format="",
-        cursor_granularity="P1D",
-        config={},
-        parameters={},
-    )
-
-    retriever = SimpleRetriever(
-        name="stream_name",
-        primary_key=primary_key,
-        requester=requester,
-        paginator=paginator,
-        record_selector=record_selector,
-        stream_slicer=stream_slicer,
-        parameters={},
-        config={},
-    )
-
-    assert retriever._last_response is None
-    assert retriever._records_from_last_response == []
-    assert retriever._parse_response(response, stream_state={}, records_schema={}) == request_response_logs
-    assert retriever._last_response == response
-    assert retriever._records_from_last_response == request_response_logs
-
-    [r for r in retriever.read_records(SyncMode.full_refresh)]
-    paginator.reset.assert_called()
-
-
-@pytest.mark.parametrize(
-    "test_name, paginator_mapping, stream_slicer_mapping, expected_mapping",
-    [
-        ("test_empty_headers", {}, {}, {}),
-        ("test_header_from_pagination_and_slicer", {"offset": 1000}, {"key": "value"}, {"key": "value", "offset": 1000}),
-        ("test_header_from_stream_slicer", {}, {"slice": "slice_value"}, {"slice": "slice_value"}),
-        ("test_duplicate_header_slicer_paginator", {"k": "v"}, {"k": "slice_value"}, None),
-    ],
-)
-def test_get_request_options_from_pagination(test_name, paginator_mapping, stream_slicer_mapping, expected_mapping):
-    # This test does not test request headers because they must be strings
-    paginator = MagicMock()
-    paginator.get_request_params.return_value = paginator_mapping
-    paginator.get_request_body_data.return_value = paginator_mapping
-    paginator.get_request_body_json.return_value = paginator_mapping
-
-    stream_slicer = MagicMock()
-    stream_slicer.get_request_params.return_value = stream_slicer_mapping
-    stream_slicer.get_request_body_data.return_value = stream_slicer_mapping
-    stream_slicer.get_request_body_json.return_value = stream_slicer_mapping
-
-    record_selector = MagicMock()
-    retriever = SimpleRetriever(
-        name="stream_name",
-        primary_key=primary_key,
-        requester=MagicMock(),
-        record_selector=record_selector,
-        paginator=paginator,
-        stream_slicer=stream_slicer,
-        parameters={},
-        config={},
-    )
-
-    request_option_type_to_method = {
-        RequestOptionType.request_parameter: retriever._request_params,
-        RequestOptionType.body_data: retriever._request_body_data,
-        RequestOptionType.body_json: retriever._request_body_json,
-    }
-
-    for _, method in request_option_type_to_method.items():
-        if expected_mapping is not None:
-            actual_mapping = method(None, None, None)
-            assert expected_mapping == actual_mapping
-        else:
-            try:
-                method(None, None, None)
-                assert False
-            except ValueError:
-                pass
-
-
-@pytest.mark.parametrize(
-    "test_name, paginator_mapping, expected_mapping",
-    [
-        ("test_only_base_headers", {}, {"key": "value"}),
-        ("test_header_from_pagination", {"offset": 1000}, {"key": "value", "offset": "1000"}),
-        ("test_duplicate_header", {"key": 1000}, None),
-    ],
-)
-def test_get_request_headers(test_name, paginator_mapping, expected_mapping):
-    # This test is separate from the other request options because request headers must be strings
-    paginator = MagicMock()
-    paginator.get_request_headers.return_value = paginator_mapping
-    requester = MagicMock(use_cache=False)
-
-    stream_slicer = MagicMock()
-    stream_slicer.get_request_headers.return_value = {"key": "value"}
-
-    record_selector = MagicMock()
-    retriever = SimpleRetriever(
-        name="stream_name",
-        primary_key=primary_key,
-        requester=requester,
-        record_selector=record_selector,
-        stream_slicer=stream_slicer,
-        paginator=paginator,
-        parameters={},
-        config={},
-    )
-
-    request_option_type_to_method = {
-        RequestOptionType.header: retriever._request_headers,
-    }
-
-    for _, method in request_option_type_to_method.items():
-        if expected_mapping:
-            actual_mapping = method(None, None, None)
-            assert expected_mapping == actual_mapping
-        else:
-            try:
-                method(None, None, None)
-                assert False
-            except ValueError:
-                pass
-
-
-@pytest.mark.parametrize(
-    "test_name, paginator_mapping, ignore_stream_slicer_parameters_on_paginated_requests, next_page_token, expected_mapping",
-    [
-        ("test_do_not_ignore_stream_slicer_params_if_ignore_is_true_but_no_next_page_token", {"key_from_pagination": "1000"}, True, None, {"key_from_pagination": "1000"}),
-        ("test_do_not_ignore_stream_slicer_params_if_ignore_is_false_and_no_next_page_token", {"key_from_pagination": "1000"}, False, None, {"key_from_pagination": "1000", "key_from_slicer": "value"}),
-        ("test_ignore_stream_slicer_params_on_paginated_request", {"key_from_pagination": "1000"}, True, {"page": 2}, {"key_from_pagination": "1000"}),
-        ("test_do_not_ignore_stream_slicer_params_on_paginated_request", {"key_from_pagination": "1000"}, False, {"page": 2}, {"key_from_pagination": "1000", "key_from_slicer": "value"}),
-    ],
-)
-def test_ignore_stream_slicer_parameters_on_paginated_requests(test_name, paginator_mapping, ignore_stream_slicer_parameters_on_paginated_requests, next_page_token, expected_mapping):
-    # This test is separate from the other request options because request headers must be strings
-    paginator = MagicMock()
-    paginator.get_request_headers.return_value = paginator_mapping
-    requester = MagicMock(use_cache=False)
-
-    stream_slicer = MagicMock()
-    stream_slicer.get_request_headers.return_value = {"key_from_slicer": "value"}
-
-    record_selector = MagicMock()
-    retriever = SimpleRetriever(
-        name="stream_name",
-        primary_key=primary_key,
-        requester=requester,
-        record_selector=record_selector,
-        stream_slicer=stream_slicer,
-        paginator=paginator,
-        ignore_stream_slicer_parameters_on_paginated_requests=ignore_stream_slicer_parameters_on_paginated_requests,
-        parameters={},
-        config={},
-    )
-
-    request_option_type_to_method = {
-        RequestOptionType.header: retriever._request_headers,
-    }
-
-    for _, method in request_option_type_to_method.items():
-        actual_mapping = method(None, None, next_page_token={"next_page_token": "1000"})
-        assert expected_mapping == actual_mapping
-
-
-@pytest.mark.parametrize(
-    "test_name, slicer_body_data, paginator_body_data, expected_body_data",
-    [
-        ("test_only_slicer_mapping", {"key": "value"}, {}, {"key": "value"}),
-        ("test_only_slicer_string", "key=value", {}, "key=value"),
-        ("test_slicer_mapping_and_paginator_no_duplicate", {"key": "value"}, {"offset": 1000}, {"key": "value", "offset": 1000}),
-        ("test_slicer_mapping_and_paginator_with_duplicate", {"key": "value"}, {"key": 1000}, None),
-        ("test_slicer_string_and_paginator", "key=value", {"offset": 1000}, None),
-    ],
-)
-def test_request_body_data(test_name, slicer_body_data, paginator_body_data, expected_body_data):
-    paginator = MagicMock()
-    paginator.get_request_body_data.return_value = paginator_body_data
-    requester = MagicMock(use_cache=False)
-
-    stream_slicer = MagicMock()
-    stream_slicer.get_request_body_data.return_value = slicer_body_data
-
-    record_selector = MagicMock()
-    retriever = SimpleRetriever(
-        name="stream_name",
-        primary_key=primary_key,
-        requester=requester,
-        record_selector=record_selector,
-        paginator=paginator,
-        stream_slicer=stream_slicer,
-        parameters={},
-        config={},
-    )
-
-    if expected_body_data:
-        actual_body_data = retriever._request_body_data(None, None, None)
-        assert expected_body_data == actual_body_data
-    else:
-        try:
-            retriever._request_body_data(None, None, None)
-            assert False
-        except ValueError:
-            pass
-
-
-@pytest.mark.parametrize(
-    "test_name, requester_path, paginator_path, expected_path",
-    [
-        ("test_path_from_requester", "/v1/path", None, None),
-        ("test_path_from_paginator", "/v1/path/", "/v2/paginator", "/v2/paginator"),
-    ],
-)
-def test_path(test_name, requester_path, paginator_path, expected_path):
-    paginator = MagicMock()
-    paginator.path.return_value = paginator_path
-    requester = MagicMock(use_cache=False)
-
-    requester.get_path.return_value = requester_path
-
-    record_selector = MagicMock()
-    retriever = SimpleRetriever(
-        name="stream_name",
-        primary_key=primary_key,
-        requester=requester,
-        record_selector=record_selector,
-        paginator=paginator,
-        parameters={},
-        config={},
-    )
-
-    actual_path = retriever._paginator_path()
-    assert expected_path == actual_path
-
-
-def test_limit_stream_slices():
-    maximum_number_of_slices = 4
-    stream_slicer = MagicMock()
-    stream_slicer.stream_slices.return_value = _generate_slices(maximum_number_of_slices * 2)
-    retriever = SimpleRetrieverTestReadDecorator(
-        name="stream_name",
-        primary_key=primary_key,
-        requester=MagicMock(),
-        paginator=MagicMock(),
-        record_selector=MagicMock(),
-        stream_slicer=stream_slicer,
-        maximum_number_of_slices=maximum_number_of_slices,
-        parameters={},
-        config={},
-    )
-
-    truncated_slices = list(retriever.stream_slices())
-
-    assert truncated_slices == _generate_slices(maximum_number_of_slices)
-
-
-@pytest.mark.parametrize(
-    "test_name, first_greater_than_second",
-    [
-        ("test_first_greater_than_second", True),
-        ("test_second_greater_than_first", False),
-    ],
-)
-def test_when_read_records_then_cursor_close_slice_with_greater_record(test_name, first_greater_than_second):
-    first_record = Record({"first": 1}, {})
-    second_record = Record({"second": 2}, {})
-    records = [first_record, second_record]
-    record_selector = MagicMock()
-    record_selector.select_records.return_value = records
-    cursor = MagicMock(spec=Cursor)
-    cursor.is_greater_than_or_equal.return_value = first_greater_than_second
-    paginator = MagicMock()
-    paginator.get_request_headers.return_value = {}
-
-    retriever = SimpleRetriever(
-        name="stream_name",
-        primary_key=primary_key,
-        requester=MagicMock(),
-        paginator=paginator,
-        record_selector=record_selector,
-        stream_slicer=cursor,
-        cursor=cursor,
-        parameters={},
-        config={},
-    )
-    stream_slice = {"repository": "airbyte"}
-
-    def retriever_read_pages(_, __, ___):
-        return retriever._parse_records(response=MagicMock(), stream_state={}, stream_slice=stream_slice, records_schema={})
-
-    with patch.object(
-        SimpleRetriever,
-        "_read_pages",
-        return_value=iter([first_record, second_record]),
-        side_effect=retriever_read_pages,
-    ):
-        list(retriever.read_records(stream_slice=stream_slice, records_schema={}))
-        cursor.close_slice.assert_called_once_with(stream_slice, first_record if first_greater_than_second else second_record)
-
-
-def test_given_stream_data_is_not_record_when_read_records_then_update_slice_with_optional_record():
-    stream_data = [AirbyteMessage(type=Type.LOG, log=AirbyteLogMessage(level=Level.INFO, message="a log message"))]
-    record_selector = MagicMock()
-    record_selector.select_records.return_value = []
-    cursor = MagicMock(spec=Cursor)
-
-    retriever = SimpleRetriever(
-        name="stream_name",
-        primary_key=primary_key,
-        requester=MagicMock(),
-        paginator=Mock(),
-        record_selector=record_selector,
-        stream_slicer=cursor,
-        cursor=cursor,
-        parameters={},
-        config={},
-    )
-    stream_slice = {"repository": "airbyte"}
-
-    def retriever_read_pages(_, __, ___):
-        return retriever._parse_records(response=MagicMock(), stream_state={}, stream_slice=stream_slice, records_schema={})
-
-    with patch.object(
-        SimpleRetriever,
-        "_read_pages",
-        return_value=iter(stream_data),
-        side_effect=retriever_read_pages,
-    ):
-        list(retriever.read_records(stream_slice=stream_slice, records_schema={}))
-        cursor.observe.assert_not_called()
-        cursor.close_slice.assert_called_once_with(stream_slice, None)
-
-
-def _generate_slices(number_of_slices):
-    return [{"date": f"2022-01-0{day + 1}"} for day in range(number_of_slices)]
-
-
-@patch.object(SimpleRetriever, "_read_pages", return_value=iter([]))
-def test_given_state_selector_when_read_records_use_stream_state(http_stream_read_pages, mocker):
-    requester = MagicMock()
-    paginator = MagicMock()
-    record_selector = MagicMock()
-    cursor = MagicMock(spec=Cursor)
-    cursor.select_state = MagicMock(return_value=A_SLICE_STATE)
-    cursor.get_stream_state = MagicMock(return_value=A_STREAM_STATE)
-
-    retriever = SimpleRetriever(
-        name="stream_name",
-        primary_key=primary_key,
-        requester=requester,
-        paginator=paginator,
-        record_selector=record_selector,
-        stream_slicer=cursor,
-        cursor=cursor,
-        parameters={},
-        config={},
-    )
-
-    list(retriever.read_records(stream_slice=A_STREAM_SLICE, records_schema={}))
-
-    http_stream_read_pages.assert_called_once_with(mocker.ANY, A_STREAM_STATE, A_STREAM_SLICE)
-
-
-def test_emit_log_request_response_messages(mocker):
-    record_selector = MagicMock()
-    record_selector.select_records.return_value = records
-
-    request = requests.PreparedRequest()
-    request.headers = {"header": "value"}
-    request.url = "http://byrde.enterprises.com/casinos"
-
-    response = requests.Response()
-    response.request = request
-    response.status_code = 200
-
-    format_http_message_mock = mocker.patch("airbyte_cdk.sources.declarative.retrievers.simple_retriever.format_http_message")
-    requester = MagicMock()
-    retriever = SimpleRetrieverTestReadDecorator(
-        name="stream_name",
-        primary_key=primary_key,
-        requester=requester,
-        paginator=MagicMock(),
-        record_selector=record_selector,
-        stream_slicer=SinglePartitionRouter(parameters={}),
-        parameters={},
-        config={},
-    )
-
-    retriever._fetch_next_page(stream_state={}, stream_slice={})
-
-    assert requester.send_request.call_args_list[0][1]["log_formatter"] is not None
-    assert requester.send_request.call_args_list[0][1]["log_formatter"](response) == format_http_message_mock.return_value

unit_tests/sources/declarative/schema/__init__.py

@@ -1,6 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-from .source_test.SourceTest import SourceTest
-
-__all__ = ["SourceTest"]

unit_tests/sources/declarative/schema/source_test/SourceTest.py

@@ -1,8 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-
-
-class SourceTest:
-    def __init__(self):
-        pass

unit_tests/sources/declarative/schema/source_test/__init__.py

@@ -1,3 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#

unit_tests/sources/declarative/schema/test_default_schema_loader.py

@@ -1,32 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-
-from unittest.mock import MagicMock
-
-import pytest
-from airbyte_cdk.sources.declarative.schema import DefaultSchemaLoader
-
-
-@pytest.mark.parametrize(
-    "found_schema, found_error, expected_schema",
-    [
-        pytest.param(
-            {"type": "object", "properties": {}}, None, {"type": "object", "properties": {}}, id="test_has_schema_in_default_location"
-        ),
-        pytest.param(None, FileNotFoundError, {}, id="test_schema_file_does_not_exist"),
-    ],
-)
-def test_get_json_schema(found_schema, found_error, expected_schema):
-    default_schema_loader = DefaultSchemaLoader({}, {})
-
-    json_file_schema_loader = MagicMock()
-    if found_schema:
-        json_file_schema_loader.get_json_schema.return_value = {"type": "object", "properties": {}}
-    if found_error:
-        json_file_schema_loader.get_json_schema.side_effect = found_error
-
-    default_schema_loader.default_loader = json_file_schema_loader
-
-    actual_schema = default_schema_loader.get_json_schema()
-    assert actual_schema == expected_schema

unit_tests/sources/declarative/schema/test_inline_schema_loader.py

@@ -1,19 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-
-import pytest
-from airbyte_cdk.sources.declarative.schema import InlineSchemaLoader
-
-
-@pytest.mark.parametrize(
-    "test_name, input_schema, expected_schema",
-    [
-        ("schema", {"k": "string"}, {"k": "string"}),
-        ("empty_schema", {}, {}),
-    ],
-)
-def test_static_schema_loads(test_name, input_schema, expected_schema):
-    schema_loader = InlineSchemaLoader(input_schema, {})
-
-    assert schema_loader.get_json_schema() == expected_schema

unit_tests/sources/declarative/schema/test_json_file_schema_loader.py

@@ -1,26 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-
-import pytest
-from airbyte_cdk.sources.declarative.schema import JsonFileSchemaLoader
-
-
-@pytest.mark.parametrize(
-    "test_name, input_path, expected_resource, expected_path",
-    [
-        ("path_prefixed_with_dot", "./source_example/schemas/lists.json", "source_example", "schemas/lists.json"),
-        ("path_prefixed_with_slash", "/source_example/schemas/lists.json", "source_example", "schemas/lists.json"),
-        ("path_starting_with_source", "source_example/schemas/lists.json", "source_example", "schemas/lists.json"),
-        ("path_starting_missing_source", "schemas/lists.json", "schemas", "lists.json"),
-        ("path_with_file_only", "lists.json", "", "lists.json"),
-        ("empty_path_does_not_crash", "", "", ""),
-        ("empty_path_with_slash_does_not_crash", "/", "", ""),
-    ],
-)
-def test_extract_resource_and_schema_path(test_name, input_path, expected_resource, expected_path):
-    json_schema = JsonFileSchemaLoader({}, {}, input_path)
-    actual_resource, actual_path = json_schema.extract_resource_and_schema_path(input_path)
-
-    assert actual_resource == expected_resource
-    assert actual_path == expected_path

unit_tests/sources/declarative/states/__init__.py

@@ -1,3 +0,0 @@
-#
-# Copyright (c) 2021 Airbyte, Inc., all rights reserved.
-#