如何为含多节点与依赖服务的Go项目搭建集成测试环境？

一、用Docker快速拉起MySQL和Redis依赖 #

用容器化工具可以快速隔离、启动测试所需的外部服务，避免本地环境干扰：

• 编写docker-compose.yml配置文件：

version: '3.8'
services:
  mysql:
    image: mysql:8.0
    environment:
      MYSQL_ROOT_PASSWORD: test123
      MYSQL_DATABASE: test_db
    ports:
      - "3306:3306"
    healthcheck:
      test: ["CMD", "mysqladmin", "ping", "-h", "localhost"]
      interval: 5s
      timeout: 5s
      retries: 5
  redis:
    image: redis:7-alpine
    ports:
      - "6379:6379"
    healthcheck:
      test: ["CMD", "redis-cli", "ping"]
      interval: 3s
      timeout: 3s
      retries: 5

• 启动依赖服务：docker-compose up -d，等待健康检查通过后再进行下一步。

二、自动化启动三个Go服务节点 #

手动启动服务无法满足测试自动化需求，建议通过脚本或代码自动拉起并管理进程：

1. 提前编译服务二进制文件（减少重复编译开销）：

mkdir -p bin
go build -o ./bin/http main.go
go build -o ./bin/tasks main.go
go build -o ./bin/scheduler main.go

2. 后台启动服务并记录PID（方便测试结束后清理）：

mkdir -p logs pids
./bin/http http > ./logs/http.log 2>&1 & echo $! > ./pids/http.pid
./bin/tasks tasks > ./logs/tasks.log 2>&1 & echo $! > ./pids/tasks.pid
./bin/scheduler scheduler > ./logs/scheduler.log 2>&1 & echo $! > ./pids/scheduler.pid

3. 添加健康检查逻辑：比如检查HTTP服务的监听端口、调用健康接口，或验证Redis/MySQL的连通性，确保服务完全启动后再执行测试。

三、用Makefile串联完整测试流程 #

编写Makefile可以一键完成「启动依赖→编译服务→启动服务→执行测试→清理资源」的全流程：

TEST_ENV=test

.PHONY: test
test: start-deps build-services start-services wait-for-services run-tests stop-services stop-deps

start-deps:
	docker-compose up -d
	@echo "等待依赖服务就绪..."
	@sleep 10 # 也可以用脚本检查健康状态替代固定等待

build-services:
	mkdir -p bin logs pids
	go build -o ./bin/http main.go
	go build -o ./bin/tasks main.go
	go build -o ./bin/scheduler main.go

start-services:
	./bin/http http > ./logs/http.log 2>&1 & echo $! > ./pids/http.pid
	./bin/tasks tasks > ./logs/tasks.log 2>&1 & echo $! > ./pids/tasks.pid
	./bin/scheduler scheduler > ./logs/scheduler.log 2>&1 & echo $! > ./pids/scheduler.pid

wait-for-services:
	@echo "等待Go服务启动..."
	@sleep 5 # 可替换为HTTP健康检查脚本

run-tests:
	go test ./integration-tests -v -tags=$(TEST_ENV)

stop-services:
	@for pid in $(shell cat ./pids/*.pid 2>/dev/null); do kill $$pid || true; done
	rm -rf ./pids/* ./logs/*

stop-deps:
	docker-compose down -v

四、集成测试用例编写示例 #

测试时直接验证完整数据流，比如发送HTTP请求后，检查Redis队列处理情况、MySQL数据状态：

package integrationtests

import (
	"net/http"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	// 导入你的Redis、Gorm客户端包
)

func TestFullDataFlow(t *testing.T) {
	// 1. 发送请求到HTTP节点
	resp, err := http.Post("http://localhost:8080/your-api-endpoint", "application/json", nil)
	assert.NoError(t, err)
	assert.Equal(t, http.StatusOK, resp.StatusCode)

	// 2. 等待Tasks节点处理消息
	time.Sleep(2 * time.Second)

	// 3. 验证Redis任务队列已被处理（示例）
	// redisClient := redis.NewClient(...)
	// queueLen, err := redisClient.LLen("task_queue").Result()
	// assert.NoError(t, err)
	// assert.Equal(t, int64(0), queueLen)

	// 4. 等待Scheduler执行一次检查（可在测试环境缩短周期到1秒）
	time.Sleep(31 * time.Second)

	// 5. 验证MySQL中状态是否符合预期
	// var status YourStatusModel
	// err = gormDB.First(&status, "trigger_condition = ?", true).Error
	// assert.NoError(t, err)
	// assert.True(t, status.XTriggered)
}

五、优化建议 #

• 测试环境独立配置：给测试用例单独配置Redis数据库、MySQL库名，避免和开发环境数据冲突。
• 缩短Scheduler周期：在测试模式下通过命令行参数或配置文件，将30秒检查周期改为1秒，加快测试执行速度。
• 用testcontainers-go替代Docker Compose：可以在Go测试代码中直接启动、管理容器，无需额外配置文件，更灵活。
• 强制资源清理：测试结束后确保所有进程、容器都被终止，避免残留资源占用端口或内存。

内容的提问来源于stack exchange，提问作者Mohammad Safakhou