airbyte-cdk 6.45.0.post23.dev14384860031__py3-none-any.whl → 6.45.1__py3-none-any.whl

airbyte_cdk/sources/declarative/interpolation/filters.py

@@ -4,9 +4,10 @@
 
 import base64
 import hashlib
+import hmac as hmac_lib
 import json
 import re
-from typing import Any, Optional
+from typing import Any, Dict, Optional
 
 
 def hash(value: Any, hash_type: str = "md5", salt: Optional[str] = None) -> str:
@@ -135,5 +136,51 @@ def regex_search(value: str, regex: str) -> str:
     return ""
 
 
-_filters_list = [hash, base64encode, base64decode, base64binascii_decode, string, regex_search]
+def hmac(value: Any, key: str, hash_type: str = "sha256") -> str:
+    """
+    Implementation of a custom Jinja2 hmac filter with SHA-256 support.
+
+    This filter creates a Hash-based Message Authentication Code (HMAC) using a cryptographic
+    hash function and a secret key. Currently only supports SHA-256, and returns hexdigest of the signature.
+
+    Example usage in a low code connector:
+
+    auth_headers:
+      $ref: "#/definitions/base_auth"
+      $parameters:
+        signature: "{{ 'message_to_sign' | hmac('my_secret_key') }}"
+
+    :param value: The message to be authenticated
+    :param key: The secret key for the HMAC
+    :param hash_type: Hash algorithm to use (default: sha256)
+    :return: HMAC digest as a hexadecimal string
+    """
+    # Define allowed hash functions
+    ALLOWED_HASH_TYPES: Dict[str, Any] = {
+        "sha256": hashlib.sha256,
+    }
+
+    if hash_type not in ALLOWED_HASH_TYPES:
+        raise ValueError(
+            f"Hash type '{hash_type}' is not allowed. Allowed types: {', '.join(ALLOWED_HASH_TYPES.keys())}"
+        )
+
+    hmac_obj = hmac_lib.new(
+        key=str(key).encode("utf-8"),
+        msg=str(value).encode("utf-8"),
+        digestmod=ALLOWED_HASH_TYPES[hash_type],
+    )
+
+    return hmac_obj.hexdigest()
+
+
+_filters_list = [
+    hash,
+    base64encode,
+    base64decode,
+    base64binascii_decode,
+    string,
+    regex_search,
+    hmac,
+]
 filters = {f.__name__: f for f in _filters_list}

airbyte_cdk/test/entrypoint_wrapper.py

@@ -82,10 +82,6 @@ class EntrypointOutput:
     def state_messages(self) -> List[AirbyteMessage]:
         return self._get_message_by_types([Type.STATE])
 
-    @property
-    def connection_status_messages(self) -> List[AirbyteMessage]:
-        return self._get_message_by_types([Type.CONNECTION_STATUS])
-
     @property
     def most_recent_state(self) -> Any:
         state_messages = self._get_message_by_types([Type.STATE])

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

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

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

@@ -101,7 +101,7 @@ airbyte_cdk/sources/declarative/incremental/per_partition_cursor.py,sha256=9IAJT
 airbyte_cdk/sources/declarative/incremental/per_partition_with_global.py,sha256=2YBOA2NnwAeIKlIhSwUB_W-FaGnPcmrG_liY7b4mV2Y,8365
 airbyte_cdk/sources/declarative/incremental/resumable_full_refresh_cursor.py,sha256=10LFv1QPM-agVKl6eaANmEBOfd7gZgBrkoTcMggsieQ,4809
 airbyte_cdk/sources/declarative/interpolation/__init__.py,sha256=Kh7FxhfetyNVDnAQ9zSxNe4oUbb8CvoW7Mqz7cs2iPg,437
-airbyte_cdk/sources/declarative/interpolation/filters.py,sha256=9tqTLFlmIp47QtN6zTToLh-YmO9b4Mbylc3bivEa448,4210
+airbyte_cdk/sources/declarative/interpolation/filters.py,sha256=cYap5zzOxIJWCLIfbkNlpyfUhjZ8FklLroIG4WGzYVs,5537
 airbyte_cdk/sources/declarative/interpolation/interpolated_boolean.py,sha256=8F3ntT_Mfo8cO9n6dCq8rTfJIpfKmzRCsVtVdhzaoGc,1964
 airbyte_cdk/sources/declarative/interpolation/interpolated_mapping.py,sha256=h36RIng4GZ9v4o_fRmgJjTNOtWmhK7NOILU1oSKPE4Q,2083
 airbyte_cdk/sources/declarative/interpolation/interpolated_nested_mapping.py,sha256=vjwvkLk7_l6YDcFClwjCMcTleRjQBh7-dzny7PUaoG8,1857
@@ -336,26 +336,13 @@ airbyte_cdk/sql/shared/sql_processor.py,sha256=fPWLcDaCBXQRV1ugdWIZ6x53FYGQ1BpMF
 airbyte_cdk/sql/types.py,sha256=XEIhRAo_ASd0kVLBkdLf5bHiRhNple-IJrC9TibcDdY,5880
 airbyte_cdk/test/__init__.py,sha256=f_XdkOg4_63QT2k3BbKY34209lppwgw-svzfZstQEq4,199
 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=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=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=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/entrypoint_wrapper.py,sha256=9XBii_YguQp0d8cykn3hy102FsJcwIBQzSB7co5ho0s,9802
 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=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
@@ -379,9 +366,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.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-6.45.1.dist-info/LICENSE.txt,sha256=Wfe61S4BaGPj404v8lrAbvhjYR68SHlkzeYrg3_bbuM,1051
+airbyte_cdk-6.45.1.dist-info/LICENSE_SHORT,sha256=aqF6D1NcESmpn-cqsxBtszTEnHKnlsp8L4x9wAh3Nxg,55
+airbyte_cdk-6.45.1.dist-info/METADATA,sha256=l16uftPwU9g9GDgSMmhyD0UWLcVSgNiN5SlMpuUJwMs,6071
+airbyte_cdk-6.45.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+airbyte_cdk-6.45.1.dist-info/entry_points.txt,sha256=fj-e3PAQvsxsQzyyq8UkG1k8spunWnD4BAH2AwlR6NM,95
+airbyte_cdk-6.45.1.dist-info/RECORD,,

airbyte_cdk/test/declarative/__init__.py

@@ -1,6 +0,0 @@
-# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
-"""Declarative tests framework.
-
-This module provides fixtures and utilities for testing Airbyte sources and destinations
-in a declarative way.
-"""

airbyte_cdk/test/declarative/models/__init__.py

@@ -1,7 +0,0 @@
-from airbyte_cdk.test.declarative.models.scenario import (
-    ConnectorTestScenario,
-)
-
-__all__ = [
-    "ConnectorTestScenario",
-]

airbyte_cdk/test/declarative/models/scenario.py

@@ -1,74 +0,0 @@
-# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
-"""Run acceptance tests in PyTest.
-
-These tests leverage the same `acceptance-test-config.yml` configuration files as the
-acceptance tests in CAT, but they run in PyTest instead of CAT. This allows us to run
-the acceptance tests in the same local environment as we are developing in, speeding
-up iteration cycles.
-"""
-
-from __future__ import annotations
-
-from pathlib import Path
-from typing import Any, Literal, cast
-
-import yaml
-from pydantic import BaseModel
-
-
-class ConnectorTestScenario(BaseModel):
-    """Acceptance test instance, as a Pydantic model.
-
-    This class represents an acceptance test instance, which is a single test case
-    that can be run against a connector. It is used to deserialize and validate the
-    acceptance test configuration file.
-    """
-
-    class AcceptanceTestExpectRecords(BaseModel):
-        path: Path
-        exact_order: bool = False
-
-    class AcceptanceTestFileTypes(BaseModel):
-        skip_test: bool
-        bypass_reason: str
-
-    config_path: Path | None = None
-    config_dict: dict[str, Any] | None = None
-
-    id: str | None = None
-
-    configured_catalog_path: Path | None = None
-    timeout_seconds: int | None = None
-    expect_records: AcceptanceTestExpectRecords | None = None
-    file_types: AcceptanceTestFileTypes | None = None
-    status: Literal["succeed", "failed"] | None = None
-
-    def get_config_dict(self) -> dict[str, Any]:
-        """Return the config dictionary.
-
-        If a config dictionary has already been loaded, return it. Otherwise, load
-        the config file and return the dictionary.
-        """
-        if self.config_dict:
-            return self.config_dict
-
-        if self.config_path:
-            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" or False
-
-    @property
-    def instance_name(self) -> str:
-        return self.config_path.stem if self.config_path else "Unnamed Scenario"
-
-    def __str__(self) -> str:
-        if self.id:
-            return f"'{self.id}' Test Scenario"
-        if self.config_path:
-            return f"'{self.config_path.name}' Test Scenario"
-
-        return f"'{hash(self)}' Test Scenario"

airbyte_cdk/test/declarative/test_suites/__init__.py

@@ -1,24 +0,0 @@
-# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
-"""Declarative test suites.
-
-Here we have base classes for a robust set of declarative connector test suites.
-"""
-
-from airbyte_cdk.test.declarative.test_suites.connector_base import (
-    ConnectorTestScenario,
-    generate_tests,
-)
-from airbyte_cdk.test.declarative.test_suites.declarative_sources import (
-    DeclarativeSourceTestSuite,
-)
-from airbyte_cdk.test.declarative.test_suites.destination_base import DestinationTestSuiteBase
-from airbyte_cdk.test.declarative.test_suites.source_base import SourceTestSuiteBase
-
-__all__ = [
-    "ConnectorTestScenario",
-    "ConnectorTestSuiteBase",
-    "DeclarativeSourceTestSuite",
-    "DestinationTestSuiteBase",
-    "SourceTestSuiteBase",
-    "generate_tests",
-]

airbyte_cdk/test/declarative/test_suites/connector_base.py

@@ -1,202 +0,0 @@
-# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
-"""Base class for connector test suites."""
-
-from __future__ import annotations
-
-import abc
-import functools
-import inspect
-import sys
-from pathlib import Path
-from typing import Any, Callable, Literal
-
-import pytest
-import yaml
-from pydantic import BaseModel
-from typing_extensions import override
-
-from airbyte_cdk import Connector
-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,
-)
-from airbyte_cdk.test.declarative.utils.job_runner import run_test_job
-
-ACCEPTANCE_TEST_CONFIG = "acceptance-test-config.yml"
-
-
-class JavaClass(str):
-    """A string that represents a Java class."""
-
-
-class DockerImage(str):
-    """A string that represents a Docker image."""
-
-
-class RunnableConnector(abc.ABC):
-    """A connector that can be run in a test scenario."""
-
-    @abc.abstractmethod
-    def launch(cls, args: list[str] | None) -> None: ...
-
-
-def generate_tests(metafunc) -> None:
-    """
-    A helper for pytest_generate_tests hook.
-
-    If a test method (in a class subclassed from our base class)
-    declares an argument 'instance', this function retrieves the
-    'scenarios' attribute from the test class and parametrizes that
-    test with the values from 'scenarios'.
-
-    ## Usage
-
-    ```python
-    from airbyte_cdk.test.declarative.test_suites.connector_base import (
-        generate_tests,
-        ConnectorTestSuiteBase,
-    )
-
-    def pytest_generate_tests(metafunc):
-        generate_tests(metafunc)
-
-    class TestMyConnector(ConnectorTestSuiteBase):
-        ...
-
-    ```
-    """
-    # Check if the test function requires an 'instance' argument
-    if "instance" in metafunc.fixturenames:
-        # Retrieve the test class
-        test_class = metafunc.cls
-        if test_class is None:
-            raise ValueError("Expected a class here.")
-        # Get the 'scenarios' attribute from the class
-        scenarios_attr = getattr(test_class, "get_scenarios", None)
-        if scenarios_attr is None:
-            raise ValueError(
-                f"Test class {test_class} does not have a 'scenarios' attribute. "
-                "Please define the 'scenarios' attribute in the test class."
-            )
-
-        scenarios = test_class.get_scenarios()
-        ids = [str(scenario) for scenario in scenarios]
-        metafunc.parametrize("instance", scenarios, ids=ids)
-
-
-class ConnectorTestSuiteBase(abc.ABC):
-    """Base class for connector test suites."""
-
-    acceptance_test_file_path = Path("./acceptance-test-config.json")
-    """The path to the acceptance test config file.
-
-    By default, this is set to the `acceptance-test-config.json` file in
-    the root of the connector source directory.
-    """
-
-    connector: type[Connector] | Path | JavaClass | DockerImage | None = None
-    """The connector class or path to the connector to test."""
-
-    working_dir: Path | None = None
-    """The root directory of the connector source code."""
-
-    @classmethod
-    def create_connector(
-        cls, scenario: ConnectorTestScenario
-    ) -> Source | AbstractSource | ConcurrentDeclarativeSource | RunnableConnector:
-        """Instantiate the connector class."""
-        raise NotImplementedError("Subclasses must implement this method.")
-
-    def run_test_scenario(
-        self,
-        verb: Literal["read", "check", "discover"],
-        test_scenario: ConnectorTestScenario,
-        *,
-        catalog: dict | None = None,
-    ) -> entrypoint_wrapper.EntrypointOutput:
-        """Run a test job from provided CLI args and return the result."""
-        return run_test_job(
-            self.create_connector(test_scenario),
-            verb,
-            test_instance=test_scenario,
-            catalog=catalog,
-        )
-
-    # Test Definitions
-
-    def test_check(
-        self,
-        instance: ConnectorTestScenario,
-    ) -> None:
-        """Run `connection` acceptance tests."""
-        result = self.run_test_scenario(
-            "check",
-            test_scenario=instance,
-        )
-        conn_status_messages: list[AirbyteMessage] = [
-            msg for msg in result._messages if msg.type == Type.CONNECTION_STATUS
-        ]  # noqa: SLF001  # Non-public API
-        assert len(conn_status_messages) == 1, (
-            "Expected exactly one CONNECTION_STATUS message. Got: \n" + "\n".join(result._messages)
-        )
-
-    @classmethod
-    @property
-    def acceptance_test_config_path(self) -> Path:
-        """Get the path to the acceptance test config file.
-
-        Check vwd and parent directories of cwd for the config file, and return the first one found.
-
-        Give up if the config file is not found in any parent directory.
-        """
-        current_dir = Path.cwd()
-        for parent_dir in current_dir.parents:
-            config_path = parent_dir / ACCEPTANCE_TEST_CONFIG
-            if config_path.exists():
-                return config_path
-        raise FileNotFoundError(
-            f"Acceptance test config file not found in any parent directory from : {Path.cwd()}"
-        )
-
-    @classmethod
-    def get_scenarios(
-        cls,
-    ) -> list[ConnectorTestScenario]:
-        """Get acceptance tests for a given category.
-
-        This has to be a separate function because pytest does not allow
-        parametrization of fixtures with arguments from the test class itself.
-        """
-        category = "connection"
-        all_tests_config = yaml.safe_load(cls.acceptance_test_config_path.read_text())
-        if "acceptance_tests" not in all_tests_config:
-            raise ValueError(
-                f"Acceptance tests config not found in {cls.acceptance_test_config_path}."
-                f" Found only: {str(all_tests_config)}."
-            )
-        if category not in all_tests_config["acceptance_tests"]:
-            return []
-        if "tests" not in all_tests_config["acceptance_tests"][category]:
-            raise ValueError(f"No tests found for category {category}")
-
-        tests_scenarios = [
-            ConnectorTestScenario.model_validate(test)
-            for test in all_tests_config["acceptance_tests"][category]["tests"]
-            if "iam_role" not in test["config_path"]
-        ]
-        working_dir = cls.working_dir or Path()
-        for test in tests_scenarios:
-            if test.config_path:
-                test.config_path = working_dir / test.config_path
-            if test.configured_catalog_path:
-                test.configured_catalog_path = working_dir / test.configured_catalog_path
-        return tests_scenarios

airbyte_cdk/test/declarative/test_suites/declarative_sources.py

@@ -1,48 +0,0 @@
-import os
-from hashlib import md5
-from pathlib import Path
-from typing import Any, cast
-
-import yaml
-
-from airbyte_cdk.sources.declarative.concurrent_declarative_source import (
-    ConcurrentDeclarativeSource,
-)
-from airbyte_cdk.test.declarative.models import ConnectorTestScenario
-from airbyte_cdk.test.declarative.test_suites.source_base import (
-    SourceTestSuiteBase,
-)
-
-
-def md5_checksum(file_path: Path) -> str:
-    with open(file_path, "rb") as file:
-        return md5(file.read()).hexdigest()
-
-
-class DeclarativeSourceTestSuite(SourceTestSuiteBase):
-    manifest_path = Path("manifest.yaml")
-    components_py_path: Path | None = None
-
-    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()
-        # source_config = connector_test.get_source_config()
-
-        manifest_dict = yaml.safe_load(self.manifest_path.read_text())
-        if self.components_py_path and self.components_py_path.exists():
-            os.environ["AIRBYTE_ENABLE_UNSAFE_CODE"] = "true"
-            config["__injected_components_py"] = self.components_py_path.read_text()
-            config["__injected_components_py_checksums"] = {
-                "md5": md5_checksum(self.components_py_path),
-            }
-
-        return ConcurrentDeclarativeSource(
-            config=config,
-            catalog=None,
-            state=None,
-            source_config=manifest_dict,
-        )

airbyte_cdk/test/declarative/test_suites/destination_base.py

@@ -1,12 +0,0 @@
-# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
-"""Base class for destination test suites."""
-
-from airbyte_cdk.test.declarative.test_suites.connector_base import ConnectorTestSuiteBase
-
-
-class DestinationTestSuiteBase(ConnectorTestSuiteBase):
-    """Base class for destination test suites.
-
-    This class provides a base set of functionality for testing destination connectors, and it
-    inherits all generic connector tests from the `ConnectorTestSuiteBase` class.
-    """

airbyte_cdk/test/declarative/test_suites/source_base.py

@@ -1,129 +0,0 @@
-# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
-"""Base class for source test suites."""
-
-from dataclasses import asdict
-from pathlib import Path
-
-import pytest
-
-from airbyte_cdk.models import (
-    AirbyteMessage,
-    AirbyteStream,
-    ConfiguredAirbyteCatalog,
-    ConfiguredAirbyteStream,
-    DestinationSyncMode,
-    SyncMode,
-    Type,
-)
-from airbyte_cdk.test import entrypoint_wrapper
-from airbyte_cdk.test.declarative.models import (
-    ConnectorTestScenario,
-)
-from airbyte_cdk.test.declarative.test_suites.connector_base import (
-    ConnectorTestSuiteBase,
-)
-from airbyte_cdk.test.declarative.utils.job_runner import run_test_job
-
-
-class SourceTestSuiteBase(ConnectorTestSuiteBase):
-    """Base class for source test suites.
-
-    This class provides a base set of functionality for testing source connectors, and it
-    inherits all generic connector tests from the `ConnectorTestSuiteBase` class.
-    """
-
-    def test_check(
-        self,
-        instance: ConnectorTestScenario,
-    ) -> None:
-        """Run `connection` acceptance tests."""
-        result: entrypoint_wrapper.EntrypointOutput = run_test_job(
-            self.create_connector(instance),
-            "check",
-            test_instance=instance,
-        )
-        conn_status_messages: list[AirbyteMessage] = [
-            msg for msg in result._messages if msg.type == Type.CONNECTION_STATUS
-        ]  # noqa: SLF001  # Non-public API
-        assert len(conn_status_messages) == 1, (
-            "Expected exactly one CONNECTION_STATUS message. Got: \n" + "\n".join(result._messages)
-        )
-
-    def test_basic_read(
-        self,
-        instance: ConnectorTestScenario,
-    ) -> None:
-        """Run acceptance tests."""
-        discover_result = run_test_job(
-            self.create_connector(instance),
-            "discover",
-            test_instance=instance,
-        )
-        if instance.expect_exception:
-            assert discover_result.errors, "Expected exception but got none."
-            return
-
-        configured_catalog = ConfiguredAirbyteCatalog(
-            streams=[
-                ConfiguredAirbyteStream(
-                    stream=stream,
-                    sync_mode=SyncMode.full_refresh,
-                    destination_sync_mode=DestinationSyncMode.append_dedup,
-                )
-                for stream in discover_result.catalog.catalog.streams
-            ]
-        )
-        result = run_test_job(
-            self.create_connector(instance),
-            "read",
-            test_instance=instance,
-            catalog=configured_catalog,
-        )
-
-        if not result.records:
-            raise AssertionError("Expected records but got none.")  # noqa: TRY003
-
-    def test_fail_with_bad_catalog(
-        self,
-        instance: ConnectorTestScenario,
-    ) -> None:
-        """Test that a bad catalog fails."""
-        invalid_configured_catalog = ConfiguredAirbyteCatalog(
-            streams=[
-                # Create ConfiguredAirbyteStream which is deliberately invalid
-                # with regard to the Airbyte Protocol.
-                # This should cause the connector to fail.
-                ConfiguredAirbyteStream(
-                    stream=AirbyteStream(
-                        name="__AIRBYTE__stream_that_does_not_exist",
-                        json_schema={
-                            "type": "object",
-                            "properties": {"f1": {"type": "string"}},
-                        },
-                        supported_sync_modes=[SyncMode.full_refresh],
-                    ),
-                    sync_mode="INVALID",
-                    destination_sync_mode="INVALID",
-                )
-            ]
-        )
-        # Set expected status to "failed" to ensure the test fails if the connector.
-        instance.status = "failed"
-        result = self.run_test_scenario(
-            "read",
-            test_scenario=instance,
-            catalog=asdict(invalid_configured_catalog),
-        )
-        assert result.errors, "Expected errors but got none."
-        assert result.trace_messages, "Expected trace messages but got none."
-
-    def test_discover(
-        self,
-        instance: ConnectorTestScenario,
-    ) -> None:
-        """Run acceptance tests."""
-        run_test_job(
-            self.create_connector(instance),
-            "check",
-            test_instance=instance,
-        )

airbyte_cdk/test/declarative/utils/job_runner.py

@@ -1,128 +0,0 @@
-import tempfile
-import uuid
-from pathlib import Path
-from typing import Any, Callable, Literal
-
-import orjson
-
-from airbyte_cdk import Connector
-from airbyte_cdk.models import (
-    Status,
-)
-from airbyte_cdk.sources.abstract_source import AbstractSource
-from airbyte_cdk.sources.declarative.declarative_source import DeclarativeSource
-from airbyte_cdk.test import entrypoint_wrapper
-from airbyte_cdk.test.declarative.models import (
-    ConnectorTestScenario,
-)
-
-
-def run_test_job(
-    connector: Connector | type[Connector] | Callable[[], Connector],
-    verb: Literal["read", "check", "discover"],
-    test_instance: ConnectorTestScenario,
-    *,
-    catalog: dict[str, Any] | None = None,
-) -> entrypoint_wrapper.EntrypointOutput:
-    """Run a test job from provided CLI args and return the result."""
-    if not connector:
-        raise ValueError("Connector is required")
-
-    connector_obj: Connector
-    if isinstance(connector, type):
-        connector_obj = connector()
-    elif isinstance(connector, Connector):
-        connector_obj = connector
-    elif isinstance(connector, DeclarativeSource | AbstractSource):
-        connector_obj = connector
-    elif isinstance(connector, Callable):
-        try:
-            connector_obj = connector()
-        except Exception as ex:
-            if not test_instance.expect_exception:
-                raise
-
-            return entrypoint_wrapper.EntrypointOutput(
-                messages=[],
-                uncaught_exception=ex,
-            )
-    else:
-        raise ValueError(f"Invalid source type: {type(connector)}")
-
-    args: list[str] = [verb]
-    if test_instance.config_path:
-        args += ["--config", str(test_instance.config_path)]
-    elif test_instance.config_dict:
-        config_path = (
-            Path(tempfile.gettempdir()) / "airbyte-test" / f"temp_config_{uuid.uuid4().hex}.json"
-        )
-        config_path.parent.mkdir(parents=True, exist_ok=True)
-        config_path.write_text(orjson.dumps(test_instance.config_dict).decode())
-        args += ["--config", str(config_path)]
-
-    catalog_path: Path | None = None
-    if verb not in ["discover", "check"]:
-        # We need a catalog for read.
-        if catalog:
-            # Write the catalog to a temp json file and pass the path to the file as an argument.
-            catalog_path = (
-                Path(tempfile.gettempdir())
-                / "airbyte-test"
-                / f"temp_catalog_{uuid.uuid4().hex}.json"
-            )
-            catalog_path.parent.mkdir(parents=True, exist_ok=True)
-            catalog_path.write_text(orjson.dumps(catalog).decode())
-        elif test_instance.configured_catalog_path:
-            catalog_path = Path(test_instance.configured_catalog_path)
-
-        if catalog_path:
-            args += ["--catalog", str(catalog_path)]
-
-    # This is a bit of a hack because the source needs the catalog early.
-    # Because it *also* can fail, we have ot redundantly wrap it in a try/except block.
-
-    result: entrypoint_wrapper.EntrypointOutput = entrypoint_wrapper._run_command(  # noqa: SLF001  # Non-public API
-        source=connector_obj,
-        args=args,
-        expecting_exception=test_instance.expect_exception,
-    )
-    if result.errors and not test_instance.expect_exception:
-        raise AssertionError(
-            "\n\n".join(
-                [str(err.trace.error).replace("\\n", "\n") for err in result.errors],
-            )
-        )
-
-    if verb == "check":
-        # Check is expected to fail gracefully without an exception.
-        # Instead, we assert that we have a CONNECTION_STATUS message with
-        # a failure status.
-        assert not result.errors, "Expected no errors from check. Got:\n" + "\n".join(
-            [str(error) for error in result.errors]
-        )
-        assert len(result.connection_status_messages) == 1, (
-            "Expected exactly one CONNECTION_STATUS message. Got "
-            f"{len(result.connection_status_messages)}:\n"
-            + "\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(msg) for msg in result.connection_status_messages])
-            )
-        return result
-
-    # For all other verbs, we assert check that an exception is raised (or not).
-    if test_instance.expect_exception:
-        if not result.errors:
-            raise AssertionError("Expected exception but got none.")
-
-        return result
-    if result.errors:
-        raise AssertionError(
-            "\n\n".join(
-                [str(err.trace.error).replace("\\n", "\n") for err in result.errors],
-            )
-        )
-
-    return result

airbyte_cdk/test/fixtures/auto.py

@@ -1,14 +0,0 @@
-"""Auto-use fixtures for pytest.
-
-WARNING: Importing this module will automatically apply these fixtures. If you want to selectively
-enable fixtures in a different context, you can import directly from the `fixtures.general` module.
-
-
-Usage:
-
-```python
-from airbyte_cdk.test.fixtures import auto
-# OR
-from airbyte_cdk.test.fixtures.auto import *
-```
-"""

airbyte_cdk/test/pytest_config/plugin.py

@@ -1,46 +0,0 @@
-"""Global pytest configuration for the Airbyte CDK tests."""
-
-from pathlib import Path
-from typing import Optional
-
-import pytest
-
-
-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: pytest.Config) -> None:
-    config.addinivalue_line("markers", "connector: mark test as a connector test")
-
-
-def pytest_addoption(parser: pytest.Parser) -> None:
-    parser.addoption(
-        "--run-connector",
-        action="store_true",
-        default=False,
-        help="run connector tests",
-    )
-
-
-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")
-    for item in items:
-        if "connector" in item.keywords:
-            item.add_marker(skip_connector)
-
-
-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: pytest.Item, nextitem: Optional[pytest.Item]) -> None:
-    # This hook is called after each test function is executed
-    print(f"Tearing down test: {item.name}")