|
|
|
|
| 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 未标记，不应显示已签")
|
| }
|
|
|
| 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("昨日签到记录不应显示为今日已签")
|
| }
|
| }
|
|
|
|
|
|
|
| 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
|
| }
|
|
|
