hfb_sys/backend/internal/modules/payment/repository_logic_test.go

package payment

import (
	"errors"
	"testing"

	"hfb_sys/backend/internal/model"
)

// TestPaymentOrderStates 测试支付单状态
func TestPaymentOrderStates(t *testing.T) {
	states := []string{"pending", "paying", "paid", "failed", "closed", "refunding", "refunded"}

	for _, state := range states {
		if state == "" {
			t.Fatal("payment order state should not be empty")
		}
	}
}

// TestRefundBizTypesAreValid 测试退款业务类型有效性
func TestRefundBizTypesAreValid(t *testing.T) {
	validTypes := map[string]bool{
		"cancel_refund":      true,
		"admin_close_refund": true,
		"admin_refund":       true,
		"checkout_refund":    true,
		"deposit_refund":     true,
		"rent_refund":        true,
		"arbitration_refund": true,
	}

	for _, bizType := range refundBizTypes {
		if !validTypes[bizType] {
			t.Fatalf("unexpected refund biz type: %s", bizType)
		}
	}

	if len(refundBizTypes) != len(validTypes) {
		t.Fatalf("refundBizTypes count = %d, want %d", len(refundBizTypes), len(validTypes))
	}
}

// TestCanReuseOrderPaymentLogic 测试支付单复用逻辑
func TestCanReuseOrderPaymentWithSameMerchant(t *testing.T) {
	payment := model.PaymentOrder{
		Status:     "paying",
		Provider:   "lakala",
		MerchantID: "M123",
	}

	config := runtimePaymentConfig{
		Provider:   "lakala",
		MerchantID: "M123",
	}

	if !canReuseOrderPayment(payment, config) {
		t.Fatal("should reuse payment with same merchant and paying status")
	}
}

func TestCanReuseOrderPaymentWithDifferentMerchant(t *testing.T) {
	payment := model.PaymentOrder{
		Status:     "paying",
		Provider:   "lakala",
		MerchantID: "M123",
	}

	config := runtimePaymentConfig{
		Provider:   "lakala",
		MerchantID: "M456", // 不同商户
	}

	if canReuseOrderPayment(payment, config) {
		t.Fatal("should not reuse payment with different merchant")
	}
}

func TestCanReuseOrderPaymentWithTerminalStatus(t *testing.T) {
	terminalStatuses := []string{"failed", "closed"}

	config := runtimePaymentConfig{
		Provider:   "lakala",
		MerchantID: "M123",
	}

	for _, status := range terminalStatuses {
		payment := model.PaymentOrder{
			Status:     status,
			Provider:   "lakala",
			MerchantID: "M123",
		}

		if canReuseOrderPayment(payment, config) {
			t.Fatalf("should not reuse payment with terminal status: %s", status)
		}
	}
}

func TestCanReuseOrderPaymentWithPaidStatus(t *testing.T) {
	payment := model.PaymentOrder{
		Status:     "paid",
		Provider:   "lakala",
		MerchantID: "M123",
	}

	config := runtimePaymentConfig{
		Provider:   "lakala",
		MerchantID: "M123",
	}

	// 已支付的订单可以复用（幂等）
	if !canReuseOrderPayment(payment, config) {
		t.Fatal("should reuse payment with paid status for idempotency")
	}
}

// TestGeneratePaymentNo 测试支付单号生成
func TestPaymentNoShouldHavePrefix(t *testing.T) {
	// 支付单号应该以特定前缀开头（如 PAY）
	// 这是一个示例测试，实际格式需要根据代码确认
	paymentNo := "PAY20260610123456"

	if len(paymentNo) < 3 {
		t.Fatal("paymentNo should have meaningful length")
	}
}

func TestPaymentNoUniquenessAssumption(t *testing.T) {
	// 支付单号应该是唯一的
	// 实际实现中通常使用时间戳+随机数保证唯一性
	seen := make(map[string]bool)

	// 模拟多个支付单号
	paymentNos := []string{
		"PAY20260610123456001",
		"PAY20260610123456002",
		"PAY20260610123456003",
	}

	for _, no := range paymentNos {
		if seen[no] {
			t.Fatalf("duplicate paymentNo: %s", no)
		}
		seen[no] = true
	}
}

// TestPaymentDTOValidation 测试支付 DTO 基本结构
func TestPaymentDTOHasRequiredFields(t *testing.T) {
	dto := PaymentDTO{
		ID:         1,
		PaymentNo:  "PAY123456",
		OrderNo:    "ORD123456",
		AmountCent: 10000,
		Status:     "paid",
	}

	if dto.ID == 0 {
		t.Fatal("PaymentDTO.ID should not be zero")
	}
	if dto.PaymentNo == "" {
		t.Fatal("PaymentDTO.PaymentNo should not be empty")
	}
	if dto.AmountCent <= 0 {
		t.Fatal("PaymentDTO.AmountCent should be positive")
	}
	if dto.Status == "" {
		t.Fatal("PaymentDTO.Status should not be empty")
	}
}

// TestRefundDTOValidation 测试退款 DTO 基本结构
func TestRefundDTOHasRequiredFields(t *testing.T) {
	dto := RefundDTO{
		OrderID:    100,
		AmountCent: 5000,
		Status:     "refunding",
		BizType:    "cancel_refund",
	}

	if dto.OrderID == 0 {
		t.Fatal("RefundDTO.OrderID should not be zero")
	}
	if dto.AmountCent <= 0 {
		t.Fatal("RefundDTO.AmountCent should be positive")
	}
	if dto.Status == "" {
		t.Fatal("RefundDTO.Status should not be empty")
	}
	if dto.BizType == "" {
		t.Fatal("RefundDTO.BizType should not be empty")
	}
}

// TestChannelSourceConstants 测试渠道来源常量
func TestChannelSourceConstantsAreUnique(t *testing.T) {
	sources := []string{
		channelSourceCreate,
		channelSourceQuery,
		channelSourceNotify,
		channelSourceMock,
	}

	seen := make(map[string]bool)
	for _, source := range sources {
		if seen[source] {
			t.Fatalf("duplicate channel source: %s", source)
		}
		seen[source] = true
	}

	if len(seen) != 4 {
		t.Fatalf("expected 4 unique channel sources, got %d", len(seen))
	}
}

// TestRuntimePaymentConfigValidation 测试运行时支付配置
func TestRuntimePaymentConfigRequiredFields(t *testing.T) {
	config := runtimePaymentConfig{
		ID:         1,
		Provider:   "lakala",
		MerchantID: "M123",
		PayWay:     "ZFBZF",
	}

	if config.ID == 0 {
		t.Fatal("config.ID should not be zero")
	}
	if config.Provider == "" {
		t.Fatal("config.Provider should not be empty")
	}
	if config.MerchantID == "" {
		t.Fatal("config.MerchantID should not be empty")
	}
}

// TestServiceInputValidation 测试 Service 输入验证
func TestServiceStartRequiresNonZeroOrderID(t *testing.T) {
	svc := &Service{repo: &Repository{}}

	_, err := svc.Start(t.Context(), 1, 0, StartPaymentRequest{}, "127.0.0.1")
	if !errors.Is(err, ErrPaymentCannotStart) {
		t.Fatalf("error = %v, want ErrPaymentCannotStart", err)
	}
}

func TestServiceStartRequiresNonZeroUserID(t *testing.T) {
	svc := &Service{repo: &Repository{}}

	_, err := svc.Start(t.Context(), 0, 100, StartPaymentRequest{}, "127.0.0.1")
	if !errors.Is(err, ErrPaymentCannotStart) {
		t.Fatalf("error = %v, want ErrPaymentCannotStart", err)
	}
}

func TestServiceStartRefundRequiresPositiveAmount(t *testing.T) {
	svc := &Service{repo: &Repository{}}

	_, err := svc.StartRefund(t.Context(), 100, 0, "cancel_refund", "test")
	if !errors.Is(err, ErrRefundCannotStart) {
		t.Fatalf("error = %v, want ErrRefundCannotStart", err)
	}

	_, err = svc.StartRefund(t.Context(), 100, -1000, "cancel_refund", "test")
	if !errors.Is(err, ErrRefundCannotStart) {
		t.Fatalf("error = %v, want ErrRefundCannotStart", err)
	}
}

func TestServiceStartRefundRequiresNonZeroOrderID(t *testing.T) {
	svc := &Service{repo: &Repository{}}

	_, err := svc.StartRefund(t.Context(), 0, 1000, "cancel_refund", "test")
	if !errors.Is(err, ErrRefundCannotStart) {
		t.Fatalf("error = %v, want ErrRefundCannotStart", err)
	}
}

// TestPaymentErrorTypes 测试错误类型定义
func TestPaymentErrorsAreDefined(t *testing.T) {
	errors := []error{
		ErrDependencyUnavailable,
		ErrPaymentUnavailable,
		ErrPaymentCannotStart,
		ErrPaymentVerifyFailed,
		ErrPaymentNotFound,
		ErrRefundCannotStart,
	}

	for i, err := range errors {
		if err == nil {
			t.Fatalf("error[%d] should not be nil", i)
		}
		if err.Error() == "" {
			t.Fatalf("error[%d] should have message", i)
		}
	}
}