junhong_cmp_fiber/tests/integration/error_code_validation_test.go

package integration

import (
	"encoding/json"
	"fmt"
	"testing"
	"time"

	"github.com/break/junhong_cmp_fiber/pkg/errors"
	"github.com/break/junhong_cmp_fiber/tests/testutils/integ"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestErrorCodeValidation_PackageNotFound(t *testing.T) {
	env := integ.NewIntegrationTestEnv(t)

	t.Run("套餐不存在返回404", func(t *testing.T) {
		resp, err := env.AsSuperAdmin().Request("GET", "/api/admin/packages/99999", nil)
		require.NoError(t, err)
		defer resp.Body.Close()

		var result map[string]interface{}
		err = json.NewDecoder(resp.Body).Decode(&result)
		require.NoError(t, err)

		// 验证 HTTP 状态码
		assert.Equal(t, 404, resp.StatusCode, "应返回 404 Not Found")

		// 验证错误码
		code, ok := result["code"].(float64)
		require.True(t, ok, "响应应包含 code 字段")
		assert.Equal(t, float64(errors.CodeNotFound), code, "应返回 CodeNotFound")
	})
}

func TestErrorCodeValidation_InsufficientBalance(t *testing.T) {
	env := integ.NewIntegrationTestEnv(t)

	t.Run("余额不足返回400", func(t *testing.T) {
		// 创建测试店铺和提现申请
		// 这里需要先创建一个店铺，然后申请提现金额 > 余额
		// 由于涉及较多前置步骤，这里仅验证错误码映射正确性

		// 假设有一个提现接口，提现金额大于余额
		body := []byte(`{"amount": 1000000000}`) // 10亿分，肯定超出余额
		resp, err := env.AsSuperAdmin().Request("POST", "/api/admin/commission_withdrawals", body)

		// 如果接口不存在或需要特定条件，跳过此测试
		if err != nil || resp.StatusCode == 404 {
			t.Skip("提现接口需要特定前置条件，跳过测试")
			return
		}
		defer resp.Body.Close()

		// 如果成功请求，验证错误码
		if resp.StatusCode != 200 {
			var result map[string]interface{}
			err = json.NewDecoder(resp.Body).Decode(&result)
			require.NoError(t, err)

			code, ok := result["code"].(float64)
			if ok && code == float64(errors.CodeInsufficientBalance) {
				assert.Equal(t, 400, resp.StatusCode, "余额不足应返回 400")
			}
		}
	})
}

func TestErrorCodeValidation_ShopCodeDuplicate(t *testing.T) {
	env := integ.NewIntegrationTestEnv(t)

	t.Run("店铺代码重复返回409", func(t *testing.T) {
		// 创建第一个店铺
		shopCode := fmt.Sprintf("TEST_SHOP_%d", time.Now().UnixNano())
		body1 := fmt.Sprintf(`{
			"shop_name": "测试店铺1",
			"shop_code": "%s",
			"level": 1,
			"contact_name": "联系人1",
			"contact_phone": "13800138001",
			"status": 1
		}`, shopCode)

		resp1, err := env.AsSuperAdmin().Request("POST", "/api/admin/shops", []byte(body1))
		require.NoError(t, err)
		defer resp1.Body.Close()

		if resp1.StatusCode != 200 {
			t.Skipf("创建店铺失败，状态码: %d", resp1.StatusCode)
			return
		}

		// 尝试创建重复店铺代码
		body2 := fmt.Sprintf(`{
			"shop_name": "测试店铺2",
			"shop_code": "%s",
			"level": 1,
			"contact_name": "联系人2",
			"contact_phone": "13800138002",
			"status": 1
		}`, shopCode)

		resp2, err := env.AsSuperAdmin().Request("POST", "/api/admin/shops", []byte(body2))
		require.NoError(t, err)
		defer resp2.Body.Close()

		var result map[string]interface{}
		err = json.NewDecoder(resp2.Body).Decode(&result)
		require.NoError(t, err)

		// 验证 HTTP 状态码
		assert.Equal(t, 409, resp2.StatusCode, "重复店铺代码应返回 409 Conflict")

		// 验证错误码
		code, ok := result["code"].(float64)
		require.True(t, ok, "响应应包含 code 字段")
		assert.Equal(t, float64(errors.CodeShopCodeExists), code, "应返回 CodeShopCodeExists")
	})
}

func TestErrorCodeValidation_LogLevels(t *testing.T) {
	t.Run("验证日志级别配置", func(t *testing.T) {
		// 4xx 错误应该是 WARN 级别
		// 5xx 错误应该是 ERROR 级别
		// 这个在 pkg/errors/handler.go 中已经实现

		// 验证错误码的 HTTP 状态码映射
		testCases := []struct {
			code           int
			expectedStatus int
			expectedLevel  string
		}{
			{errors.CodeNotFound, 404, "WARN"},
			{errors.CodeInvalidParam, 400, "WARN"},
			{errors.CodeShopCodeExists, 409, "WARN"},
			{errors.CodeInsufficientBalance, 400, "WARN"},
			{errors.CodeInternalError, 500, "ERROR"},
		}

		for _, tc := range testCases {
			httpStatus := errors.GetHTTPStatus(tc.code)
			assert.Equal(t, tc.expectedStatus, httpStatus,
				"错误码 %d 应映射到 HTTP %d", tc.code, tc.expectedStatus)

			// 验证日志级别（4xx -> WARN, 5xx -> ERROR）
			expectedLevel := "WARN"
			if httpStatus >= 500 {
				expectedLevel = "ERROR"
			}
			assert.Equal(t, expectedLevel, tc.expectedLevel,
				"HTTP %d 应使用 %s 级别日志", httpStatus, expectedLevel)
		}
	})
}