This commit is contained in:
75
logic/controller/pet_manage_test.go
Normal file
75
logic/controller/pet_manage_test.go
Normal file
@@ -0,0 +1,75 @@
|
||||
package controller
|
||||
|
||||
import (
|
||||
"blazing/common/data/xmlres"
|
||||
"blazing/common/socket/errorcode"
|
||||
"blazing/logic/service/player"
|
||||
playermodel "blazing/modules/player/model"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func firstPetIDForControllerTest(t *testing.T) int {
|
||||
t.Helper()
|
||||
|
||||
for id := range xmlres.PetMAP {
|
||||
return id
|
||||
}
|
||||
|
||||
t.Fatal("xmlres.PetMAP is empty")
|
||||
return 0
|
||||
}
|
||||
|
||||
func TestSetPetExpCapsLevelAt100(t *testing.T) {
|
||||
petID := firstPetIDForControllerTest(t)
|
||||
petInfo := playermodel.GenPetInfo(petID, 31, 0, 0, 99, nil, 0)
|
||||
if petInfo == nil {
|
||||
t.Fatal("failed to generate test pet")
|
||||
}
|
||||
|
||||
expPool := petInfo.NextLvExp + 10_000
|
||||
testPlayer := player.NewPlayer(nil)
|
||||
testPlayer.Info = &playermodel.PlayerInfo{
|
||||
ExpPool: expPool,
|
||||
PetList: []playermodel.PetInfo{*petInfo},
|
||||
}
|
||||
|
||||
currentPet := &testPlayer.Info.PetList[0]
|
||||
result, err := (Controller{}).SetPetExp(&PetSetExpInboundInfo{
|
||||
CatchTime: currentPet.CatchTime,
|
||||
Exp: expPool,
|
||||
}, testPlayer)
|
||||
if err != 0 {
|
||||
t.Fatalf("expected SetPetExp to succeed, got err=%d", err)
|
||||
}
|
||||
if currentPet.Level != 100 {
|
||||
t.Fatalf("expected pet level to stop at 100, got %d", currentPet.Level)
|
||||
}
|
||||
if result.Exp != 10_000 {
|
||||
t.Fatalf("expected overflow exp to remain in pool, got %d", result.Exp)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSetPetExpRejectsPetAtLevel100(t *testing.T) {
|
||||
petID := firstPetIDForControllerTest(t)
|
||||
petInfo := playermodel.GenPetInfo(petID, 31, 0, 0, 100, nil, 0)
|
||||
if petInfo == nil {
|
||||
t.Fatal("failed to generate test pet")
|
||||
}
|
||||
|
||||
testPlayer := player.NewPlayer(nil)
|
||||
testPlayer.Info = &playermodel.PlayerInfo{
|
||||
ExpPool: 50_000,
|
||||
PetList: []playermodel.PetInfo{*petInfo},
|
||||
}
|
||||
|
||||
result, err := (Controller{}).SetPetExp(&PetSetExpInboundInfo{
|
||||
CatchTime: petInfo.CatchTime,
|
||||
Exp: 12_345,
|
||||
}, testPlayer)
|
||||
if err != errorcode.ErrorCodes.ErrSystemError {
|
||||
t.Fatalf("expected level-100 pet to be rejected, got err=%d", err)
|
||||
}
|
||||
if result.Exp != 50_000 {
|
||||
t.Fatalf("expected exp pool to remain unchanged, got %d", result.Exp)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user