修复(#16): 修复订单退款状态与后台恢复冲突 (#16)

Co-authored-by: multica-agent <github@multica.ai>

queue/logic/order/order_status_recovery_test.go

@@ -0,0 +1,78 @@
+package orderLogic
+
+import (
+	"context"
+	"fmt"
+	"strings"
+	"testing"
+
+	"github.com/DATA-DOG/go-sqlmock"
+	"github.com/perfect-panel/server/internal/svc"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+)
+
+func TestReleaseClaimOnlyReleasesClaimedStatus(t *testing.T) {
+	const orderNo = "ORD-CLAIMED-1"
+
+	db, mock, cleanup := newOrderQueueTestDB(t)
+	defer cleanup()
+
+	mock.ExpectBegin()
+	mock.ExpectExec("UPDATE `order`").
+		WithArgs(OrderStatusPaid, sqlmock.AnyArg(), orderNo, OrderStatusClaimed).
+		WillReturnResult(sqlmock.NewResult(0, 1))
+	mock.ExpectCommit()
+
+	logic := NewActivateOrderLogic(&svc.ServiceContext{DB: db})
+	if err := logic.releaseClaim(context.Background(), orderNo); err != nil {
+		t.Fatalf("releaseClaim error: %v", err)
+	}
+	assertOrderQueueExpectations(t, mock)
+}
+
+func TestStuckOrderRecoveryScansOnlyClaimedStatus(t *testing.T) {
+	db, mock, cleanup := newOrderQueueTestDB(t)
+	defer cleanup()
+
+	mock.ExpectQuery("FROM `order`").
+		WithArgs(OrderStatusClaimed, sqlmock.AnyArg()).
+		WillReturnRows(sqlmock.NewRows([]string{"id", "order_no", "status"}))
+
+	logic := NewStuckOrderRecoveryLogic(&svc.ServiceContext{DB: db})
+	if err := logic.ProcessTask(context.Background(), nil); err != nil {
+		t.Fatalf("ProcessTask error: %v", err)
+	}
+	assertOrderQueueExpectations(t, mock)
+}
+
+func newOrderQueueTestDB(t *testing.T) (*gorm.DB, sqlmock.Sqlmock, func()) {
+	t.Helper()
+
+	sqlDB, mock, err := sqlmock.New(sqlmock.QueryMatcherOption(sqlmock.QueryMatcherFunc(func(expectedSQL, actualSQL string) error {
+		if strings.Contains(actualSQL, expectedSQL) {
+			return nil
+		}
+		return fmt.Errorf("actual sql %q does not contain %q", actualSQL, expectedSQL)
+	})))
+	if err != nil {
+		t.Fatalf("create sqlmock: %v", err)
+	}
+
+	db, err := gorm.Open(mysql.New(mysql.Config{Conn: sqlDB, SkipInitializeWithVersion: true}), &gorm.Config{})
+	if err != nil {
+		_ = sqlDB.Close()
+		t.Fatalf("open gorm db: %v", err)
+	}
+
+	return db, mock, func() {
+		_ = sqlDB.Close()
+	}
+}
+
+func assertOrderQueueExpectations(t *testing.T, mock sqlmock.Sqlmock) {
+	t.Helper()
+	if err := mock.ExpectationsWereMet(); err != nil {
+		t.Fatalf("unmet sql expectations: %v", err)
+	}
+}