94 lines
2.6 KiB
Go
94 lines
2.6 KiB
Go
 
|
package settings
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"path/filepath"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/havline/havline/internal/db"
|
||
|
|
)
|
||
|
|
|
||
|
|
func newTestStore(t *testing.T) *Store {
|
||
|
|
t.Helper()
|
||
|
|
conn, err := db.Open(filepath.Join(t.TempDir(), "havline.db"))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
t.Cleanup(func() { conn.Close() })
|
||
|
|
if err := db.Migrate(context.Background(), conn, filepath.Join("..", "..", "migrations")); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
return NewStore(conn)
|
||
|
|
}
|
||
|
|
|
||
|
|
// TestApplyTimezoneSetsLocal：设置里的时区必须真的作用到 time.Local，
|
||
|
|
// 否则「改了时区但通知时间不变」这个黑洞会一直存在。
|
||
|
|
func TestApplyTimezoneSetsLocal(t *testing.T) {
|
||
|
|
original := time.Local
|
||
|
|
t.Cleanup(func() { time.Local = original })
|
||
|
|
|
||
|
|
store := newTestStore(t)
|
||
|
|
ctx := context.Background()
|
||
|
|
if err := store.Set(ctx, KeyTimezone, "Asia/Tokyo"); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
loc, err := ApplyTimezone(ctx, store)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if loc.String() != "Asia/Tokyo" {
|
||
|
|
t.Fatalf("应返回 Asia/Tokyo，实际 %s", loc)
|
||
|
|
}
|
||
|
|
if time.Local.String() != "Asia/Tokyo" {
|
||
|
|
t.Fatalf("time.Local 应被设为 Asia/Tokyo，实际 %s", time.Local)
|
||
|
|
}
|
||
|
|
// 同一个瞬时在东京比上海早一小时
|
||
|
|
_, offset := time.Now().In(time.Local).Zone()
|
||
|
|
if offset != 9*3600 {
|
||
|
|
t.Fatalf("东京时区偏移应为 +9h，实际 %d 秒", offset)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// TestApplyTimezoneFallsBackToDefault：没写过设置时用注册表里的默认值。
|
||
|
|
func TestApplyTimezoneFallsBackToDefault(t *testing.T) {
|
||
|
|
original := time.Local
|
||
|
|
t.Cleanup(func() { time.Local = original })
|
||
|
|
|
||
|
|
store := newTestStore(t)
|
||
|
|
loc, err := ApplyTimezone(context.Background(), store)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if loc.String() != Defaults[KeyTimezone] {
|
||
|
|
t.Fatalf("应使用默认时区 %q，实际 %s", Defaults[KeyTimezone], loc)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// TestLoadTimezoneRejectsInvalid：非法时区要给出可读的中文报错，
|
||
|
|
// 而不是把 "" 或 UTC 悄悄当成合法值写库。
|
||
|
|
func TestLoadTimezoneRejectsInvalid(t *testing.T) {
|
||
|
|
cases := []string{"", " ", "Mars/Olympus", "Asia/Shanghai; rm -rf /"}
|
||
|
|
for _, input := range cases {
|
||
|
|
if _, err := LoadTimezone(input); err == nil {
|
||
|
|
t.Fatalf("%q 应被拒绝", input)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if _, err := LoadTimezone("UTC"); err != nil {
|
||
|
|
t.Fatalf("UTC 应合法：%v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// TestLoadTimezoneErrorMentionsName：报错里要带上是哪个值不合法，便于前端提示。
|
||
|
|
func TestLoadTimezoneErrorMentionsName(t *testing.T) {
|
||
|
|
_, err := LoadTimezone("Bad/Zone")
|
||
|
|
if err == nil {
|
||
|
|
t.Fatal("非法时区应报错")
|
||
|
|
}
|
||
|
|
if !strings.Contains(err.Error(), "Bad/Zone") {
|
||
|
|
t.Fatalf("报错应包含原始输入，实际：%v", err)
|
||
|
|
}
|
||
|
|
}
|