99 lines
2.9 KiB
Go
99 lines
2.9 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"affiliate_dash/internal/model"
|
|
"affiliate_dash/internal/testdb"
|
|
)
|
|
|
|
func TestAutoPurchaseCooldown_ManualAttemptDoesNotBlock(t *testing.T) {
|
|
now := time.Now()
|
|
manualJustNow := now.Add(-time.Minute)
|
|
autoLongAgo := now.Add(-30 * time.Minute)
|
|
cfg := &model.RockConfig{
|
|
PurchaseCooldownMinutes: 10,
|
|
LastAttemptAt: &manualJustNow, // 刚刚手动采购过
|
|
LastAutoAttemptAt: &autoLongAgo, // 上一次自动尝试已过冷却
|
|
}
|
|
if remain := rockAutoPurchaseCooldownRemaining(cfg, now); remain != 0 {
|
|
t.Fatalf("手动采购不应阻塞自动补货，剩余冷却 %v", remain)
|
|
}
|
|
}
|
|
|
|
func TestAutoPurchaseCooldown_UsesLastAutoAttempt(t *testing.T) {
|
|
now := time.Now()
|
|
recentAuto := now.Add(-2 * time.Minute)
|
|
cfg := &model.RockConfig{
|
|
PurchaseCooldownMinutes: 10,
|
|
LastAutoAttemptAt: &recentAuto,
|
|
}
|
|
remain := rockAutoPurchaseCooldownRemaining(cfg, now)
|
|
if remain < 7*time.Minute || remain > 8*time.Minute {
|
|
t.Fatalf("剩余冷却时间不符: %v", remain)
|
|
}
|
|
}
|
|
|
|
func TestAutoPurchaseCooldown_BlockedAttemptWritesNoLog(t *testing.T) {
|
|
db := testdb.New(t, &model.RockConfig{}, &model.RockPurchaseLog{})
|
|
svc := NewRockService(db, NewRockClient("https://example.com"))
|
|
ctx := context.Background()
|
|
|
|
if _, err := svc.GetConfig(ctx); err != nil {
|
|
t.Fatalf("初始化配置失败: %v", err)
|
|
}
|
|
recent := time.Now().Add(-time.Minute)
|
|
if err := db.Model(&model.RockConfig{}).Where("id = ?", 1).
|
|
Updates(map[string]interface{}{
|
|
"purchase_cooldown_minutes": 10,
|
|
"last_auto_attempt_at": recent,
|
|
}).Error; err != nil {
|
|
t.Fatalf("写入冷却时间失败: %v", err)
|
|
}
|
|
|
|
if _, err := svc.PurchaseTokens(ctx, 10, true); !errors.Is(err, ErrAutoPurchaseCooldown) {
|
|
t.Fatalf("应命中自动采购冷却，实际 err=%v", err)
|
|
}
|
|
|
|
var count int64
|
|
if err := db.Model(&model.RockPurchaseLog{}).Count(&count).Error; err != nil {
|
|
t.Fatalf("统计采购流水失败: %v", err)
|
|
}
|
|
if count != 0 {
|
|
t.Fatalf("冷却拦截不应生成采购流水，实际 %d 条", count)
|
|
}
|
|
}
|
|
|
|
func TestGetStats_ExposesAutoPurchaseCooldown(t *testing.T) {
|
|
db := newRockTestDB(t)
|
|
svc := NewRockService(db, NewRockClient("https://example.com"))
|
|
ctx := context.Background()
|
|
|
|
if _, err := svc.GetConfig(ctx); err != nil {
|
|
t.Fatalf("初始化配置失败: %v", err)
|
|
}
|
|
recent := time.Now().Add(-time.Minute)
|
|
if err := db.Model(&model.RockConfig{}).Where("id = ?", 1).
|
|
Updates(map[string]interface{}{
|
|
"auto_purchase_enabled": true,
|
|
"purchase_cooldown_minutes": 10,
|
|
"last_auto_attempt_at": recent,
|
|
}).Error; err != nil {
|
|
t.Fatalf("写入冷却配置失败: %v", err)
|
|
}
|
|
|
|
stats, err := svc.GetStats(ctx)
|
|
if err != nil {
|
|
t.Fatalf("查询统计失败: %v", err)
|
|
}
|
|
if stats.AutoPurchaseCooldownSeconds <= 0 {
|
|
t.Fatalf("应返回剩余冷却秒数: %+v", stats)
|
|
}
|
|
if stats.NextAutoPurchaseAt == "" {
|
|
t.Fatalf("应返回下次可自动补货时间: %+v", stats)
|
|
}
|
|
}
|