210 lines
8.4 KiB
Go
210 lines
8.4 KiB
Go
package gamemanagement
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"hyapp-admin-server/internal/integration/activityclient"
|
|
"hyapp-admin-server/internal/integration/gameclient"
|
|
"hyapp-admin-server/internal/integration/userclient"
|
|
"hyapp-admin-server/internal/integration/walletclient"
|
|
"hyapp-admin-server/internal/middleware"
|
|
activityv1 "hyapp.local/api/proto/activity/v1"
|
|
gamev1 "hyapp.local/api/proto/game/v1"
|
|
walletv1 "hyapp.local/api/proto/wallet/v1"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type fakeRobotProfileSource struct {
|
|
profiles []RobotProfile
|
|
requested int
|
|
}
|
|
|
|
func (s *fakeRobotProfileSource) RandomRobotProfiles(_ context.Context, count int) ([]RobotProfile, error) {
|
|
s.requested = count
|
|
return s.profiles[:count], nil
|
|
}
|
|
|
|
type fakeRobotUserClient struct {
|
|
userclient.Client
|
|
created []userclient.QuickCreateAccountRequest
|
|
}
|
|
|
|
func (c *fakeRobotUserClient) QuickCreateAccount(_ context.Context, req userclient.QuickCreateAccountRequest) (*userclient.QuickCreateAccountResult, error) {
|
|
c.created = append(c.created, req)
|
|
return &userclient.QuickCreateAccountResult{UserID: int64(9000 + len(c.created)), DisplayUserID: "robot"}, nil
|
|
}
|
|
|
|
func (c *fakeRobotUserClient) BatchGetUsers(_ context.Context, _ userclient.BatchGetUsersRequest) (map[int64]*userclient.User, error) {
|
|
users := make(map[int64]*userclient.User, len(c.created))
|
|
for index, created := range c.created {
|
|
userID := int64(9001 + index)
|
|
users[userID] = &userclient.User{UserID: userID, DisplayUserID: "robot", Username: created.Username, Avatar: created.Avatar}
|
|
}
|
|
return users, nil
|
|
}
|
|
|
|
type fakeRobotGameClient struct {
|
|
gameclient.Client
|
|
registered []int64
|
|
}
|
|
|
|
func (c *fakeRobotGameClient) RegisterDiceRobots(_ context.Context, req *gamev1.RegisterDiceRobotsRequest) (*gamev1.RegisterDiceRobotsResponse, error) {
|
|
c.registered = append(c.registered, req.GetUserIds()...)
|
|
robots := make([]*gamev1.DiceRobot, 0, len(req.GetUserIds()))
|
|
for _, userID := range req.GetUserIds() {
|
|
robots = append(robots, &gamev1.DiceRobot{AppCode: "lalu", GameId: "dice", UserId: userID, Status: "active"})
|
|
}
|
|
return &gamev1.RegisterDiceRobotsResponse{Robots: robots}, nil
|
|
}
|
|
|
|
type fakeRobotWalletClient struct {
|
|
walletclient.Client
|
|
grants []string
|
|
equips []int64
|
|
}
|
|
|
|
func (c *fakeRobotWalletClient) ListResources(_ context.Context, req *walletv1.ListResourcesRequest) (*walletv1.ListResourcesResponse, error) {
|
|
switch req.GetResourceType() {
|
|
case "avatar_frame":
|
|
return &walletv1.ListResourcesResponse{Resources: []*walletv1.Resource{{ResourceId: 101, ResourceType: "avatar_frame", Status: "active", ManagerGrantEnabled: true}}}, nil
|
|
case "vehicle":
|
|
return &walletv1.ListResourcesResponse{Resources: []*walletv1.Resource{{ResourceId: 201, ResourceType: "vehicle", Status: "active", ManagerGrantEnabled: true}}}, nil
|
|
case "badge":
|
|
return &walletv1.ListResourcesResponse{Resources: []*walletv1.Resource{
|
|
{ResourceId: 301, ResourceType: "badge", Status: "active", ManagerGrantEnabled: true, MetadataJson: `{"badge_form":"strip"}`},
|
|
{ResourceId: 302, ResourceType: "badge", Status: "active", ManagerGrantEnabled: true, MetadataJson: `{"badge_form":"tile"}`},
|
|
}}, nil
|
|
default:
|
|
return &walletv1.ListResourcesResponse{}, nil
|
|
}
|
|
}
|
|
|
|
func (c *fakeRobotWalletClient) GrantResource(_ context.Context, req *walletv1.GrantResourceRequest) (*walletv1.ResourceGrantResponse, error) {
|
|
c.grants = append(c.grants, req.GetCommandId())
|
|
return &walletv1.ResourceGrantResponse{Grant: &walletv1.ResourceGrant{Items: []*walletv1.ResourceGrantItem{{
|
|
ResourceId: req.GetResourceId(),
|
|
EntitlementId: "entitlement",
|
|
}}}}, nil
|
|
}
|
|
|
|
func (c *fakeRobotWalletClient) EquipUserResource(_ context.Context, req *walletv1.EquipUserResourceRequest) (*walletv1.EquipUserResourceResponse, error) {
|
|
c.equips = append(c.equips, req.GetResourceId())
|
|
return &walletv1.EquipUserResourceResponse{}, nil
|
|
}
|
|
|
|
type fakeRobotActivityClient struct {
|
|
activityclient.Client
|
|
levels []*activityv1.SetUserLevelRequest
|
|
}
|
|
|
|
func (c *fakeRobotActivityClient) SetUserLevel(_ context.Context, req *activityv1.SetUserLevelRequest) (*activityv1.SetUserLevelResponse, error) {
|
|
c.levels = append(c.levels, req)
|
|
return &activityv1.SetUserLevelResponse{Status: "consumed", Track: req.GetTrack(), NewLevel: req.GetLevel()}, nil
|
|
}
|
|
|
|
func TestGenerateDiceRobotsUsesLikeiProfiles(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
profiles := &fakeRobotProfileSource{profiles: []RobotProfile{
|
|
{Nickname: "likei-one", Avatar: "https://media.haiyihy.com/avatar/one.jpg"},
|
|
{Nickname: "likei-two", Avatar: "https://media.haiyihy.com/avatar/two.jpg"},
|
|
}}
|
|
users := &fakeRobotUserClient{}
|
|
games := &fakeRobotGameClient{}
|
|
wallets := &fakeRobotWalletClient{}
|
|
activities := &fakeRobotActivityClient{}
|
|
handler := New(games, users, nil, WithRobotProfileSource(profiles), WithRobotAppearanceServices(wallets, activities))
|
|
|
|
recorder := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(recorder)
|
|
c.Set(middleware.ContextRequestID, "req-robot-test")
|
|
c.Set(middleware.ContextUserID, uint(7))
|
|
c.Request = httptest.NewRequest(http.MethodPost, "/api/v1/admin/games/self/dice/robots:generate", bytes.NewBufferString(`{
|
|
"count": 2,
|
|
"nicknameLanguage": "ar",
|
|
"avatarUrls": ["https://randomuser.me/api/portraits/men/1.jpg"],
|
|
"country": "SA"
|
|
}`))
|
|
c.Request.Header.Set("Content-Type", "application/json")
|
|
|
|
handler.GenerateDiceRobots(c)
|
|
|
|
if recorder.Code != http.StatusOK {
|
|
t.Fatalf("status = %d, body = %s", recorder.Code, recorder.Body.String())
|
|
}
|
|
if profiles.requested != 2 {
|
|
t.Fatalf("profile source requested %d profiles, want 2", profiles.requested)
|
|
}
|
|
if len(users.created) != 2 {
|
|
t.Fatalf("created users = %d, want 2", len(users.created))
|
|
}
|
|
if users.created[0].Username != "likei-one" || users.created[0].Avatar != "https://media.haiyihy.com/avatar/one.jpg" {
|
|
t.Fatalf("first robot profile = %#v", users.created[0])
|
|
}
|
|
if users.created[1].Username != "likei-two" || users.created[1].Avatar != "https://media.haiyihy.com/avatar/two.jpg" {
|
|
t.Fatalf("second robot profile = %#v", users.created[1])
|
|
}
|
|
if users.created[0].Language != "ar" || users.created[0].Source != "game_robot" || users.created[0].Country != "SA" {
|
|
t.Fatalf("robot account metadata = %#v", users.created[0])
|
|
}
|
|
if len(games.registered) != 2 || games.registered[0] != 9001 || games.registered[1] != 9002 {
|
|
t.Fatalf("registered user ids = %#v", games.registered)
|
|
}
|
|
if len(wallets.grants) != 8 || len(wallets.equips) != 4 || len(activities.levels) != 4 {
|
|
t.Fatalf("robot appearance init mismatch: grants=%d equips=%d levels=%d", len(wallets.grants), len(wallets.equips), len(activities.levels))
|
|
}
|
|
for _, level := range activities.levels {
|
|
if level.GetLevel() < robotMinDisplayLevel || level.GetLevel() > robotMaxDisplayLevel {
|
|
t.Fatalf("robot level out of range: %+v", level)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestGenerateDiceRobotsFallsBackToRandomProfilesWhenLikeiMissing(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
users := &fakeRobotUserClient{}
|
|
games := &fakeRobotGameClient{}
|
|
handler := New(games, users, nil, WithRobotAppearanceServices(&fakeRobotWalletClient{}, &fakeRobotActivityClient{}))
|
|
|
|
recorder := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(recorder)
|
|
c.Set(middleware.ContextRequestID, "req-robot-random")
|
|
c.Set(middleware.ContextUserID, uint(7))
|
|
c.Request = httptest.NewRequest(http.MethodPost, "/api/v1/admin/games/self/dice/robots:generate", bytes.NewBufferString(`{
|
|
"count": 3,
|
|
"country": "TH"
|
|
}`))
|
|
c.Request.Header.Set("Content-Type", "application/json")
|
|
|
|
handler.GenerateDiceRobots(c)
|
|
|
|
if recorder.Code != http.StatusOK {
|
|
t.Fatalf("status = %d, body = %s", recorder.Code, recorder.Body.String())
|
|
}
|
|
if len(users.created) != 3 {
|
|
t.Fatalf("created users = %d, want 3", len(users.created))
|
|
}
|
|
for index, created := range users.created {
|
|
if !strings.HasPrefix(created.Username, "Robot ") {
|
|
t.Fatalf("created[%d] username should be generated, got %#v", index, created)
|
|
}
|
|
if !strings.HasPrefix(created.Avatar, "https://api.dicebear.com/9.x/adventurer/png?seed=hyapp-robot-") {
|
|
t.Fatalf("created[%d] avatar should be generated, got %#v", index, created)
|
|
}
|
|
if created.Gender != "male" && created.Gender != "female" {
|
|
t.Fatalf("created[%d] gender should be random male/female, got %#v", index, created)
|
|
}
|
|
if created.Country != "TH" || created.Source != "game_robot" {
|
|
t.Fatalf("created[%d] metadata = %#v", index, created)
|
|
}
|
|
}
|
|
if len(games.registered) != 3 || games.registered[0] != 9001 || games.registered[2] != 9003 {
|
|
t.Fatalf("registered user ids = %#v", games.registered)
|
|
}
|
|
}
|