package store_test

import (
	"errors"
	"testing"
	"time"

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

func TestGetOrderAndAttemptByRef(t *testing.T) {
	s := store.NewOrderStore(model.OpenTestDB(t))
	seedOrder(t, s, "PAY-Q1") // 复用 order_test.go 的 seedOrder(建 order + attempt provider_ref="P-PAY-Q1")

	o, err := s.GetOrder("PAY-Q1")
	if err != nil || o.OutTradeNo != "PAY-Q1" {
		t.Fatalf("GetOrder = %+v, %v", o, err)
	}
	if _, err := s.GetOrder("NOPE"); !errors.Is(err, store.ErrOrderNotFound) {
		t.Fatalf("缺单应 ErrOrderNotFound, got %v", err)
	}

	att, err := s.AttemptByProviderRef("P-PAY-Q1")
	if err != nil || att.OutTradeNo != "PAY-Q1" || att.Channel != "crypto" {
		t.Fatalf("AttemptByProviderRef = %+v, %v", att, err)
	}
	if _, err := s.AttemptByProviderRef("P-UNKNOWN"); !errors.Is(err, store.ErrAttemptNotFound) {
		t.Fatalf("缺尝试应 ErrAttemptNotFound, got %v", err)
	}
}

func TestListPendingAndExpire(t *testing.T) {
	s := store.NewOrderStore(model.OpenTestDB(t))
	seedOrder(t, s, "PAY-Q2")

	pending, err := s.ListAttemptsByStatus(model.AttemptPending, 10)
	if err != nil || len(pending) != 1 {
		t.Fatalf("ListAttemptsByStatus pending = %d, %v", len(pending), err)
	}

	n, err := s.ExpirePendingAttempts("PAY-Q2")
	if err != nil || n != 1 {
		t.Fatalf("ExpirePendingAttempts = %d, %v", n, err)
	}
	pending2, _ := s.ListAttemptsByStatus(model.AttemptPending, 10)
	if len(pending2) != 0 {
		t.Fatalf("弃过期后应无 pending, got %d", len(pending2))
	}
	_ = time.Now
}

func TestExpireStaleOrders(t *testing.T) {
	db := model.OpenTestDB(t)
	s := store.NewOrderStore(db)
	now := time.Date(2026, 7, 10, 12, 0, 0, 0, time.UTC)

	// 三张单:旧 pending(应过期)、新 pending(未到点)、旧 paid(不动)。
	mk := func(no string, st model.OrderStatusV2, createdAgo time.Duration) {
		o := &model.OrderV2{OutTradeNo: no, AmountMinor: 100, Currency: "USDT", Status: st}
		if err := s.CreateOrder(o); err != nil {
			t.Fatalf("create %s: %v", no, err)
		}
		// 直接回填 created_at 制造账龄(GORM 默认写 now)。
		if err := db.Model(&model.OrderV2{}).Where("out_trade_no = ?", no).
			Update("created_at", now.Add(-createdAgo)).Error; err != nil {
			t.Fatalf("backdate %s: %v", no, err)
		}
	}
	mk("STALE", model.OrderPendingV2, 2*time.Hour)   // 旧 pending
	mk("FRESH", model.OrderPendingV2, 5*time.Minute) // 新 pending
	mk("PAID", model.OrderPaidV2, 2*time.Hour)       // 旧 paid

	cutoff := now.Add(-time.Hour) // TTL=1h
	n, err := s.ExpireStaleOrders(cutoff, 500)
	if err != nil {
		t.Fatalf("expire: %v", err)
	}
	if n != 1 {
		t.Fatalf("应关闭 1 张(仅 STALE), got %d", n)
	}
	assertStatus := func(no string, want model.OrderStatusV2) {
		o, _ := s.GetOrder(no)
		if o.Status != want {
			t.Fatalf("%s status = %v, want %v", no, o.Status, want)
		}
	}
	assertStatus("STALE", model.OrderExpiredV2)
	assertStatus("FRESH", model.OrderPendingV2)
	assertStatus("PAID", model.OrderPaidV2)

	// 幂等:再跑一次不再关闭(STALE 已非 pending)。
	if n2, _ := s.ExpireStaleOrders(cutoff, 500); n2 != 0 {
		t.Fatalf("重跑应 0, got %d", n2)
	}
}