102 lines
3.1 KiB
Go
102 lines
3.1 KiB
Go
package roomadmin
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"hyapp-admin-server/internal/integration/robotclient"
|
|
)
|
|
|
|
func TestNormalizeRoomListSortKeepsOnlyContributionSort(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
query listQuery
|
|
wantOrder string
|
|
}{
|
|
{
|
|
name: "default",
|
|
query: listQuery{},
|
|
wantOrder: "rle.created_at_ms DESC, rle.room_id DESC",
|
|
},
|
|
{
|
|
name: "contribution_desc",
|
|
query: listQuery{SortBy: "room_contribution", SortDirection: "desc"},
|
|
wantOrder: "rle.heat DESC, rle.created_at_ms DESC, rle.room_id DESC",
|
|
},
|
|
{
|
|
name: "contribution_asc",
|
|
query: listQuery{SortBy: "heat", SortDirection: "asc"},
|
|
wantOrder: "rle.heat ASC, rle.created_at_ms DESC, rle.room_id DESC",
|
|
},
|
|
{
|
|
name: "unknown_falls_back",
|
|
query: listQuery{SortBy: "owner_user_id", SortDirection: "asc"},
|
|
wantOrder: "rle.created_at_ms DESC, rle.room_id DESC",
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
query := normalizeListQuery(test.query)
|
|
if got := roomListOrderBy(query); got != test.wantOrder {
|
|
t.Fatalf("order by mismatch: got %q want %q", got, test.wantOrder)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestRobotRoomProfileFromOwnerUsesOwnerNameAndAvatar(t *testing.T) {
|
|
name, avatar, err := robotRoomProfileFromOwner(RoomOwner{
|
|
Username: " Robot Host ",
|
|
Avatar: " https://cdn.example.com/robot.png ",
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("robot room owner profile should be valid: %v", err)
|
|
}
|
|
if name != "Robot Host" || avatar != "https://cdn.example.com/robot.png" {
|
|
t.Fatalf("owner profile mismatch: name=%q avatar=%q", name, avatar)
|
|
}
|
|
}
|
|
|
|
func TestRobotRoomProfileFromOwnerRequiresNameAndAvatar(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
owner RoomOwner
|
|
}{
|
|
{name: "missing_name", owner: RoomOwner{Avatar: "https://cdn.example.com/robot.png"}},
|
|
{name: "missing_avatar", owner: RoomOwner{Username: "Robot Host"}},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
_, _, err := robotRoomProfileFromOwner(test.owner)
|
|
if !errors.Is(err, ErrInvalidArgument) {
|
|
t.Fatalf("expected ErrInvalidArgument, got %v", err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestAvailableRoomRobotsFromGamePoolKeepsOnlyRoomAvailableUsers(t *testing.T) {
|
|
robots := []robotclient.Robot{
|
|
{UserID: 101, Status: "active", LastUsedAtMS: 1000, UsedCount: 2},
|
|
{UserID: 102, Status: "active", LastUsedAtMS: 2000, UsedCount: 3},
|
|
{UserID: 103, Status: "active", LastUsedAtMS: 3000, UsedCount: 4},
|
|
}
|
|
owners := map[int64]RoomOwner{
|
|
101: {UserID: "101", Username: "Robot A"},
|
|
103: {UserID: "103", Username: "Robot C"},
|
|
}
|
|
|
|
items := availableRoomRobotsFromGamePool(robots, []int64{103, 101}, owners)
|
|
if len(items) != 2 {
|
|
t.Fatalf("available robots length mismatch: got %d want 2", len(items))
|
|
}
|
|
if items[0].UserID != "101" || items[0].User.Username != "Robot A" || items[0].UsedCount != 2 {
|
|
t.Fatalf("first available robot mismatch: %+v", items[0])
|
|
}
|
|
if items[1].UserID != "103" || items[1].User.Username != "Robot C" || items[1].LastUsedAtMS != 3000 {
|
|
t.Fatalf("second available robot mismatch: %+v", items[1])
|
|
}
|
|
}
|