add testing

datahub-project · Jan 24, 2025 · 8cab3d4 · 8cab3d4
1 parent 128939e
commit 8cab3d4
Show file tree

Hide file tree

Showing 19 changed files with 3,848 additions and 37 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/sql/mariadb.py b/metadata-ingestion/src/datahub/ingestion/source/sql/mariadb.py
@@ -20,17 +20,16 @@
     MySQLProcedureContainer,
     MySQLStoredProcedure,
 )
-from datahub.ingestion.source.sql.mysql.source import MySQLConnectionConfig
 
 logger: logging.Logger = logging.getLogger(__name__)
 
 
-class MariaDBConnectionConfig(MySQLConnectionConfig):
+class MariaDBConfig(MySQLConfig):
     host_port: str = Field(default="localhost:3306", description="MariaDB host URL.")
 
 
 @platform_name("MariaDB")
-@config_class(MariaDBConnectionConfig)
+@config_class(MariaDBConfig)
 @support_status(SupportStatus.CERTIFIED)
 @capability(SourceCapability.PLATFORM_INSTANCE, "Enabled by default")
 @capability(SourceCapability.DOMAINS, "Supported via the `domain` config field")

diff --git a/metadata-ingestion/tests/integration/mariadb/docker-compose.yml b/metadata-ingestion/tests/integration/mariadb/docker-compose.yml
@@ -0,0 +1,12 @@
+version: "3"
+services:
+  testmariadb:
+    image: mariadb:10.6
+    container_name: testmariadb
+    environment:
+      MARIADB_ROOT_PASSWORD: password
+      MARIADB_DATABASE: test_db
+    ports:
+      - "53300:3306"
+    volumes:
+      - ./setup:/docker-entrypoint-initdb.d