package codes_test import ( "context" "testing" "time" "github.com/wangjia/pangolin/server/internal/codes" ) func TestListBatchesAndVoidBatchOnLibTables(t *testing.T) { ctx := context.Background() db := openMigratedSQLite(t) store := codes.NewStore(db) svc := codes.NewService(store, nil, 5, time.Hour) res, err := svc.CreateBatch(ctx, codes.BatchRequest{ PlanCode: codes.PlanPro, DurationDays: 30, Count: 3, Channel: codes.ChannelManual, Note: "n", CreatedBy: "admin:1", }) if err != nil { t.Fatalf("mint: %v", err) } infos, total, err := store.ListBatches(ctx, 50, 0) if err != nil { t.Fatalf("ListBatches: %v", err) } if total != 1 || len(infos) != 1 { t.Fatalf("total=%d len=%d", total, len(infos)) } bi := infos[0] if bi.ID != res.BatchID || bi.Channel != codes.ChannelManual || bi.CreatedBy != "admin:1" || bi.Note != "n" || bi.Total != 3 || bi.Unused != 3 || bi.Redeemed != 0 || bi.Void != 0 { t.Fatalf("BatchInfo = %+v", bi) } n, err := store.VoidBatch(ctx, res.BatchID) if err != nil || n != 3 { t.Fatalf("VoidBatch = %d, %v", n, err) } infos, _, err = store.ListBatches(ctx, 50, 0) if err != nil || infos[0].Void != 3 || infos[0].Unused != 0 { t.Fatalf("after void: %+v (err=%v)", infos[0], err) } // 二次 void:0 行(只动 unused)。 if n, err := store.VoidBatch(ctx, res.BatchID); err != nil || n != 0 { t.Fatalf("re-void = %d, %v", n, err) } }