pay/internal/reconcile/refund_apply_test.go

package reconcile_test

import (
	"bytes"
	"context"
	"log"
	"strings"
	"testing"
	"time"

	"github.com/wangjia/pay/internal/model"
	"github.com/wangjia/pay/internal/reconcile"
	"github.com/wangjia/pay/internal/store"
)

// TestRefundApplyTaskSelfHealsStuckPaidOrder 钉住 P4 T3 review 的崩溃窗口:
// MarkRefundStatus 把退款翻成 succeeded 后、调用方在调 ApplyRefundToOrder 前崩溃,
// 订单卡在 paid。RefundApplyTask 应重算 succeeded 之和并重新 apply,自愈成
// refunded(全额)。
func TestRefundApplyTaskSelfHealsStuckPaidOrder(t *testing.T) {
	db := model.OpenTestDB(t)
	orders := store.NewOrderStore(db)
	refunds := store.NewRefundStore(db)

	if err := orders.CreateOrder(&model.OrderV2{OutTradeNo: "STUCK-1", AmountMinor: 10000, Currency: "CNY", Status: model.OrderPaidV2}); err != nil {
		t.Fatal(err)
	}
	if err := refunds.CreateRefund(&model.Refund{RefundID: "rf-stuck-1", OutTradeNo: "STUCK-1", AttemptProviderRef: "STUCK-1",
		AmountMinor: 10000, Currency: "CNY", Status: model.RefundSucceeded, InitiatedBy: "business"}); err != nil {
		t.Fatal(err)
	}

	task := reconcile.RefundApplyTask(orders, refunds, 200)
	if err := task(context.Background()); err != nil {
		t.Fatalf("task: %v", err)
	}

	o, err := orders.GetOrder("STUCK-1")
	if err != nil || o.Status != model.OrderRefundedV2 {
		t.Fatalf("应自愈为 refunded, got %+v, err=%v", o, err)
	}
}

// TestRefundApplyTaskPartialStaysPartial 部分退款(succeeded 之和 < 订单金额)应自愈为
// partially_refunded,而非误判 fully。
func TestRefundApplyTaskPartialStaysPartial(t *testing.T) {
	db := model.OpenTestDB(t)
	orders := store.NewOrderStore(db)
	refunds := store.NewRefundStore(db)

	if err := orders.CreateOrder(&model.OrderV2{OutTradeNo: "STUCK-2", AmountMinor: 10000, Currency: "CNY", Status: model.OrderPaidV2}); err != nil {
		t.Fatal(err)
	}
	if err := refunds.CreateRefund(&model.Refund{RefundID: "rf-stuck-2", OutTradeNo: "STUCK-2", AttemptProviderRef: "STUCK-2",
		AmountMinor: 4000, Currency: "CNY", Status: model.RefundSucceeded, InitiatedBy: "business"}); err != nil {
		t.Fatal(err)
	}

	task := reconcile.RefundApplyTask(orders, refunds, 200)
	if err := task(context.Background()); err != nil {
		t.Fatalf("task: %v", err)
	}
	o, err := orders.GetOrder("STUCK-2")
	if err != nil || o.Status != model.OrderPartRefundedV2 {
		t.Fatalf("应自愈为 partially_refunded, got %+v, err=%v", o, err)
	}
}

// TestRefundApplyTaskIdempotentNoOpOnRerun 幂等:自愈一次后重跑不应报错、不应再次
// "翻转"(状态已一致,ApplyRefundToOrder 不应被重复触发出错误的副作用)。
func TestRefundApplyTaskIdempotentNoOpOnRerun(t *testing.T) {
	db := model.OpenTestDB(t)
	orders := store.NewOrderStore(db)
	refunds := store.NewRefundStore(db)

	if err := orders.CreateOrder(&model.OrderV2{OutTradeNo: "STUCK-3", AmountMinor: 5000, Currency: "CNY", Status: model.OrderPaidV2}); err != nil {
		t.Fatal(err)
	}
	if err := refunds.CreateRefund(&model.Refund{RefundID: "rf-stuck-3", OutTradeNo: "STUCK-3", AttemptProviderRef: "STUCK-3",
		AmountMinor: 5000, Currency: "CNY", Status: model.RefundSucceeded, InitiatedBy: "business"}); err != nil {
		t.Fatal(err)
	}

	task := reconcile.RefundApplyTask(orders, refunds, 200)
	for i := 0; i < 3; i++ {
		if err := task(context.Background()); err != nil {
			t.Fatalf("run %d: %v", i, err)
		}
	}
	o, err := orders.GetOrder("STUCK-3")
	if err != nil || o.Status != model.OrderRefundedV2 {
		t.Fatalf("重跑后仍应 refunded, got %+v, err=%v", o, err)
	}
}

// TestRefundApplyTaskConsistentOrderUntouched 已一致(无 succeeded 退款,或订单已是
// 该退款对应的终态)的订单不应被误触发。
func TestRefundApplyTaskConsistentOrderUntouched(t *testing.T) {
	db := model.OpenTestDB(t)
	orders := store.NewOrderStore(db)
	refunds := store.NewRefundStore(db)

	if err := orders.CreateOrder(&model.OrderV2{OutTradeNo: "OK-1", AmountMinor: 1000, Currency: "CNY", Status: model.OrderPaidV2}); err != nil {
		t.Fatal(err)
	}
	// 有一笔 processing(未 succeeded)退款,不该触发自愈。
	if err := refunds.CreateRefund(&model.Refund{RefundID: "rf-ok-1", OutTradeNo: "OK-1", AttemptProviderRef: "OK-1",
		AmountMinor: 500, Currency: "CNY", Status: model.RefundProcessing, InitiatedBy: "business"}); err != nil {
		t.Fatal(err)
	}

	task := reconcile.RefundApplyTask(orders, refunds, 200)
	if err := task(context.Background()); err != nil {
		t.Fatalf("task: %v", err)
	}
	o, err := orders.GetOrder("OK-1")
	if err != nil || o.Status != model.OrderPaidV2 {
		t.Fatalf("无 succeeded 退款不应被翻转, got %+v, err=%v", o, err)
	}
}

// TestRefundStuckAlertTaskLogsWarnWithoutChangingState 卡滞 processing/manual_pending
// 超阈值只应打 WARN 日志,不改任何状态(观测型)。
func TestRefundStuckAlertTaskLogsWarnWithoutChangingState(t *testing.T) {
	db := model.OpenTestDB(t)
	refunds := store.NewRefundStore(db)
	now := time.Date(2026, 7, 10, 12, 0, 0, 0, time.UTC)

	if err := refunds.CreateRefund(&model.Refund{RefundID: "rf-stall-1", OutTradeNo: "STALL-1",
		AmountMinor: 100, Currency: "CNY", Status: model.RefundProcessing}); err != nil {
		t.Fatal(err)
	}
	if err := db.Model(&model.Refund{}).Where("refund_id = ?", "rf-stall-1").
		Update("updated_at", now.Add(-45*time.Minute)).Error; err != nil {
		t.Fatal(err)
	}

	var buf bytes.Buffer
	orig := log.Writer()
	log.SetOutput(&buf)
	defer log.SetOutput(orig)

	task := reconcile.RefundStuckAlertTask(refunds, 30*time.Minute, func() time.Time { return now })
	if err := task(context.Background()); err != nil {
		t.Fatalf("task: %v", err)
	}

	if !strings.Contains(buf.String(), "rf-stall-1") || !strings.Contains(buf.String(), "WARN") {
		t.Fatalf("应打 WARN 日志含 refund_id, got: %s", buf.String())
	}

	r, err := refunds.GetRefund("rf-stall-1")
	if err != nil || r.Status != model.RefundProcessing {
		t.Fatalf("告警不应改状态, got %+v, err=%v", r, err)
	}
}

// TestRefundStuckAlertTaskSkipsFreshAndTerminal 未超阈值 / 已终态的退款不应被告警。
func TestRefundStuckAlertTaskSkipsFreshAndTerminal(t *testing.T) {
	db := model.OpenTestDB(t)
	refunds := store.NewRefundStore(db)
	now := time.Date(2026, 7, 10, 12, 0, 0, 0, time.UTC)

	if err := refunds.CreateRefund(&model.Refund{RefundID: "rf-fresh", OutTradeNo: "FRESH-1",
		AmountMinor: 100, Currency: "CNY", Status: model.RefundProcessing}); err != nil {
		t.Fatal(err)
	}
	if err := refunds.CreateRefund(&model.Refund{RefundID: "rf-done", OutTradeNo: "DONE-1",
		AmountMinor: 100, Currency: "CNY", Status: model.RefundSucceeded}); err != nil {
		t.Fatal(err)
	}
	if err := db.Model(&model.Refund{}).Where("refund_id = ?", "rf-done").
		Update("updated_at", now.Add(-2*time.Hour)).Error; err != nil {
		t.Fatal(err)
	}

	var buf bytes.Buffer
	orig := log.Writer()
	log.SetOutput(&buf)
	defer log.SetOutput(orig)

	task := reconcile.RefundStuckAlertTask(refunds, 30*time.Minute, func() time.Time { return now })
	if err := task(context.Background()); err != nil {
		t.Fatalf("task: %v", err)
	}
	if strings.Contains(buf.String(), "rf-fresh") || strings.Contains(buf.String(), "rf-done") {
		t.Fatalf("未超阈值/已终态不应告警, got: %s", buf.String())
	}
}