Skip to content

Commit ca07d1a

Browse files
committed
enhance unit tests
1 parent 335981e commit ca07d1a

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

unit_tests/connector_builder/test_connector_builder_handler.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@
5959
from airbyte_cdk.sources.declarative.declarative_stream import DeclarativeStream
6060
from airbyte_cdk.sources.declarative.manifest_declarative_source import ManifestDeclarativeSource
6161
from airbyte_cdk.sources.declarative.retrievers.simple_retriever import SimpleRetriever
62+
from airbyte_cdk.sources.declarative.stream_slicers import StreamSlicerTestReadDecorator
6263
from airbyte_cdk.test.mock_http import HttpMocker, HttpRequest, HttpResponse
6364
from airbyte_cdk.utils.airbyte_secrets_utils import filter_secrets, update_secrets
6465
from unit_tests.connector_builder.utils import create_configured_catalog
@@ -1112,6 +1113,7 @@ def test_read_source(mock_http_stream):
11121113
streams = source.streams(config)
11131114
for s in streams:
11141115
assert isinstance(s.retriever, SimpleRetriever)
1116+
assert isinstance(s.retriever.stream_slicer, StreamSlicerTestReadDecorator)
11151117

11161118

11171119
@patch.object(
@@ -1158,6 +1160,7 @@ def test_read_source_single_page_single_slice(mock_http_stream):
11581160
streams = source.streams(config)
11591161
for s in streams:
11601162
assert isinstance(s.retriever, SimpleRetriever)
1163+
assert isinstance(s.retriever.stream_slicer, StreamSlicerTestReadDecorator)
11611164

11621165

11631166
@pytest.mark.parametrize(

unit_tests/sources/declarative/parsers/test_model_to_component_factory.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2737,6 +2737,7 @@ def test_simple_retriever_emit_log_messages():
27372737
assert retriever.log_formatter(response) == connector_builder_factory._get_log_formatter(
27382738
None, retriever.name
27392739
)(response)
2740+
assert isinstance(retriever.stream_slicer, StreamSlicerTestReadDecorator)
27402741

27412742

27422743
def test_create_page_increment():

0 commit comments

Comments
 (0)