package test

import (
	"os"
	"testing"

	"yinli-api/pkg/config"

	"github.com/stretchr/testify/assert"
)

// TestLoadConfig 测试配置加载
func TestLoadConfig(t *testing.T) {
	tests := []struct {
		name        string
		env         string
		expectError bool
	}{
		{
			name:        "加载开发环境配置",
			env:         "dev",
			expectError: false,
		},
		{
			name:        "加载预发布环境配置",
			env:         "stage",
			expectError: false,
		},
		{
			name:        "加载生产环境配置",
			env:         "prod",
			expectError: false,
		},
		{
			name:        "加载不存在的配置",
			env:         "nonexistent",
			expectError: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			cfg, err := config.LoadConfig(tt.env)

			if tt.expectError {
				assert.Error(t, err)
				assert.Nil(t, cfg)
			} else {
				assert.NoError(t, err)
				assert.NotNil(t, cfg)

				// 验证配置结构
				assert.NotEmpty(t, cfg.Server.Port)
				assert.NotEmpty(t, cfg.Server.Mode)
				assert.NotEmpty(t, cfg.Database.Host)
				assert.NotEmpty(t, cfg.Database.Username)
				assert.NotEmpty(t, cfg.Database.DBName)
				assert.NotEmpty(t, cfg.Redis.Host)
				assert.NotEmpty(t, cfg.JWT.Secret)
			}
		})
	}
}

// TestGetDSN 测试数据库连接字符串生成
func TestGetDSN(t *testing.T) {
	cfg, err := config.LoadConfig("dev")
	assert.NoError(t, err)
	assert.NotNil(t, cfg)

	dsn := cfg.GetDSN()
	assert.NotEmpty(t, dsn)
	assert.Contains(t, dsn, cfg.Database.Username)
	assert.Contains(t, dsn, cfg.Database.Host)
	assert.Contains(t, dsn, cfg.Database.DBName)
}

// TestGetRedisAddr 测试Redis连接地址生成
func TestGetRedisAddr(t *testing.T) {
	cfg, err := config.LoadConfig("dev")
	assert.NoError(t, err)
	assert.NotNil(t, cfg)

	addr := cfg.GetRedisAddr()
	assert.NotEmpty(t, addr)
	assert.Contains(t, addr, cfg.Redis.Host)
}

// TestGetEnv 测试环境变量获取
func TestGetEnv(t *testing.T) {
	// 设置测试环境变量
	os.Setenv("TEST_VAR", "test_value")
	defer os.Unsetenv("TEST_VAR")

	tests := []struct {
		name         string
		key          string
		defaultValue string
		expected     string
	}{
		{
			name:         "存在的环境变量",
			key:          "TEST_VAR",
			defaultValue: "default",
			expected:     "test_value",
		},
		{
			name:         "不存在的环境变量",
			key:          "NON_EXISTENT_VAR",
			defaultValue: "default",
			expected:     "default",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := config.GetEnv(tt.key, tt.defaultValue)
			assert.Equal(t, tt.expected, result)
		})
	}
}

// TestConfigValidation 测试配置验证
func TestConfigValidation(t *testing.T) {
	cfg, err := config.LoadConfig("dev")
	assert.NoError(t, err)
	assert.NotNil(t, cfg)

	// 验证服务器配置
	assert.True(t, cfg.Server.Port != "")
	assert.Contains(t, []string{"debug", "release"}, cfg.Server.Mode)

	// 验证数据库配置
	assert.True(t, cfg.Database.Port > 0)
	assert.True(t, cfg.Database.MaxIdleConns > 0)
	assert.True(t, cfg.Database.MaxOpenConns > 0)

	// 验证Redis配置
	assert.True(t, cfg.Redis.Port > 0)
	assert.True(t, cfg.Redis.PoolSize > 0)

	// 验证JWT配置
	assert.True(t, cfg.JWT.ExpireHours > 0)
	assert.NotEmpty(t, cfg.JWT.Secret)

	// 验证频率限制配置
	if cfg.RateLimit.Enabled {
		assert.True(t, cfg.RateLimit.Requests > 0)
		assert.True(t, cfg.RateLimit.Window > 0)
	}

	// 验证CORS配置
	assert.True(t, len(cfg.CORS.AllowOrigins) > 0)
	assert.True(t, len(cfg.CORS.AllowMethods) > 0)
	assert.True(t, len(cfg.CORS.AllowHeaders) > 0)
}