fix(redemption): add status filtering and cleanup action

2026-07-04 16:15:47 +08:00
parent 86021d8ed2
commit 12603a7765
10 changed files with 272 additions and 130 deletions
@@ -0,0 +1,100 @@
+package model
+
+import (
+	"testing"
+
+	"github.com/QuantumNous/new-api/common"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"gorm.io/gorm"
+)
+
+func TestSearchRedemptionsFiltersAndPaginates(t *testing.T) {
+	require.NoError(t, DB.AutoMigrate(&Redemption{}))
+	require.NoError(t, DB.Session(&gorm.Session{AllowGlobalUpdate: true}).Unscoped().Delete(&Redemption{}).Error)
+	t.Cleanup(func() {
+		require.NoError(t, DB.Session(&gorm.Session{AllowGlobalUpdate: true}).Unscoped().Delete(&Redemption{}).Error)
+	})
+
+	now := common.GetTimestamp()
+	redemptions := []Redemption{
+		{Id: 1, Name: "alpha-active", Key: "00000000000000000000000000000001", Status: common.RedemptionCodeStatusEnabled, ExpiredTime: 0},
+		{Id: 2, Name: "alpha-future", Key: "00000000000000000000000000000002", Status: common.RedemptionCodeStatusEnabled, ExpiredTime: now + 3600},
+		{Id: 3, Name: "alpha-expired", Key: "00000000000000000000000000000003", Status: common.RedemptionCodeStatusEnabled, ExpiredTime: now - 10},
+		{Id: 4, Name: "beta-disabled", Key: "00000000000000000000000000000004", Status: common.RedemptionCodeStatusDisabled, ExpiredTime: 0},
+		{Id: 5, Name: "beta-used", Key: "00000000000000000000000000000005", Status: common.RedemptionCodeStatusUsed, ExpiredTime: 0},
+	}
+	require.NoError(t, DB.Create(&redemptions).Error)
+
+	tests := []struct {
+		name      string
+		keyword   string
+		status    string
+		startIdx  int
+		num       int
+		wantTotal int64
+		wantIds   []int
+	}{
+		{
+			name:      "no filters returns all rows",
+			num:       10,
+			wantTotal: 5,
+			wantIds:   []int{5, 4, 3, 2, 1},
+		},
+		{
+			name:      "keyword filters by name prefix",
+			keyword:   "alpha",
+			num:       10,
+			wantTotal: 3,
+			wantIds:   []int{3, 2, 1},
+		},
+		{
+			name:      "enabled status excludes expired rows",
+			status:    "1",
+			num:       10,
+			wantTotal: 2,
+			wantIds:   []int{2, 1},
+		},
+		{
+			name:      "expired status returns enabled expired rows",
+			status:    "expired",
+			num:       10,
+			wantTotal: 1,
+			wantIds:   []int{3},
+		},
+		{
+			name:      "disabled status",
+			status:    "2",
+			num:       10,
+			wantTotal: 1,
+			wantIds:   []int{4},
+		},
+		{
+			name:      "used status",
+			status:    "3",
+			num:       10,
+			wantTotal: 1,
+			wantIds:   []int{5},
+		},
+		{
+			name:      "pagination keeps unpaged total",
+			startIdx:  1,
+			num:       2,
+			wantTotal: 5,
+			wantIds:   []int{4, 3},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			rows, total, err := SearchRedemptions(tt.keyword, tt.status, tt.startIdx, tt.num)
+			require.NoError(t, err)
+			assert.Equal(t, tt.wantTotal, total)
+			gotIds := make([]int, 0, len(rows))
+			for _, row := range rows {
+				gotIds = append(gotIds, row.Id)
+			}
+			assert.Equal(t, tt.wantIds, gotIds)
+		})
+	}
+}