82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
package httpapi
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.118im.com/im_team/im_workbench/internal/store"
|
|
)
|
|
|
|
func TestPayloadUserIDsAreStrings(t *testing.T) {
|
|
now := time.Date(2026, 9, 16, 9, 30, 0, 0, time.UTC)
|
|
userA, err := parseUserID("usrag7w9xJ6hto")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
userB, err := parseUserID("usr1RqTFgHSenM")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
payloads := []map[string]any{
|
|
organizationMemberPayload(store.OrganizationMember{UserID: userA}),
|
|
attendanceRecordPayload(&store.AttendanceRecord{RecordID: "att_1", UserID: userA, Date: now, CreatedAt: now, UpdatedAt: now}),
|
|
attendanceAppealPayload(&store.AttendanceAppeal{AppealID: "atta_1", UserID: userA, SubmittedAt: now, CreatedAt: now, UpdatedAt: now}),
|
|
meetingPayload(store.Meeting{
|
|
MeetingID: "mtg_1",
|
|
OrganizerID: userA,
|
|
RecorderID: userB,
|
|
StartAt: now,
|
|
EndAt: now.Add(time.Hour),
|
|
CreatedAt: now,
|
|
UpdatedAt: now,
|
|
Participants: []store.MeetingParticipant{
|
|
{UserID: userA, Response: "accepted"},
|
|
{UserID: userB, Response: "pending"},
|
|
},
|
|
}, userA),
|
|
meetingNoteFilePayload(store.MeetingNoteFile{NoteFileID: "mnote_1", UploaderID: userA, CreatedAt: now, UpdatedAt: now}, userA),
|
|
dailyReportPayload(store.DailyReport{ReportID: "dr_1", AuthorID: userA, ReportDate: now, CreatedAt: now, UpdatedAt: now}),
|
|
monthlyReportPayload(store.MonthlyReport{ReportID: "mr_1", AuthorID: userA, CreatedAt: now, UpdatedAt: now}),
|
|
mailboxSubmissionPayload(store.MailboxSubmission{SubmissionID: "mb_1", SenderID: userA, SubmittedAt: now}),
|
|
}
|
|
|
|
for _, payload := range payloads {
|
|
assertNoNumericUserIDs(t, payload)
|
|
}
|
|
}
|
|
|
|
func assertNoNumericUserIDs(t *testing.T, value any) {
|
|
t.Helper()
|
|
switch typed := value.(type) {
|
|
case map[string]any:
|
|
for key, item := range typed {
|
|
if isUserIDKey(key) {
|
|
if _, ok := item.(string); !ok {
|
|
t.Fatalf("%s is %T, want string", key, item)
|
|
}
|
|
}
|
|
assertNoNumericUserIDs(t, item)
|
|
}
|
|
case []map[string]any:
|
|
for _, item := range typed {
|
|
assertNoNumericUserIDs(t, item)
|
|
}
|
|
case []any:
|
|
for _, item := range typed {
|
|
assertNoNumericUserIDs(t, item)
|
|
}
|
|
}
|
|
}
|
|
|
|
func isUserIDKey(key string) bool {
|
|
return key == "user_id" ||
|
|
key == "organizer_id" ||
|
|
key == "recorder_user_id" ||
|
|
key == "uploader_user_id" ||
|
|
key == "author_id" ||
|
|
key == "sender_id" ||
|
|
strings.HasSuffix(key, "_user_id")
|
|
}
|