junhong_cmp_fiber/tests/integration/health_test.go

package integration

import (
	"context"
	"net/http/httptest"
	"testing"

	"github.com/break/junhong_cmp_fiber/internal/handler"
	"github.com/gofiber/fiber/v2"
	"github.com/redis/go-redis/v9"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"go.uber.org/zap"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
)

// TestHealthCheckNormal 测试健康检查 - 正常状态
func TestHealthCheckNormal(t *testing.T) {
	// 初始化日志
	logger, _ := zap.NewDevelopment()

	// 初始化内存数据库
	tx, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
	require.NoError(t, err)

	// 初始化 Redis 客户端（使用本地 Redis）
	rdb := redis.NewClient(&redis.Options{
		Addr: "localhost:6379",
		DB:   0,
	})
	defer func() { _ = rdb.Close() }()

	// 创建 Fiber 应用
	app := fiber.New()

	// 创建健康检查处理器
	healthHandler := handler.NewHealthHandler(tx, rdb, logger)
	app.Get("/health", healthHandler.Check)

	// 发送测试请求
	req := httptest.NewRequest("GET", "/health", nil)
	resp, err := app.Test(req)
	require.NoError(t, err)
	defer resp.Body.Close()

	// 验证响应状态码
	assert.Equal(t, 200, resp.StatusCode)

	// 验证响应内容
	// 注意：这里可以进一步解析 JSON 响应体验证详细信息
}

// TestHealthCheckDatabaseDown 测试健康检查 - 数据库异常
func TestHealthCheckDatabaseDown(t *testing.T) {
	t.Skip("需要模拟数据库连接失败的场景")

	// 初始化日志
	logger, _ := zap.NewDevelopment()

	// 初始化一个会失败的数据库连接
	tx, err := gorm.Open(sqlite.Open("/invalid/path/test.tx"), &gorm.Config{})
	if err != nil {
		// 预期会失败
		t.Log("数据库连接失败（预期行为）")
	}

	// 初始化 Redis 客户端
	rdb := redis.NewClient(&redis.Options{
		Addr: "localhost:6379",
		DB:   0,
	})
	defer func() { _ = rdb.Close() }()

	// 创建 Fiber 应用
	app := fiber.New()

	// 创建健康检查处理器
	healthHandler := handler.NewHealthHandler(tx, rdb, logger)
	app.Get("/health", healthHandler.Check)

	// 发送测试请求
	req := httptest.NewRequest("GET", "/health", nil)
	resp, err := app.Test(req)
	require.NoError(t, err)
	defer resp.Body.Close()

	// 验证响应状态码应该是 503 (Service Unavailable)
	assert.Equal(t, 503, resp.StatusCode)
}

// TestHealthCheckRedisDown 测试健康检查 - Redis 异常
func TestHealthCheckRedisDown(t *testing.T) {
	// 初始化日志
	logger, _ := zap.NewDevelopment()

	// 初始化内存数据库
	tx, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
	require.NoError(t, err)

	// 初始化一个连接到无效地址的 Redis 客户端
	rdb := redis.NewClient(&redis.Options{
		Addr: "localhost:9999", // 无效端口
		DB:   0,
	})
	defer func() { _ = rdb.Close() }()

	// 创建 Fiber 应用
	app := fiber.New()

	// 创建健康检查处理器
	healthHandler := handler.NewHealthHandler(tx, rdb, logger)
	app.Get("/health", healthHandler.Check)

	// 发送测试请求
	req := httptest.NewRequest("GET", "/health", nil)
	resp, err := app.Test(req)
	require.NoError(t, err)
	defer resp.Body.Close()

	// 验证响应状态码应该是 503 (Service Unavailable)
	assert.Equal(t, 503, resp.StatusCode)
}

// TestHealthCheckDetailed 测试健康检查 - 验证详细信息
func TestHealthCheckDetailed(t *testing.T) {
	// 初始化日志
	logger, _ := zap.NewDevelopment()

	// 初始化内存数据库
	tx, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
	require.NoError(t, err)

	// 初始化 Redis 客户端
	rdb := redis.NewClient(&redis.Options{
		Addr: "localhost:6379",
		DB:   0,
	})
	defer func() { _ = rdb.Close() }()

	// 测试 Redis 连接
	ctx := context.Background()
	_, err = rdb.Ping(ctx).Result()
	if err != nil {
		t.Skip("Redis 未运行，跳过测试")
	}

	// 创建 Fiber 应用
	app := fiber.New()

	// 创建健康检查处理器
	healthHandler := handler.NewHealthHandler(tx, rdb, logger)
	app.Get("/health", healthHandler.Check)

	// 发送测试请求
	req := httptest.NewRequest("GET", "/health", nil)
	resp, err := app.Test(req)
	require.NoError(t, err)
	defer resp.Body.Close()

	// 验证响应状态码
	assert.Equal(t, 200, resp.StatusCode)

	// TODO: 解析 JSON 响应并验证包含以下字段：
	// - status: "healthy"
	// - postgres: "up"
	// - redis: "up"
	// - timestamp
}