mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-16 07:33:06 +08:00
Ryan ChouandClaude Opus 5
32ad1dcdce
订阅首窗初始化在开通日零点时，automaticWindowStartAt 会把实际推进锚点修正为 StartsAt，但展示层（calculateProgress / WeeklyResetTime / MonthlyResetTime） 仍按窗口起点+周期计算，导致仪表盘显示的重置时间早于实际重置时间（差值 = StartsAt 的时分秒）。用户在显示的重置时间过后请求仍会收到 429 WEEKLY_LIMIT_EXCEEDED。 抽出 windowResetAnchor 供推进与展示共用，保证两者恒一致。非 legacy 锚点 （手动重置或已推进过的窗口）保持权威，不做修正。 日窗口不在本 PR 范围内：#5380 已把日额度改为按日历日对齐（DailyResetTime = 窗口起点所在日的次日 0 点），与周/月的期限对齐滚动窗口语义不同，不共用本修正。 附一个守护用例钉死这条边界。 Fixes #5348 Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
108 lines
4.3 KiB
Go
108 lines
4.3 KiB
Go
package service
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/pkg/timezone"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// 周窗口初始化在开通日零点（legacy anchor）时，automaticWindowStartAt 会把实际
|
|
// 推进锚点修正为 StartsAt。展示层的 WeeklyResetTime 必须应用同一修正，
|
|
// 否则用户看到的重置时间会比实际重置时间早（差值 = StartsAt 的时分秒）。
|
|
func TestWeeklyResetTime_LegacyMidnightAnchor_UsesStartsAt(t *testing.T) {
|
|
startsAt := time.Date(2026, 7, 31, 13, 37, 6, 0, time.FixedZone("UTC+8", 8*3600))
|
|
windowStart := startOfDay(startsAt)
|
|
|
|
sub := &UserSubscription{
|
|
StartsAt: startsAt,
|
|
ExpiresAt: startsAt.AddDate(0, 0, 30),
|
|
WeeklyWindowStart: ptrTime(windowStart),
|
|
}
|
|
|
|
got := sub.WeeklyResetTime()
|
|
require.NotNil(t, got)
|
|
assert.True(t, startsAt.Add(7*24*time.Hour).Equal(*got),
|
|
"legacy 午夜锚点应按 StartsAt+7d 计算重置时间，而不是窗口起点+7d")
|
|
}
|
|
|
|
// 非 legacy 锚点（手动重置或已推进过的窗口）保持权威，不做修正。
|
|
func TestWeeklyResetTime_RegularAnchor_UsesWindowStart(t *testing.T) {
|
|
startsAt := time.Date(2026, 7, 31, 13, 37, 6, 0, time.FixedZone("UTC+8", 8*3600))
|
|
windowStart := startsAt.Add(7 * 24 * time.Hour) // 已推进过一轮，非开通日零点
|
|
|
|
sub := &UserSubscription{
|
|
StartsAt: startsAt,
|
|
ExpiresAt: startsAt.AddDate(0, 0, 30),
|
|
WeeklyWindowStart: ptrTime(windowStart),
|
|
}
|
|
|
|
got := sub.WeeklyResetTime()
|
|
require.NotNil(t, got)
|
|
assert.True(t, windowStart.Add(7*24*time.Hour).Equal(*got),
|
|
"非 legacy 锚点应按窗口起点+7d 计算重置时间")
|
|
}
|
|
|
|
// 展示与执行一致性：WeeklyResetTime 前一秒窗口不应推进，到点后应推进。
|
|
func TestWeeklyResetTime_MatchesAutomaticWindowStart(t *testing.T) {
|
|
startsAt := time.Date(2026, 7, 31, 13, 37, 6, 0, time.FixedZone("UTC+8", 8*3600))
|
|
windowStart := startOfDay(startsAt)
|
|
|
|
sub := &UserSubscription{
|
|
StartsAt: startsAt,
|
|
ExpiresAt: startsAt.AddDate(0, 0, 30),
|
|
WeeklyWindowStart: ptrTime(windowStart),
|
|
}
|
|
|
|
resetAt := *sub.WeeklyResetTime()
|
|
|
|
_, ok := sub.automaticWindowStartAt(sub.WeeklyWindowStart, 7*24*time.Hour, resetAt.Add(-time.Second))
|
|
assert.False(t, ok, "展示的重置时间之前窗口不应可推进")
|
|
|
|
newStart, ok := sub.automaticWindowStartAt(sub.WeeklyWindowStart, 7*24*time.Hour, resetAt)
|
|
assert.True(t, ok, "到达展示的重置时间后窗口应可推进")
|
|
assert.True(t, resetAt.Equal(newStart), "推进后的新窗口起点应等于展示的重置时间")
|
|
}
|
|
|
|
// 月窗口与周窗口同属期限对齐滚动窗口，应用同一 legacy 锚点修正。
|
|
// 日窗口按日历日对齐（见 automaticDailyWindowStartAt），不走此修正，故不在此覆盖。
|
|
func TestMonthlyResetTime_LegacyMidnightAnchor_UsesStartsAt(t *testing.T) {
|
|
startsAt := time.Date(2026, 7, 31, 13, 37, 6, 0, time.FixedZone("UTC+8", 8*3600))
|
|
windowStart := startOfDay(startsAt)
|
|
|
|
sub := &UserSubscription{
|
|
StartsAt: startsAt,
|
|
ExpiresAt: startsAt.AddDate(0, 0, 60),
|
|
MonthlyWindowStart: ptrTime(windowStart),
|
|
}
|
|
|
|
monthly := sub.MonthlyResetTime()
|
|
require.NotNil(t, monthly)
|
|
assert.True(t, startsAt.Add(30*24*time.Hour).Equal(*monthly),
|
|
"legacy 午夜锚点应按 StartsAt+30d 计算重置时间，而不是窗口起点+30d")
|
|
}
|
|
|
|
// 日窗口不受本修正影响：DailyResetTime 保持 #5380 的日历日对齐语义。
|
|
// 基准取配置时区的 0 点（与 subscription_daily_midnight_reset_test.go 同构），
|
|
// 保证断言在任意本地时区下都成立。
|
|
func TestDailyResetTime_UnaffectedByWindowResetAnchor(t *testing.T) {
|
|
base := timezone.StartOfDay(time.Date(2026, 7, 31, 12, 0, 0, 0, timezone.Location()))
|
|
startsAt := base.Add(13*time.Hour + 37*time.Minute + 6*time.Second)
|
|
windowStart := base // legacy 锚点：开通日 0 点
|
|
|
|
sub := &UserSubscription{
|
|
StartsAt: startsAt,
|
|
ExpiresAt: startsAt.AddDate(0, 0, 30),
|
|
DailyWindowStart: ptrTime(windowStart),
|
|
}
|
|
|
|
got := sub.DailyResetTime()
|
|
require.NotNil(t, got)
|
|
assert.True(t, got.Equal(base.AddDate(0, 0, 1)),
|
|
"日窗口应保持日历日对齐（窗口起点所在日的次日 0 点）")
|
|
assert.False(t, got.Equal(startsAt.Add(24*time.Hour)),
|
|
"日窗口不应被 windowResetAnchor 修正成 StartsAt+24h（那是周/月的语义）")
|
|
}
|