chore: enable docker in docker tests to pass on arm machines

Author: terry-docker

Dockerfile

@@ -1,5 +1,8 @@
 ARG PYTHON_VERSION=3.10
 FROM python:${PYTHON_VERSION}-slim-bookworm
+ARG POETRY_EXTRAS
+
+ENV PYTHONPATH=/workspace
 
 ENV POETRY_NO_INTERACTION=1 \
     POETRY_VIRTUALENVS_IN_PROJECT=1 \
@@ -20,10 +23,18 @@ RUN bash -c 'python -m venv /opt/poetry-venv && source $_/bin/activate && pip in
 # install dependencies with poetry
 COPY pyproject.toml .
 COPY poetry.lock .
-RUN poetry install --all-extras --with dev --no-root
+RUN if [ "$POETRY_EXTRAS" = "" ]; then \
+        poetry install --all-extras --with dev --no-root; \
+    else \
+        poetry install --extras "$POETRY_EXTRAS" --with dev --no-root; \
+    fi
 
 # copy project source
 COPY . .
 
 # install project with poetry
-RUN poetry install --all-extras --with dev
+RUN if [ "$POETRY_EXTRAS" = "" ]; then \
+        poetry install --all-extras --with dev; \
+    else \
+        poetry install --extras "$POETRY_EXTRAS" --with dev; \
+    fi

core/__init__.py

@@ -4,8 +4,11 @@
 from typing import Callable
 from testcontainers.core.container import DockerClient
 from pprint import pprint
+from testcontainers.core.utils import is_arm
 import sys
 
+from .list_arm_extras import get_arm_extras
+
 PROJECT_DIR = Path(__file__).parent.parent.parent.resolve()
 
 
@@ -25,11 +28,16 @@ def python_testcontainer_image() -> str:
     py_version = ".".join(map(str, sys.version_info[:2]))
     image_name = f"testcontainers-python:{py_version}"
     client = DockerClient()
+    build_args = {"PYTHON_VERSION": py_version}
+
+    if is_arm():
+        build_args["POETRY_EXTRAS"] = get_arm_extras()
+
     client.build(
         path=str(PROJECT_DIR),
         tag=image_name,
         rm=False,
-        buildargs={"PYTHON_VERSION": py_version},
+        buildargs=build_args,
     )
     return image_name
 

core/tests/__init__.py

@@ -0,0 +1,18 @@
+from pathlib import Path
+from testcontainers.core.utils import is_arm
+
+try:
+    import tomllib  # Python 3.11+
+except ImportError:
+    import tomli as tomllib
+
+SKIPPED_EXTRAS = {"db2"}  # skip incompatible extras
+
+
+def get_arm_extras():
+    with Path("pyproject.toml").open("rb") as f:
+        data = tomllib.load(f)
+
+    extras = data["tool"]["poetry"]["extras"]
+    skip = SKIPPED_EXTRAS
+    return " ".join(k for k in extras if k not in skip)