Fix integration tests timeout by pre-pulling Docker image

- Add Docker image pre-pull step to CI workflow before running tests
- Add explicit image pull with error handling in test fixtures
- Remove CI skip condition for integration tests
- Improve error messages for Docker-related failures

Co-authored-by: rnovatorov <20299819+rnovatorov@users.noreply.github.com>

Author: Copilot

.github/workflows/ci.yml

@@ -27,6 +27,9 @@ jobs:
       - name: Install dependencies
         run: make install-deps
 
+      - name: Pull Docker images for integration tests
+        run: docker pull eclipse-mosquitto:latest
+
       - name: Run checks
         run: make check
 

Makefile

@@ -42,12 +42,7 @@ test-unit:
 
 .PHONY: test-integration
 test-integration:
-# TODO: Figure out why integration tests time out in CI environment.
-ifdef CI
-	$(warning skipping integration tests in CI environment)
-else
 	pipenv run pytest -vv --capture=no tests/integration
-endif
 
 .PHONY: get-pipenv
 get-pipenv:

tests/integration/conftest.py

@@ -32,6 +32,22 @@ def fixture_mosquitto_container(
     docker_client: docker.DockerClient,
 ) -> Generator[docker.models.containers.Container, None, None]:
     name = "enapter-python-sdk-integration-tests-mosquitto"
+    image = os.getenv("MOSQUITTO_IMAGE", "eclipse-mosquitto:latest")
+
+    # Ensure the image is available locally
+    try:
+        docker_client.images.get(image)
+    except docker.errors.ImageNotFound:
+        # Pull the image if not available locally
+        # This is handled by the CI workflow, but we keep it here for local testing
+        try:
+            docker_client.images.pull(image)
+        except docker.errors.APIError as e:
+            raise RuntimeError(
+                f"Failed to pull Docker image {image}. "
+                f"Please ensure Docker is running and you have network connectivity. "
+                f"Error: {e}"
+            ) from e
 
     try:
         old_mosquitto = docker_client.containers.get(name)
@@ -41,7 +57,7 @@ def fixture_mosquitto_container(
         old_mosquitto.remove(force=True)
 
     mosquitto = docker_client.containers.run(
-        os.getenv("MOSQUITTO_IMAGE", "eclipse-mosquitto:latest"),
+        image,
         ["mosquitto", "-c", "/mosquitto-no-auth.conf"],
         name=name,
         network="bridge",