use proper destination test suite class

Author: aaronsteers

airbyte_cdk/cli/airbyte_cdk/_connector.py

@@ -67,7 +67,7 @@
 
 TEST_FILE_TEMPLATE = '''
 # Copyright (c) 2025 Airbyte, Inc., all rights reserved.
-"""FAST Airbyte Standard Tests for the {connector_name} source."""
+"""FAST Airbyte Standard Tests for the {connector_name} connector."""
 
 #from airbyte_cdk.test.standard_tests import {base_class_name}
 from airbyte_cdk.test.standard_tests.util import create_connector_test_suite
@@ -81,11 +81,13 @@
     connector_directory=Path(),
 )
 
+# Uncomment the following lines to create a custom test suite class:
+#
 # class TestSuite({base_class_name}):
-#     """Test suite for the {connector_name} source.
-
-#     This class inherits from SourceTestSuiteBase and implements all of the tests in the suite.
-
+#     """Test suite for the `{connector_name}` connector.
+#
+#     This class inherits from `{base_class_name}` and implements all of the tests in the suite.
+#
 #     As long as the class name starts with "Test", pytest will automatically discover and run the
 #     tests in this class.
 #     """

airbyte_cdk/test/standard_tests/util.py

@@ -10,22 +10,14 @@
 from airbyte_cdk.test.standard_tests.declarative_sources import (
     DeclarativeSourceTestSuite,
 )
+from airbyte_cdk.test.standard_tests.destination_base import DestinationTestSuiteBase
 from airbyte_cdk.test.standard_tests.docker_base import DockerConnectorTestSuite
 from airbyte_cdk.test.standard_tests.source_base import SourceTestSuiteBase
 from airbyte_cdk.utils.connector_paths import (
     METADATA_YAML,
     find_connector_root_from_name,
 )
 
-TEST_CLASS_MAPPING: dict[
-    Literal["python", "manifest-only", "java"],
-    type[DockerConnectorTestSuite],
-] = {
-    "python": SourceTestSuiteBase,
-    "manifest-only": DeclarativeSourceTestSuite,
-    "java": DockerConnectorTestSuite,
-}
-
 
 def create_connector_test_suite(
     *,
@@ -55,17 +47,25 @@ def create_connector_test_suite(
         )
     metadata_dict: dict[str, Any] = yaml.safe_load(metadata_yaml_path.read_text())
     metadata_tags = metadata_dict["data"].get("tags", [])
-    for language_option in TEST_CLASS_MAPPING:
-        if f"language:{language_option}" in metadata_tags:
-            language = language_option
-            test_suite_class = TEST_CLASS_MAPPING[language]
-            break
-    else:
+    language_tag = next((tag for tag in metadata_tags if tag.startswith("language:")), None)
+    if not language_tag:
         raise ValueError(
-            f"Unsupported connector type. "
-            f"Supported language values are: {', '.join(TEST_CLASS_MAPPING.keys())}. "
+            f"Metadata YAML file '{metadata_yaml_path}' does not contain a 'language' tag. "
+            "Please ensure the metadata file is correctly configured."
             f"Found tags: {', '.join(metadata_tags)}"
         )
+    language = language_tag.split(":", 1)[1]
+
+    if language == "java":
+        test_suite_class = DockerConnectorTestSuite
+    elif language == "manifest-only":
+        test_suite_class = DeclarativeSourceTestSuite
+    elif language == "python" and connector_name.startswith("source-"):
+        test_suite_class = SourceTestSuiteBase
+    elif language == "python" and connector_name.startswith("destination-"):
+        test_suite_class = DestinationTestSuiteBase
+    else:
+        raise ValueError("Unsupported language for connector '{connector_name}': {language}")
 
     subclass_overrides: dict[str, Any] = {
         "get_connector_root_dir": classmethod(lambda cls: connector_directory),