Merge pull request #93 from bruin-data/fix/skip-test-clickhouse

Enable mysql source in test

ingestr/main_test.py

@@ -38,6 +38,7 @@
 from testcontainers.localstack import LocalStackContainer  # type: ignore
 from testcontainers.postgres import PostgresContainer  # type: ignore
 from typer.testing import CliRunner
+from testcontainers.mysql import MySqlContainer  # type: ignore
 
 from ingestr.main import app
 from ingestr.src.appstore.errors import (
@@ -447,7 +448,6 @@ def start(self) -> str:
         url = super().start()
         if self.container is None:
             raise ValueError("Container is not initialized.")
-
         port = self.container.get_exposed_port(8123)
         return (
             url.replace("clickhouse://", "clickhouse+native://") + f"?http_port={port}"
@@ -478,12 +478,12 @@ def stop_fully(self):
 clickHouseDocker = ClickhouseDockerImage(
     lambda: ClickHouseContainer(CLICKHOUSE_IMAGE).start()
 )
+mysqlDocker = DockerImage(lambda: MySqlContainer(MYSQL8_IMAGE, username="root").start())
+
 SOURCES = {
     "postgres": pgDocker,
     "duckdb": DuckDb(),
-    #     "mysql8": DockerImage(
-    #         lambda: MySqlContainer(MYSQL8_IMAGE, username="root").start()
-    #     ),
+    "mysql8": mysqlDocker,
     # "sqlserver": DockerImage(
     #     lambda: SqlServerContainer(MSSQL22_IMAGE, dialect="mssql").start(),
     #     "?driver=ODBC+Driver+18+for+SQL+Server&TrustServerCertificate=Yes",
@@ -497,7 +497,6 @@ def stop_fully(self):
     "clickhouse+native": clickHouseDocker,
 }
 
-
 @pytest.fixture(scope="session", autouse=True)
 def manage_containers():
     # Run all tests
@@ -528,9 +527,6 @@ def test_create_replace(source, dest):
         source_uri = source_future.result()
         dest_uri = dest_future.result()
 
-    print(f"source_uri: {source_uri}")
-    print(f"dest_uri: {dest_uri}")
-
     db_to_db_create_replace(source_uri, dest_uri)
     source.stop()
     dest.stop()
@@ -1203,7 +1199,7 @@ def test_kafka_to_db(dest):
     with ThreadPoolExecutor() as executor:
         dest_future = executor.submit(dest.start)
         source_future = executor.submit(
-            KafkaContainer("confluentinc/cp-kafka:7.6.0").start, timeout=60
+            KafkaContainer("confluentinc/cp-kafka:7.6.0").start, timeout=120
         )
         dest_uri = dest_future.result()
         kafka = source_future.result()