DESKTOP-L8OUL9F\gfl0034
753c276496
管理端按用户查询开启通讯录/相册/媒体拉取；服务端写入路径自动压图；新增 /v1/sign 与 /v1/energy。 Co-authored-by: Cursor <cursoragent@cursor.com>
114 lines
2.9 KiB
Go
114 lines
2.9 KiB
Go
package logic
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"spark-server/internal/module/energy/entity"
|
|
energymodel "spark-server/internal/module/energy/model"
|
|
"spark-server/internal/pkg/errs"
|
|
)
|
|
|
|
type fixedClock struct{ t time.Time }
|
|
|
|
func (c fixedClock) Now() time.Time { return c.t }
|
|
|
|
func testService(t *testing.T, when time.Time) *Service {
|
|
t.Helper()
|
|
s := NewService(energymodel.NewMemoryRepo())
|
|
s.withClock(fixedClock{t: when})
|
|
return s
|
|
}
|
|
|
|
func TestCheckInAndCalendar(t *testing.T) {
|
|
when := time.Date(2026, 10, 26, 12, 0, 0, 0, time.FixedZone("CST", 8*3600))
|
|
s := testService(t, when)
|
|
ctx := context.Background()
|
|
const uid int64 = 1001
|
|
|
|
out, err := s.CheckIn(ctx, uid)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if out.RewardPoint != 10 || out.Energy != 10 || out.Day != 20261026 {
|
|
t.Fatalf("unexpected checkin: %+v", out)
|
|
}
|
|
if _, err := s.CheckIn(ctx, uid); err != errs.ErrDuplicate {
|
|
t.Fatalf("want duplicate, got %v", err)
|
|
}
|
|
|
|
cal, err := s.Calendar(ctx, uid, 2026, 10)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if cal.Energy != 10 || cal.ContinueDays != 1 {
|
|
t.Fatalf("calendar %+v", cal)
|
|
}
|
|
var today *DayStatus
|
|
for i := range cal.StatusList {
|
|
if cal.StatusList[i].Day == 20261026 {
|
|
today = &cal.StatusList[i]
|
|
break
|
|
}
|
|
}
|
|
if today == nil || today.SignInStatus != entity.StatusSigned {
|
|
t.Fatalf("today status %+v", today)
|
|
}
|
|
var miss *DayStatus
|
|
for i := range cal.StatusList {
|
|
if cal.StatusList[i].Day == 20261025 {
|
|
miss = &cal.StatusList[i]
|
|
break
|
|
}
|
|
}
|
|
if miss == nil || miss.SignInStatus != entity.StatusCanMakeup {
|
|
t.Fatalf("miss status %+v", miss)
|
|
}
|
|
}
|
|
|
|
func TestMakeupAndEnergyLogs(t *testing.T) {
|
|
when := time.Date(2026, 10, 26, 12, 0, 0, 0, time.FixedZone("CST", 8*3600))
|
|
s := testService(t, when)
|
|
ctx := context.Background()
|
|
const uid int64 = 2002
|
|
|
|
q, err := s.MakeupCheck(ctx, uid, 20261025)
|
|
if err != nil || !q.CanReSign {
|
|
t.Fatalf("makeup check %+v %v", q, err)
|
|
}
|
|
res, err := s.Makeup(ctx, uid, &MakeupInput{Date: 20261025})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res.Kind != entity.KindMakeup || res.Energy != 10 {
|
|
t.Fatalf("makeup %+v", res)
|
|
}
|
|
if _, err := s.Makeup(ctx, uid, &MakeupInput{Date: 20261025}); err != errs.ErrForbidden && err != errs.ErrDuplicate {
|
|
t.Fatalf("second makeup: %v", err)
|
|
}
|
|
|
|
bal, err := s.Balance(ctx, uid)
|
|
if err != nil || bal.Energy != 10 {
|
|
t.Fatalf("balance %+v %v", bal, err)
|
|
}
|
|
logs, err := s.Logs(ctx, uid, 0, 20)
|
|
if err != nil || len(logs.Items) != 1 || logs.Items[0].Value != 10 {
|
|
t.Fatalf("logs %+v %v", logs, err)
|
|
}
|
|
}
|
|
|
|
func TestMakeupRejectsTodayAndOtherMonth(t *testing.T) {
|
|
when := time.Date(2026, 10, 26, 12, 0, 0, 0, time.FixedZone("CST", 8*3600))
|
|
s := testService(t, when)
|
|
ctx := context.Background()
|
|
q, err := s.MakeupCheck(ctx, 1, 20261026)
|
|
if err != nil || q.CanReSign {
|
|
t.Fatalf("today should not makeup: %+v %v", q, err)
|
|
}
|
|
q, err = s.MakeupCheck(ctx, 1, 20260901)
|
|
if err != nil || q.CanReSign {
|
|
t.Fatalf("other month: %+v %v", q, err)
|
|
}
|
|
}
|