70 lines
2.3 KiB
Go
70 lines
2.3 KiB
Go
package service
|
||
|
||
import (
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
func at(day, clock string) time.Time {
|
||
t, _ := time.ParseInLocation("2006-01-02 15:04", day+" "+clock, time.Local)
|
||
return t
|
||
}
|
||
|
||
func TestNextAutoUpdateDaily(t *testing.T) {
|
||
// 昨天 09:00 跑过，现在是今天 08:00 → 下一次今天 09:00
|
||
next := NextAutoUpdate("daily", 1, "09:00", at("2026-08-10", "09:00"), at("2026-08-11", "08:00"))
|
||
if !next.Equal(at("2026-08-11", "09:00")) {
|
||
t.Fatalf("want today 09:00, got %v", next)
|
||
}
|
||
// 今天 09:01 已跑，现在 15:00 → 明天 09:00
|
||
next = NextAutoUpdate("daily", 1, "09:00", at("2026-08-11", "09:01"), at("2026-08-11", "15:00"))
|
||
if !next.Equal(at("2026-08-12", "09:00")) {
|
||
t.Fatalf("want tomorrow 09:00, got %v", next)
|
||
}
|
||
// 昨天跑过，现在 10:00（已过 09:00）→ 今天 09:00（补跑）
|
||
next = NextAutoUpdate("daily", 1, "09:00", at("2026-08-10", "09:00"), at("2026-08-11", "10:00"))
|
||
if !next.Equal(at("2026-08-11", "09:00")) {
|
||
t.Fatalf("want catch-up today 09:00, got %v", next)
|
||
}
|
||
}
|
||
|
||
func TestNextAutoUpdateEveryNDays(t *testing.T) {
|
||
next := NextAutoUpdate("everyNDays", 3, "22:30", at("2026-08-08", "22:30"), at("2026-08-11", "08:00"))
|
||
if !next.Equal(at("2026-08-11", "22:30")) {
|
||
t.Fatalf("want 08-11 22:30, got %v", next)
|
||
}
|
||
}
|
||
|
||
func TestNextAutoUpdateEveryNHours(t *testing.T) {
|
||
next := NextAutoUpdate("everyNHours", 4, "", at("2026-08-11", "06:15"), at("2026-08-11", "08:00"))
|
||
if !next.Equal(at("2026-08-11", "10:15")) {
|
||
t.Fatalf("want 10:15, got %v", next)
|
||
}
|
||
}
|
||
|
||
func TestNextAutoUpdateBadClock(t *testing.T) {
|
||
next := NextAutoUpdate("daily", 1, "bogus", at("2026-08-10", "09:00"), at("2026-08-11", "08:00"))
|
||
if !next.Equal(at("2026-08-11", "09:00")) {
|
||
t.Fatalf("bad clock should fall back to 09:00, got %v", next)
|
||
}
|
||
}
|
||
|
||
func TestDigestDue(t *testing.T) {
|
||
if DigestDue("21:00", at("2026-08-11", "20:59")) {
|
||
t.Fatal("20:59 should not be due for 21:00")
|
||
}
|
||
if !DigestDue("21:00", at("2026-08-11", "21:00")) {
|
||
t.Fatal("21:00 sharp should be due")
|
||
}
|
||
if !DigestDue("21:00", at("2026-08-11", "23:40")) {
|
||
t.Fatal("23:40 should be due")
|
||
}
|
||
// 非法时间回退 09:00（与 parseClock 兜底一致）
|
||
if DigestDue("bogus", at("2026-08-11", "08:00")) {
|
||
t.Fatal("bad clock falls back to 09:00, 08:00 not due")
|
||
}
|
||
if !DigestDue("bogus", at("2026-08-11", "09:30")) {
|
||
t.Fatal("bad clock falls back to 09:00, 09:30 due")
|
||
}
|
||
}
|