File size: 3,232 Bytes
6d60378 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 | // checkin_status_test.go 钉住「今日已签到」观测：NoteCheckinDone 标记当日，
// statusOf 输出 CheckinDone（跨零点自然过期），持久化往返不丢。
package pool
import (
"os"
"path/filepath"
"strings"
"testing"
"time"
"github.com/linguo2625469/workbuddy2api-panel/internal/auth"
)
func TestNoteCheckinDoneMarksToday(t *testing.T) {
p := New("")
p.Add(&auth.Auth{UID: "u1"})
p.Add(&auth.Auth{UID: "u2"})
if st, _ := p.Status("u1"); st.CheckinDone {
t.Fatal("未签到账号不应显示已签")
}
p.NoteCheckinDone("u1")
st, ok := p.Status("u1")
if !ok || !st.CheckinDone {
t.Fatalf("u1 标记后应显示已签: %+v ok=%v", st, ok)
}
if st, _ := p.Status("u2"); st.CheckinDone {
t.Fatal("u2 未标记，不应显示已签")
}
// 未知 uid 不 panic、不影响其他账号。
p.NoteCheckinDone("no-such-uid")
}
func TestCheckinDonePersists(t *testing.T) {
dir := t.TempDir()
fp := filepath.Join(dir, "state.json")
p := New(fp)
p.Add(&auth.Auth{UID: "u1"})
p.NoteCheckinDone("u1")
p.Flush()
raw, err := os.ReadFile(fp)
if err != nil {
t.Fatal(err)
}
if !strings.Contains(string(raw), `"last_checkin_day"`) {
t.Fatalf("state.json 缺少 last_checkin_day:\n%s", raw)
}
p2 := New(fp)
p2.Add(&auth.Auth{UID: "u1"})
if st, _ := p2.Status("u1"); !st.CheckinDone {
t.Fatal("重启后当日已签状态丢失")
}
}
func TestCheckinDoneExpiredYesterday(t *testing.T) {
dir := t.TempDir()
fp := filepath.Join(dir, "state.json")
p := New(fp)
p.Add(&auth.Auth{UID: "u1"})
p.NoteCheckinDone("u1")
p.Flush()
raw, err := os.ReadFile(fp)
if err != nil {
t.Fatal(err)
}
yesterday := time.Now().AddDate(0, 0, -1).Format("2006-01-02")
raw = []byte(strings.Replace(string(raw), time.Now().Format("2006-01-02"), yesterday, 1))
if err := os.WriteFile(fp, raw, 0o600); err != nil {
t.Fatal(err)
}
p2 := New(fp)
p2.Add(&auth.Auth{UID: "u1"})
if st, _ := p2.Status("u1"); st.CheckinDone {
t.Fatal("昨日签到记录不应显示为今日已签")
}
}
// TestSetNicknamePersists（issue #94）：昵称更新写入 auths 文件并往返无损；
// 未变化不写盘；空昵称/未知 uid 拒绝。
func TestSetNicknamePersists(t *testing.T) {
dir := t.TempDir()
fp := filepath.Join(dir, "workbuddy-u1.json")
a := &auth.Auth{UID: "u1", Nickname: "旧名字", AccessToken: "at", FilePath: fp}
p := New("")
p.Add(a)
if !p.SetNickname("u1", "新名字") {
t.Fatal("昵称变化应返回 true")
}
reloaded, err := auth.Parse(mustRead(t, fp))
if err != nil || reloaded.Nickname != "新名字" {
t.Fatalf("回写后昵称=%q err=%v, want 新名字", reloaded.Nickname, err)
}
if p.SetNickname("u1", "新名字") {
t.Fatal("未变化不应再写盘")
}
if p.SetNickname("u1", "") || p.SetNickname("no-such", "x") {
t.Fatal("空昵称/未知 uid 应拒绝")
}
st, _ := p.Status("u1")
if st.Nickname != "新名字" {
t.Fatalf("Status 昵称=%q, want 新名字", st.Nickname)
}
}
func mustRead(t *testing.T, path string) []byte {
t.Helper()
raw, err := os.ReadFile(path)
if err != nil {
t.Fatal(err)
}
return raw
}
|