Function setupPostgresContainer

pkg/session/postgres/service_test.go:26–106

setupPostgresContainer 启动 PostgreSQL 容器用于测试

(t *testing.T)

// setupPostgresContainer 启动 PostgreSQL 容器用于测试
func setupPostgresContainer(t *testing.T) (service *Service, cleanup func()) {
	t.Helper()

	// 检查是否在 CI 环境或需要跳过集成测试
	if testing.Short() {
		t.Skip("Skipping PostgreSQL integration test in short mode")
	}
	if os.Getenv("SKIP_INTEGRATION_TESTS") != "" {
		t.Skip("Skipping PostgreSQL integration test (SKIP_INTEGRATION_TESTS is set)")
	}

	// 捕获 Docker 不可用时的 panic
	defer func() {
		if r := recover(); r != nil {
			t.Skipf("Docker not available, skipping PostgreSQL integration test: %v", r)
		}
	}()

	ctx := context.Background()

	// 创建 PostgreSQL 容器
	req := testcontainers.ContainerRequest{
		Image:        "postgres:16-alpine",
		ExposedPorts: []string{"5432/tcp"},
		Env: map[string]string{
			"POSTGRES_USER":     "test",
			"POSTGRES_PASSWORD": "test",
			"POSTGRES_DB":       "testdb",
		},
		WaitingFor: wait.ForLog("database system is ready to accept connections").
			WithOccurrence(2).
			WithStartupTimeout(60 * time.Second),
	}

	container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
		ContainerRequest: req,
		Started:          true,
	})
	if err != nil {
		t.Skipf("Failed to start PostgreSQL container (Docker may not be available): %v", err)
	}

	// 获取容器端口
	host, err := container.Host(ctx)
	if err != nil {
		t.Skipf("Failed to get container host: %v", err)
	}

	port, err := container.MappedPort(ctx, "5432")
	if err != nil {
		t.Skipf("Failed to get container port: %v", err)
	}

	// 构建 DSN
	dsn := fmt.Sprintf("host=%s port=%s user=test password=test dbname=testdb sslmode=disable",
		host, port.Port())

	// 创建服务