Merge pull request #354 from docker/add-testcontainers

Add integration tests using Testcontainers

Author: ericcurtin

.github/workflows/integration-test.yml

@@ -0,0 +1,26 @@
+name: Integration Tests
+
+on:
+  workflow_dispatch:  # Manual trigger only
+  push:
+
+jobs:
+  integration-test:
+    runs-on: ubuntu-latest
+    timeout-minutes: 10
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8
+
+      - name: Set up Go
+        uses: actions/setup-go@44694675825211faa026b3c33043df3e48a5fa00
+        with:
+          go-version: 1.24.2
+          cache: true
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@e468171a9de216ec08956ac3ada2f0791b6bd435
+
+      - name: Run integration tests
+        run: make integration-tests

Makefile

@@ -27,8 +27,11 @@ SOURCE ?=
 TAG ?=
 LICENSE ?=
 
+# Test configuration
+BUILD_DMR ?= 1
+
 # Main targets
-.PHONY: build run clean test docker-build docker-build-multiplatform docker-run docker-build-vllm docker-run-vllm docker-run-impl help validate model-distribution-tool
+.PHONY: build run clean test integration-tests docker-build docker-build-multiplatform docker-run docker-build-vllm docker-run-vllm docker-run-impl help validate model-distribution-tool
 # Default target
 .DEFAULT_GOAL := build
 
@@ -60,6 +63,19 @@ clean:
 test:
 	go test -v ./...
 
+integration-tests:
+	@echo "Running integration tests..."
+	@echo "Note: This requires Docker to be running"
+	@echo "Checking test naming conventions..."
+	@INVALID_TESTS=$$(grep "^func Test" cmd/cli/commands/integration_test.go | grep -v "^func TestIntegration"); \
+	if [ -n "$$INVALID_TESTS" ]; then \
+		echo "Error: Found test functions that don't start with 'TestIntegration':"; \
+		echo "$$INVALID_TESTS" | sed 's/func \([^(]*\).*/\1/'; \
+		exit 1; \
+	fi
+	@BUILD_DMR=$(BUILD_DMR) go test -v -race -count=1 -tags=integration -run "^TestIntegration" -timeout=5m ./cmd/cli/commands
+	@echo "Integration tests completed!"
+
 validate:
 	find . -type f -name "*.sh" | xargs shellcheck