Ilia Denisov
357 lines
8.7 KiB
Go
357 lines
8.7 KiB
Go
package game_test
|
|
|
|
import (
|
|
"iter"
|
|
"math/rand/v2"
|
|
"slices"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
e "github.com/iliadenisov/galaxy/internal/error"
|
|
"github.com/iliadenisov/galaxy/internal/model/game"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestCargoCapacity(t *testing.T) {
|
|
test := func(cargoSize float64, expectCapacity float64) {
|
|
ship := game.ShipType{
|
|
ShipTypeReport: game.ShipTypeReport{
|
|
Drive: 1,
|
|
Armament: 1,
|
|
Weapons: 1,
|
|
Shields: 1,
|
|
Cargo: cargoSize,
|
|
},
|
|
}
|
|
sg := game.ShipGroup{
|
|
Number: 1,
|
|
State: "In_Orbit",
|
|
Drive: 1.5,
|
|
Weapons: 1.1,
|
|
Shields: 2.0,
|
|
Cargo: 1.0,
|
|
}
|
|
assert.Equal(t, expectCapacity, sg.CargoCapacity(&ship))
|
|
}
|
|
test(1, 1.05)
|
|
test(5, 6.25)
|
|
test(10, 15)
|
|
test(50, 175)
|
|
test(100, 600)
|
|
}
|
|
|
|
func TestCarryingAndFullMass(t *testing.T) {
|
|
Freighter := &game.ShipType{
|
|
ShipTypeReport: game.ShipTypeReport{
|
|
Name: "Freighter",
|
|
Drive: 8,
|
|
Armament: 0,
|
|
Weapons: 0,
|
|
Shields: 2,
|
|
Cargo: 10,
|
|
},
|
|
}
|
|
sg := &game.ShipGroup{
|
|
Number: 1,
|
|
State: "In_Orbit",
|
|
Drive: 1.0,
|
|
Weapons: 1.0,
|
|
Shields: 1.0,
|
|
Cargo: 1.0,
|
|
Load: 0.0,
|
|
}
|
|
em := Freighter.EmptyMass()
|
|
assert.Equal(t, 0.0, sg.CarryingMass())
|
|
assert.Equal(t, em, sg.FullMass(Freighter))
|
|
|
|
sg.Load = 10.0
|
|
assert.Equal(t, 10.0, sg.CarryingMass())
|
|
assert.Equal(t, em+10.0, sg.FullMass(Freighter))
|
|
|
|
sg.Cargo = 2.5
|
|
assert.Equal(t, 4.0, sg.CarryingMass())
|
|
assert.Equal(t, em+4.0, sg.FullMass(Freighter))
|
|
}
|
|
|
|
func TestSpeed(t *testing.T) {
|
|
Freighter := &game.ShipType{
|
|
ShipTypeReport: game.ShipTypeReport{
|
|
Name: "Freighter",
|
|
Drive: 8,
|
|
Armament: 0,
|
|
Weapons: 0,
|
|
Shields: 2,
|
|
Cargo: 10,
|
|
},
|
|
}
|
|
sg := &game.ShipGroup{
|
|
Number: 1,
|
|
State: "In_Orbit",
|
|
Drive: 1.0,
|
|
Weapons: 1.0,
|
|
Shields: 1.0,
|
|
Cargo: 1.0,
|
|
Load: 0.0,
|
|
}
|
|
assert.Equal(t, 8.0, sg.Speed(Freighter))
|
|
sg.Load = 5.0
|
|
assert.Equal(t, 6.4, sg.Speed(Freighter))
|
|
sg.Drive = 1.5
|
|
assert.Equal(t, 9.6, sg.Speed(Freighter))
|
|
sg.Load = 10
|
|
sg.Cargo = 1.5
|
|
assert.Equal(t, 9.0, sg.Speed(Freighter))
|
|
}
|
|
|
|
func TestBombingPower(t *testing.T) {
|
|
Gunship := game.ShipType{
|
|
ShipTypeReport: game.ShipTypeReport{
|
|
Drive: 60.0,
|
|
Armament: 3,
|
|
Weapons: 30.0,
|
|
Shields: 100.0,
|
|
Cargo: 0.0,
|
|
},
|
|
}
|
|
sg := game.ShipGroup{
|
|
Number: 1,
|
|
State: "In_Orbit",
|
|
Drive: 1.0,
|
|
Weapons: 1.0,
|
|
Shields: 1.0,
|
|
Cargo: 1.0,
|
|
}
|
|
expectedBombingPower := 139.295
|
|
result := sg.BombingPower(&Gunship)
|
|
assert.Equal(t, expectedBombingPower, result)
|
|
}
|
|
|
|
func TestUpgradeCost(t *testing.T) {
|
|
Cruiser := game.ShipType{
|
|
ShipTypeReport: game.ShipTypeReport{
|
|
Name: "Cruiser",
|
|
Drive: 15,
|
|
Armament: 1,
|
|
Weapons: 15,
|
|
Shields: 15,
|
|
Cargo: 0,
|
|
},
|
|
}
|
|
|
|
sg := game.ShipGroup{
|
|
Number: 1,
|
|
State: "In_Orbit",
|
|
Drive: 1.0,
|
|
Weapons: 1.0,
|
|
Shields: 1.0,
|
|
Cargo: 1.0,
|
|
}
|
|
upgradeCost := sg.UpgradeDriveCost(&Cruiser, 2.0) +
|
|
sg.UpgradeWeaponsCost(&Cruiser, 2.0) +
|
|
sg.UpgradeShieldsCost(&Cruiser, 2.0) +
|
|
sg.UpgradeCargoCost(&Cruiser, 2.0)
|
|
assert.Equal(t, 225., upgradeCost)
|
|
}
|
|
|
|
func TestDriveEffective(t *testing.T) {
|
|
tc := []struct {
|
|
driveShipType float64
|
|
driveTech float64
|
|
expectDriveEffective float64
|
|
}{
|
|
{1, 1, 1},
|
|
{1, 2, 2},
|
|
{2, 1, 2},
|
|
{0, 1, 0},
|
|
{0, 1.5, 0},
|
|
{0, 10, 0},
|
|
{1.5, 1.5, 2.25},
|
|
}
|
|
for i := range tc {
|
|
someShip := game.ShipType{
|
|
ShipTypeReport: game.ShipTypeReport{
|
|
Drive: tc[i].driveShipType,
|
|
Armament: rand.UintN(30) + 1,
|
|
Weapons: rand.Float64()*30 + 1,
|
|
Shields: rand.Float64()*100 + 1,
|
|
Cargo: rand.Float64()*20 + 1,
|
|
},
|
|
}
|
|
sg := game.ShipGroup{
|
|
Number: rand.UintN(4) + 1,
|
|
State: "In_Orbit",
|
|
Drive: tc[i].driveTech,
|
|
Weapons: rand.Float64()*5 + 1,
|
|
Shields: rand.Float64()*5 + 1,
|
|
Cargo: rand.Float64()*5 + 1,
|
|
}
|
|
assert.Equal(t, tc[i].expectDriveEffective, sg.DriveEffective(&someShip))
|
|
}
|
|
}
|
|
|
|
func TestShipGroupEqual(t *testing.T) {
|
|
fleetId := uuid.New()
|
|
someUUID := uuid.New()
|
|
mat := game.CargoMaterial
|
|
cap := game.CargoCapital
|
|
left := &game.ShipGroup{
|
|
Index: 1,
|
|
Number: 1,
|
|
|
|
OwnerID: uuid.New(),
|
|
TypeID: uuid.New(),
|
|
FleetID: &fleetId,
|
|
State: "In_Orbit",
|
|
CargoType: &mat,
|
|
Load: 123.45,
|
|
Drive: 1.0,
|
|
Weapons: 1.0,
|
|
Shields: 1.0,
|
|
Cargo: 1.0,
|
|
}
|
|
|
|
// essential properties
|
|
right := *left
|
|
assert.True(t, left.Equal(right))
|
|
|
|
left.OwnerID = someUUID
|
|
assert.False(t, left.Equal(right))
|
|
|
|
right = *left
|
|
left.TypeID = someUUID
|
|
assert.False(t, left.Equal(right))
|
|
|
|
right = *left
|
|
left.FleetID = &someUUID
|
|
assert.False(t, left.Equal(right))
|
|
|
|
right = *left
|
|
left.FleetID = nil
|
|
assert.False(t, left.Equal(right))
|
|
|
|
right = *left
|
|
left.State = "In_Space"
|
|
assert.False(t, left.Equal(right))
|
|
|
|
right = *left
|
|
left.CargoType = &cap
|
|
assert.False(t, left.Equal(right))
|
|
|
|
right = *left
|
|
left.CargoType = nil
|
|
assert.False(t, left.Equal(right))
|
|
|
|
right = *left
|
|
left.Load = 45.123
|
|
assert.False(t, left.Equal(right))
|
|
|
|
right = *left
|
|
left.Drive = 1.1
|
|
assert.False(t, left.Equal(right))
|
|
|
|
right = *left
|
|
left.Weapons = 1.1
|
|
assert.False(t, left.Equal(right))
|
|
|
|
right = *left
|
|
left.Shields = 1.1
|
|
assert.False(t, left.Equal(right))
|
|
|
|
right = *left
|
|
left.Cargo = 1.1
|
|
assert.False(t, left.Equal(right))
|
|
|
|
// non-essential properties
|
|
right = *left
|
|
|
|
left.Index = 2
|
|
assert.True(t, left.Equal(right))
|
|
left.Number = 5
|
|
assert.True(t, left.Equal(right))
|
|
}
|
|
|
|
func TestCreateShips(t *testing.T) {
|
|
g := copyGame()
|
|
var err error
|
|
|
|
err = g.CreateShips(Race_0_idx, "Unknown_Ship_Type", R0_Planet_0_num, 2)
|
|
assert.ErrorContains(t, err, e.GenericErrorText(e.ErrInputEntityNotExists))
|
|
err = g.CreateShips(Race_0_idx, Race_0_Gunship, R1_Planet_1_num, 2)
|
|
assert.ErrorContains(t, err, e.GenericErrorText(e.ErrInputEntityNotOwned))
|
|
|
|
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 1))
|
|
assert.Len(t, collectGroups(g.ListShipGroups(Race_0_idx)), 1)
|
|
|
|
assert.NoError(t, g.CreateShips(Race_1_idx, Race_1_Freighter, R1_Planet_1_num, 1))
|
|
assert.Len(t, collectGroups(g.ListShipGroups(1)), 1)
|
|
|
|
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 6))
|
|
assert.Len(t, collectGroups(g.ListShipGroups(Race_0_idx)), 2)
|
|
|
|
assert.NoError(t, g.CreateShips(Race_1_idx, Race_1_Gunship, R1_Planet_1_num, 1))
|
|
assert.Len(t, collectGroups(g.ListShipGroups(1)), 2)
|
|
}
|
|
|
|
func TestJoinEqualGroups(t *testing.T) {
|
|
g := copyGame()
|
|
|
|
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 1)) // 1 -> 2
|
|
assert.NoError(t, g.CreateShips(Race_1_idx, Race_1_Freighter, R1_Planet_1_num, 1))
|
|
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 6)) // (2)
|
|
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 2)) // (3)
|
|
assert.NoError(t, g.CreateShips(Race_1_idx, Race_1_Gunship, R1_Planet_1_num, 1))
|
|
|
|
g.Race[Race_0_idx].Drive = 1.5
|
|
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 9)) // 4 -> 6
|
|
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 7)) // 5 -> 7
|
|
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 4)) // (6)
|
|
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 4)) // (7)
|
|
|
|
assert.Len(t, collectGroups(g.ListShipGroups(Race_0_idx)), 7)
|
|
|
|
g.Race[Race_1_idx].Shields = 2.0
|
|
assert.NoError(t, g.CreateShips(1, Race_1_Freighter, R1_Planet_1_num, 1))
|
|
assert.Len(t, collectGroups(g.ListShipGroups(Race_1_idx)), 3)
|
|
|
|
assert.NoError(t, g.JoinEqualGroups(Race_0.Name))
|
|
|
|
assert.Len(t, collectGroups(g.ListShipGroups(Race_1_idx)), 3)
|
|
assert.Len(t, collectGroups(g.ListShipGroups(Race_0_idx)), 4)
|
|
|
|
shipTypeID := func(ri int, name string) uuid.UUID {
|
|
st := slices.IndexFunc(g.Race[ri].ShipTypes, func(v game.ShipType) bool { return v.Name == name })
|
|
if st < 0 {
|
|
t.Fatalf("ShipType not found: %s", name)
|
|
return uuid.Nil
|
|
}
|
|
return g.Race[ri].ShipTypes[st].ID
|
|
}
|
|
|
|
for _, sg := range g.ListShipGroups(Race_0_idx) {
|
|
switch {
|
|
case sg.TypeID == shipTypeID(Race_0_idx, Race_0_Freighter) && sg.Drive == 1.1:
|
|
assert.Equal(t, uint(7), sg.Number)
|
|
assert.Equal(t, uint(2), sg.Index)
|
|
case sg.TypeID == shipTypeID(Race_0_idx, Race_0_Freighter) && sg.Drive == 1.5:
|
|
assert.Equal(t, uint(11), sg.Number)
|
|
assert.Equal(t, uint(7), sg.Index)
|
|
case sg.TypeID == shipTypeID(Race_0_idx, Race_0_Gunship) && sg.Drive == 1.1:
|
|
assert.Equal(t, uint(2), sg.Number)
|
|
assert.Equal(t, uint(3), sg.Index)
|
|
case sg.TypeID == shipTypeID(Race_0_idx, Race_0_Gunship) && sg.Drive == 1.5:
|
|
assert.Equal(t, uint(13), sg.Number)
|
|
assert.Equal(t, uint(6), sg.Index)
|
|
default:
|
|
t.Error("not all ship groups covered")
|
|
}
|
|
}
|
|
}
|
|
|
|
func collectGroups(i iter.Seq2[int, game.ShipGroup]) []game.ShipGroup {
|
|
result := make([]game.ShipGroup, 0)
|
|
for _, sg := range i {
|
|
result = append(result, sg)
|
|
}
|
|
return result
|
|
}
|