using testcontainers

Author: daimor

.github/workflows/python-publish.yml

@@ -28,17 +28,6 @@ jobs:
     runs-on: ubuntu-latest
     env:
       IMAGE: ${{ matrix.image }}
-    services:
-      iris:
-        image: ${{ matrix.image }}
-        env:
-          IRIS_USERNAME: sqlalchemy
-          IRIS_PASSWORD: sqlalchemy
-          IRIS_NAMESPACE: TEST
-        #  /irisHealth.sh is just useless
-        options: >-
-          --health-cmd true
-          --health-start-period 30s
     steps:
     - uses: actions/checkout@v3
     - name: Install requirements
@@ -49,7 +38,7 @@ jobs:
         pip install -e .
     - name: Run Tests
       run: |
-        pytest --dburi iris://sqlalchemy:sqlalchemy@iris:1972/TEST
+        pytest --container ${{ matrix.image }}
 
   deploy:
     needs: test

tests/conftest.py

@@ -1,9 +1,59 @@
 from sqlalchemy.dialects import registry
 import pytest
+import time
+
+from sqlalchemy.testing.plugin.plugin_base import pre
+
+from testcontainers.iris import IRISContainer
 
 registry.register("iris.iris", "sqlalchemy_iris.iris", "IRISDialect_iris")
 registry.register("iris.emb", "sqlalchemy_iris.embedded", "IRISDialect_emb")
 
 pytest.register_assert_rewrite("sqlalchemy.testing.assertions")
 
 from sqlalchemy.testing.plugin.pytestplugin import *  # noqa
+
+original_pytest_addoption = pytest_addoption
+
+
+def pytest_addoption(parser):
+    original_pytest_addoption(parser)
+
+    group = parser.getgroup("iris")
+
+    group.addoption(
+        "--container",
+        action="store",
+        default=None,
+        type=str,
+        help="Docker image with IRIS",
+    )
+
+
+@pre
+def start_container(opt, file_config):
+    global iris
+    iris = None
+    if not opt.container:
+        return
+    try:
+        print("Starting IRIS container:", opt.container)
+        iris = IRISContainer(
+            opt.container,
+            username="sqlalchemy",
+            password="sqlalchemy",
+            dbname="TEST",
+        )
+        iris.start()
+        print("dburi:", iris.get_connection_url())
+        opt.dburi = [iris.get_connection_url()]
+    except Exception as e:
+        iris = None
+        pytest.exit("Failed to start IRIS container")
+
+
+def pytest_unconfigure(config):
+    global iris
+    if iris:
+        print("Stopping IRIS container")
+        iris.stop()