airbyte-cdk 6.45.0.post20.dev14369762306__py3-none-any.whl → 6.45.0.post23.dev14384860031__py3-none-any.whl

airbyte_cdk/test/declarative/models/scenario.py

@@ -10,7 +10,7 @@ up iteration cycles.
 from __future__ import annotations
 
 from pathlib import Path
-from typing import Literal
+from typing import Any, Literal, cast
 
 import yaml
 from pydantic import BaseModel
@@ -33,7 +33,7 @@ class ConnectorTestScenario(BaseModel):
         bypass_reason: str
 
     config_path: Path | None = None
-    config_dict: dict | None = None
+    config_dict: dict[str, Any] | None = None
 
     id: str | None = None
 
@@ -43,27 +43,27 @@ class ConnectorTestScenario(BaseModel):
     file_types: AcceptanceTestFileTypes | None = None
     status: Literal["succeed", "failed"] | None = None
 
-    def get_config_dict(self) -> dict:
+    def get_config_dict(self) -> dict[str, Any]:
         """Return the config dictionary.
 
         If a config dictionary has already been loaded, return it. Otherwise, load
-        Otherwise, load the config file and return the dictionary.
+        the config file and return the dictionary.
         """
         if self.config_dict:
             return self.config_dict
 
         if self.config_path:
-            return yaml.safe_load(self.config_path.read_text())
+            return cast(dict[str, Any], yaml.safe_load(self.config_path.read_text()))
 
         raise ValueError("No config dictionary or path provided.")
 
     @property
     def expect_exception(self) -> bool:
-        return self.status and self.status == "failed"
+        return self.status and self.status == "failed" or False
 
     @property
     def instance_name(self) -> str:
-        return self.config_path.stem
+        return self.config_path.stem if self.config_path else "Unnamed Scenario"
 
     def __str__(self) -> str:
         if self.id:

airbyte_cdk/test/declarative/test_suites/connector_base.py

@@ -20,6 +20,11 @@ from airbyte_cdk.models import (
     AirbyteMessage,
     Type,
 )
+from airbyte_cdk.sources.declarative.declarative_source import (
+    AbstractSource,
+    ConcurrentDeclarativeSource,
+    Source,
+)
 from airbyte_cdk.test import entrypoint_wrapper
 from airbyte_cdk.test.declarative.models import (
     ConnectorTestScenario,
@@ -41,10 +46,10 @@ class RunnableConnector(abc.ABC):
     """A connector that can be run in a test scenario."""
 
     @abc.abstractmethod
-    def launch(cls, args: list[str] | None): ...
+    def launch(cls, args: list[str] | None) -> None: ...
 
 
-def generate_tests(metafunc):
+def generate_tests(metafunc) -> None:
     """
     A helper for pytest_generate_tests hook.
 

airbyte_cdk/test/declarative/test_suites/declarative_sources.py

@@ -26,6 +26,7 @@ class DeclarativeSourceTestSuite(SourceTestSuiteBase):
     def create_connector(
         self, connector_test: ConnectorTestScenario
     ) -> ConcurrentDeclarativeSource:
+        """Create a connector instance for the test suite."""
         config = connector_test.get_config_dict()
         # catalog = connector_test.get_catalog()
         # state = connector_test.get_state()

airbyte_cdk/test/declarative/utils/job_runner.py

@@ -1,7 +1,7 @@
 import tempfile
 import uuid
 from pathlib import Path
-from typing import Callable, Literal
+from typing import Any, Callable, Literal
 
 import orjson
 
@@ -22,7 +22,7 @@ def run_test_job(
     verb: Literal["read", "check", "discover"],
     test_instance: ConnectorTestScenario,
     *,
-    catalog: dict | None = None,
+    catalog: dict[str, Any] | None = None,
 ) -> entrypoint_wrapper.EntrypointOutput:
     """Run a test job from provided CLI args and return the result."""
     if not connector:
@@ -49,7 +49,7 @@ def run_test_job(
     else:
         raise ValueError(f"Invalid source type: {type(connector)}")
 
-    args = [verb]
+    args: list[str] = [verb]
     if test_instance.config_path:
         args += ["--config", str(test_instance.config_path)]
     elif test_instance.config_dict:
@@ -103,12 +103,12 @@ def run_test_job(
         assert len(result.connection_status_messages) == 1, (
             "Expected exactly one CONNECTION_STATUS message. Got "
             f"{len(result.connection_status_messages)}:\n"
-            + "\n".join(result.connection_status_messages)
+            + "\n".join([str(msg) for msg in result.connection_status_messages])
         )
         if test_instance.expect_exception:
             assert result.connection_status_messages[0].connectionStatus.status == Status.FAILED, (
                 "Expected CONNECTION_STATUS message to be FAILED. Got: \n"
-                + "\n".join([str(result.connection_status_messages)])
+                + "\n".join([str(msg) for msg in result.connection_status_messages])
             )
         return result
 

airbyte_cdk/test/pytest_config/plugin.py

@@ -1,18 +1,24 @@
+"""Global pytest configuration for the Airbyte CDK tests."""
+
 from pathlib import Path
+from typing import Optional
 
 import pytest
 
 
-def pytest_collect_file(parent, path):
-    if path.basename == "test_connector.py":
-        return pytest.Module.from_parent(parent, path=path)
+def pytest_collect_file(parent: Optional[pytest.Module], path: Path) -> pytest.Module | None:
+    """Collect test files based on their names."""
+    if path.name == "test_connector.py":
+        return cast(pytest.Module, pytest.Module.from_parent(parent, path=path))
+
+    return None
 
 
-def pytest_configure(config):
+def pytest_configure(config: pytest.Config) -> None:
     config.addinivalue_line("markers", "connector: mark test as a connector test")
 
 
-def pytest_addoption(parser):
+def pytest_addoption(parser: pytest.Parser) -> None:
     parser.addoption(
         "--run-connector",
         action="store_true",
@@ -21,7 +27,7 @@ def pytest_addoption(parser):
     )
 
 
-def pytest_collection_modifyitems(config, items):
+def pytest_collection_modifyitems(config: pytest.Config, items: list[pytest.Item]) -> None:
     if config.getoption("--run-connector"):
         return
     skip_connector = pytest.mark.skip(reason="need --run-connector option to run")
@@ -30,11 +36,11 @@ def pytest_collection_modifyitems(config, items):
             item.add_marker(skip_connector)
 
 
-def pytest_runtest_setup(item):
+def pytest_runtest_setup(item: pytest.Item) -> None:
     # This hook is called before each test function is executed
     print(f"Setting up test: {item.name}")
 
 
-def pytest_runtest_teardown(item, nextitem):
+def pytest_runtest_teardown(item: pytest.Item, nextitem: Optional[pytest.Item]) -> None:
     # This hook is called after each test function is executed
     print(f"Tearing down test: {item.name}")

{airbyte_cdk-6.45.0.post20.dev14369762306.dist-info → airbyte_cdk-6.45.0.post23.dev14384860031.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: airbyte-cdk
-Version: 6.45.0.post20.dev14369762306
+Version: 6.45.0.post23.dev14384860031
 Summary: A framework for writing Airbyte Connectors.
 Home-page: https://airbyte.com
 License: MIT

{airbyte_cdk-6.45.0.post20.dev14369762306.dist-info → airbyte_cdk-6.45.0.post23.dev14384860031.dist-info}/RECORD

@@ -338,25 +338,24 @@ airbyte_cdk/test/__init__.py,sha256=f_XdkOg4_63QT2k3BbKY34209lppwgw-svzfZstQEq4,
 airbyte_cdk/test/catalog_builder.py,sha256=-y05Cz1x0Dlk6oE9LSKhCozssV2gYBNtMdV5YYOPOtk,3015
 airbyte_cdk/test/declarative/__init__.py,sha256=IpPLFCCWnlpp_eTGKK_2WGRFfenbnolxYLmkQSZMO3U,205
 airbyte_cdk/test/declarative/models/__init__.py,sha256=rXoywbDd-oqEhRQLuaEQIDxykMeawHjzzeN3XIpGQN8,132
-airbyte_cdk/test/declarative/models/scenario.py,sha256=0yaThwlYrOAKRwgcHviGaiFjOlZTadxZX2BWiwrDX5w,2288
+airbyte_cdk/test/declarative/models/scenario.py,sha256=ySzQosxi2TRaw7JbQMFsuBieoH0ufXJhEY8Vu3DXtx4,2378
 airbyte_cdk/test/declarative/test_suites/__init__.py,sha256=E1KY-xxKjkKXYaAVyCs96-ZMCCoDhR24s2eQhWWOx-c,769
-airbyte_cdk/test/declarative/test_suites/connector_base.py,sha256=ei3pEj_sFCNg9QeCURsCiEGyPQcaia74G6oskIkJ-Dk,6612
-airbyte_cdk/test/declarative/test_suites/declarative_sources.py,sha256=QE0YUUiS5ANScb2QS-yR04KubwlTdZI4Hg6CqKA057c,1571
+airbyte_cdk/test/declarative/test_suites/connector_base.py,sha256=7qFjRI6NkJPOTIHndl8KvWz4eM2PiXMPBSKOBpGKk7o,6760
+airbyte_cdk/test/declarative/test_suites/declarative_sources.py,sha256=4VPGzG8keyci0x2o6_jNqx7HcAuamumJoPafudH5w_o,1633
 airbyte_cdk/test/declarative/test_suites/destination_base.py,sha256=L_l1gpNCkzfx-c7mzS1He5hTVbqR39OnfWdrYMglS7E,486
 airbyte_cdk/test/declarative/test_suites/source_base.py,sha256=1VJ9KXM1nSodFKh2VKXc2cO010JfE-df2JoVGKasrmk,4381
 airbyte_cdk/test/declarative/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-airbyte_cdk/test/declarative/utils/job_runner.py,sha256=yraJPrrlX3HK7aanBodu4IJReAUI5NzI6n3mi2Oz_1s,4782
+airbyte_cdk/test/declarative/utils/job_runner.py,sha256=AI10Nt5MEMbJt_zNNBatUPStvMEF1mstHzow-11ZVsE,4845
 airbyte_cdk/test/entrypoint_wrapper.py,sha256=TyUmVJyIuGelAv6y8Wy_BnwqIRw_drjfZWKlroljCuQ,9951
 airbyte_cdk/test/fixtures/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 airbyte_cdk/test/fixtures/auto.py,sha256=wYKu1pBUJX60Bg-K2RwNXi11Txu1Rncc8WNlOvy9Zug,354
-airbyte_cdk/test/fixtures/general.py,sha256=pvAqTR1F5CZX4sIn8J7ON5T5Vy5f7SqK6XkHpMbt8c8,255
 airbyte_cdk/test/mock_http/__init__.py,sha256=jE5kC6CQ0OXkTqKhciDnNVZHesBFVIA2YvkdFGwva7k,322
 airbyte_cdk/test/mock_http/matcher.py,sha256=4Qj8UnJKZIs-eodshryce3SN1Ayc8GZpBETmP6hTEyc,1446
 airbyte_cdk/test/mock_http/mocker.py,sha256=XgsjMtVoeMpRELPyALgrkHFauH9H5irxrz1Kcxh2yFY,8013
 airbyte_cdk/test/mock_http/request.py,sha256=tdB8cqk2vLgCDTOKffBKsM06llYs4ZecgtH6DKyx6yY,4112
 airbyte_cdk/test/mock_http/response.py,sha256=s4-cQQqTtmeej0pQDWqmG0vUWpHS-93lIWMpW3zSVyU,662
 airbyte_cdk/test/mock_http/response_builder.py,sha256=debPx_lRYBaQVSwCoKLa0F8KFk3h0qG7bWxFBATa0cc,7958
-airbyte_cdk/test/pytest_config/plugin.py,sha256=SWqyC4FQAlWgNJ0zgv9usmVvhwRXucbkmy0hLYBXTtk,1074
+airbyte_cdk/test/pytest_config/plugin.py,sha256=Z2RYpXOKTRsytb0pn0K-PvkQlQXeOppnEcX4IlBEZfI,1457
 airbyte_cdk/test/state_builder.py,sha256=kLPql9lNzUJaBg5YYRLJlY_Hy5JLHJDVyKPMZMoYM44,946
 airbyte_cdk/test/utils/__init__.py,sha256=Hu-1XT2KDoYjDF7-_ziDwv5bY3PueGjANOCbzeOegDg,57
 airbyte_cdk/test/utils/data.py,sha256=CkCR1_-rujWNmPXFR1IXTMwx1rAl06wAyIKWpDcN02w,820
@@ -380,9 +379,9 @@ airbyte_cdk/utils/slice_hasher.py,sha256=EDxgROHDbfG-QKQb59m7h_7crN1tRiawdf5uU7G
 airbyte_cdk/utils/spec_schema_transformations.py,sha256=-5HTuNsnDBAhj-oLeQXwpTGA0HdcjFOf2zTEMUTTg_Y,816
 airbyte_cdk/utils/stream_status_utils.py,sha256=ZmBoiy5HVbUEHAMrUONxZvxnvfV9CesmQJLDTAIWnWw,1171
 airbyte_cdk/utils/traced_exception.py,sha256=C8uIBuCL_E4WnBAOPSxBicD06JAldoN9fGsQDp463OY,6292
-airbyte_cdk-6.45.0.post20.dev14369762306.dist-info/LICENSE.txt,sha256=Wfe61S4BaGPj404v8lrAbvhjYR68SHlkzeYrg3_bbuM,1051
-airbyte_cdk-6.45.0.post20.dev14369762306.dist-info/LICENSE_SHORT,sha256=aqF6D1NcESmpn-cqsxBtszTEnHKnlsp8L4x9wAh3Nxg,55
-airbyte_cdk-6.45.0.post20.dev14369762306.dist-info/METADATA,sha256=zTOq3asxJHi47HUL-AdFI0s8EPUb0D_xQ6wnmbmqKLE,6093
-airbyte_cdk-6.45.0.post20.dev14369762306.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-airbyte_cdk-6.45.0.post20.dev14369762306.dist-info/entry_points.txt,sha256=fj-e3PAQvsxsQzyyq8UkG1k8spunWnD4BAH2AwlR6NM,95
-airbyte_cdk-6.45.0.post20.dev14369762306.dist-info/RECORD,,
+airbyte_cdk-6.45.0.post23.dev14384860031.dist-info/LICENSE.txt,sha256=Wfe61S4BaGPj404v8lrAbvhjYR68SHlkzeYrg3_bbuM,1051
+airbyte_cdk-6.45.0.post23.dev14384860031.dist-info/LICENSE_SHORT,sha256=aqF6D1NcESmpn-cqsxBtszTEnHKnlsp8L4x9wAh3Nxg,55
+airbyte_cdk-6.45.0.post23.dev14384860031.dist-info/METADATA,sha256=9vnnu8hLTvth6IXJb8KCu045Tl6ulUD0afpW1H0qziU,6093
+airbyte_cdk-6.45.0.post23.dev14384860031.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+airbyte_cdk-6.45.0.post23.dev14384860031.dist-info/entry_points.txt,sha256=fj-e3PAQvsxsQzyyq8UkG1k8spunWnD4BAH2AwlR6NM,95
+airbyte_cdk-6.45.0.post23.dev14384860031.dist-info/RECORD,,

airbyte_cdk/test/fixtures/general.py

@@ -1,15 +0,0 @@
-"""General fixtures for pytest.
-
-Usage:
-
-```python
-from airbyte_cdk.test.fixtures.general import *
-# OR:
-from airbyte_cdk.test.fixtures.general import connector_test_dir
-```
-"""
-
-
-@pytest.fixture
-def connector_test_dir():
-    return Path(__file__).parent