feat(cli): add source-smoke-test CLI

airbyte/cli/__init__.py

@@ -0,0 +1,7 @@
+# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
+"""CLI modules for PyAirbyte."""
+
+from airbyte.cli.pyab import cli
+
+
+__all__ = ["cli"]

airbyte/cli/__main__.py

@@ -0,0 +1,8 @@
+# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
+"""Allow running the CLI via python -m airbyte.cli."""
+
+from airbyte.cli.pyab import cli
+
+
+if __name__ == "__main__":
+    cli()

airbyte/cli.py → airbyte/cli/pyab.py

@@ -170,7 +170,7 @@ def _is_executable_path(connector_str: str) -> bool:
 
 def _get_connector_name(connector: str) -> str:
     if _is_docker_image(connector):
-        return connector.split(":")[0].split("/")[-1]
+        return connector.split(":", maxsplit=1)[0].rsplit("/", maxsplit=1)[-1]
 
     return connector
 

airbyte/cli/smoke_test_source/__init__.py

@@ -0,0 +1,18 @@
+# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
+"""Smoke test source for destination regression testing.
+
+This module provides a synthetic data source that generates test data
+covering common edge cases that break destinations: type variations,
+null handling, naming edge cases, schema variations, and batch sizes.
+
+.. warning::
+    This module is experimental and subject to change without notice.
+    The APIs and behavior may be modified or removed in future versions.
+"""
+
+from airbyte.cli.smoke_test_source.source import SourceSmokeTest
+
+
+__all__ = [
+    "SourceSmokeTest",
+]