Files
galaxy-game/internal/model/game/group_test.go
T
2025-12-11 23:00:31 +03:00

700 lines
24 KiB
Go

package game_test
import (
"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/iliadenisov/galaxy/internal/number"
"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()
assert.ErrorContains(t,
g.CreateShips(Race_0_idx, "Unknown_Ship_Type", R0_Planet_0_num, 2),
e.GenericErrorText(e.ErrInputEntityNotExists))
assert.ErrorContains(t,
g.CreateShips(Race_0_idx, Race_0_Gunship, R1_Planet_1_num, 2),
e.GenericErrorText(e.ErrInputEntityNotOwned))
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 1))
assert.Len(t, slices.Collect(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, slices.Collect(g.ListShipGroups(1)), 1)
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 6))
assert.Len(t, slices.Collect(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, slices.Collect(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, slices.Collect(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, slices.Collect(g.ListShipGroups(Race_1_idx)), 3)
assert.NoError(t, g.JoinEqualGroups(Race_0.Name))
assert.Len(t, slices.Collect(g.ListShipGroups(Race_1_idx)), 3)
assert.Len(t, slices.Collect(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 TestBreakGroup(t *testing.T) {
g := copyGame()
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 13)) // group #1 (0)
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 7)) // group #2 (1) - In_Space
g.ShipGroups[1].State = "In_Space"
fleet := "R0_Fleet"
assert.NoError(t, g.JoinShipGroupToFleet(Race_0.Name, fleet, 1, 0))
assert.ErrorContains(t,
g.BreakGroup("UnknownRace", 1, 0),
e.GenericErrorText(e.ErrInputUnknownRace))
assert.ErrorContains(t,
g.BreakGroup(Race_0.Name, 555, 0),
e.GenericErrorText(e.ErrInputEntityNotExists))
assert.ErrorContains(t,
g.BreakGroup(Race_0.Name, 1, 17),
e.GenericErrorText(e.ErrBeakGroupNumberNotEnough))
assert.ErrorContains(t,
g.BreakGroup(Race_0.Name, 2, 0),
e.GenericErrorText(e.ErrShipsBusy))
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 2)
assert.Len(t, slices.Collect(g.ListFleets(Race_0_idx)), 1)
// group #1 -> group #3 (5 new, 8 left)
assert.NoError(t, g.BreakGroup(Race_0.Name, 1, 5)) // group #3 (2)
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 3)
assert.Equal(t, uint(8), g.ShipGroups[0].Number)
assert.NotNil(t, g.ShipGroups[0].FleetID)
assert.Equal(t, uint(5), g.ShipGroups[2].Number)
assert.Equal(t, uint(3), g.ShipGroups[2].Index)
assert.Nil(t, g.ShipGroups[2].FleetID)
assert.Nil(t, g.ShipGroups[2].CargoType)
// group #1 -> group #4 (2 new, 6 left)
g.ShipGroups[0].CargoType = game.CargoColonist.Ref()
g.ShipGroups[0].Load = 32.8 // 8 ships
assert.NoError(t, g.BreakGroup(Race_0.Name, 1, 2)) // group #4 (3)
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 4)
assert.Equal(t, uint(6), g.ShipGroups[0].Number)
assert.NotNil(t, g.ShipGroups[0].FleetID)
assert.Equal(t, uint(2), g.ShipGroups[3].Number)
assert.Equal(t, uint(4), g.ShipGroups[3].Index)
assert.Nil(t, g.ShipGroups[3].FleetID)
assert.NoError(t, g.JoinShipGroupToFleet(Race_0.Name, fleet, 4, 0))
assert.NotNil(t, g.ShipGroups[3].FleetID)
assert.Equal(t, game.CargoColonist.Ref(), g.ShipGroups[0].CargoType)
assert.Equal(t, 24.6, number.Fixed3(g.ShipGroups[0].Load))
assert.Equal(t, game.CargoColonist.Ref(), g.ShipGroups[3].CargoType)
assert.Equal(t, 8.2, number.Fixed3(g.ShipGroups[3].Load))
// group #1 -> MAX 6 off the fleet
assert.NoError(t, g.BreakGroup(Race_0.Name, 1, 6)) // group #1 (0)
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 4)
assert.Equal(t, uint(6), g.ShipGroups[0].Number)
assert.Nil(t, g.ShipGroups[0].FleetID)
// group #4 -> ALL off the fleet
assert.NoError(t, g.BreakGroup(Race_0.Name, 4, 0)) // group #1 (0)
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 4)
assert.Equal(t, uint(2), g.ShipGroups[3].Number)
assert.Nil(t, g.ShipGroups[3].FleetID)
}
func TestGiveawayGroup(t *testing.T) {
g := copyGame()
assert.NoError(t, g.CreateShips(Race_0_idx, ShipType_Cruiser, R0_Planet_0_num, 11)) // group #1 (0)
assert.NoError(t, g.CreateShips(Race_1_idx, ShipType_Cruiser, R1_Planet_1_num, 23)) // group #1 (1)
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 17)) // group #2 (2) - In_Space
assert.NoError(t, g.JoinShipGroupToFleet(Race_0.Name, "R0_Fleet", 2, 0))
assert.NotNil(t, g.ShipGroups[2].FleetID)
g.ShipGroups[2].Origin = &R0_Planet_2_num
rng := 31.337
g.ShipGroups[2].Range = &rng
g.ShipGroups[2].State = "In_Space"
g.ShipGroups[2].CargoType = game.CargoMaterial.Ref()
g.ShipGroups[2].Load = 1.234
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 2)
assert.Len(t, slices.Collect(g.ListShipGroups(Race_1_idx)), 1)
assert.ErrorContains(t,
g.GiveawayGroup("UnknownRace", Race_1.Name, 2, 0),
e.GenericErrorText(e.ErrInputUnknownRace))
assert.ErrorContains(t,
g.GiveawayGroup(Race_0.Name, "UnknownRace", 2, 0),
e.GenericErrorText(e.ErrInputUnknownRace))
assert.ErrorContains(t,
g.GiveawayGroup(Race_0.Name, Race_0.Name, 2, 0),
e.GenericErrorText(e.ErrInputSameRace))
assert.ErrorContains(t,
g.GiveawayGroup(Race_0.Name, Race_1.Name, 555, 0),
e.GenericErrorText(e.ErrInputEntityNotExists))
assert.ErrorContains(t,
g.GiveawayGroup(Race_0.Name, Race_1.Name, 2, 18),
e.GenericErrorText(e.ErrBeakGroupNumberNotEnough))
assert.ErrorContains(t,
g.GiveawayGroup(Race_0.Name, Race_1.Name, 1, 0),
e.GenericErrorText(e.ErrGiveawayGroupShipsTypeNotEqual))
assert.NoError(t, g.GiveawayGroup(Race_0.Name, Race_1.Name, 2, 11)) // group #2 (3)
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 2)
assert.Len(t, slices.Collect(g.ListShipGroups(Race_1_idx)), 2)
sto := slices.IndexFunc(g.Race[Race_0_idx].ShipTypes, func(st game.ShipType) bool { return st.Name == Race_0_Gunship })
sti := slices.IndexFunc(g.Race[Race_1_idx].ShipTypes, func(st game.ShipType) bool { return st.Name == Race_0_Gunship })
assert.Equal(t, g.Race[Race_1_idx].ShipTypes[sti].Name, g.Race[Race_0_idx].ShipTypes[sto].Name)
assert.Equal(t, g.Race[Race_1_idx].ShipTypes[sti].Drive, g.Race[Race_0_idx].ShipTypes[sto].Drive)
assert.Equal(t, g.Race[Race_1_idx].ShipTypes[sti].Weapons, g.Race[Race_0_idx].ShipTypes[sto].Weapons)
assert.Equal(t, g.Race[Race_1_idx].ShipTypes[sti].Shields, g.Race[Race_0_idx].ShipTypes[sto].Shields)
assert.Equal(t, g.Race[Race_1_idx].ShipTypes[sti].Cargo, g.Race[Race_0_idx].ShipTypes[sto].Cargo)
assert.Equal(t, g.Race[Race_1_idx].ShipTypes[sti].Armament, g.Race[Race_0_idx].ShipTypes[sto].Armament)
assert.Equal(t, g.ShipGroups[2].State, g.ShipGroups[3].State)
assert.Equal(t, g.ShipGroups[2].CargoType, g.ShipGroups[3].CargoType)
assert.Equal(t, g.ShipGroups[2].Load, g.ShipGroups[3].Load)
assert.Equal(t, g.ShipGroups[2].Drive, g.ShipGroups[3].Drive)
assert.Equal(t, g.ShipGroups[2].Weapons, g.ShipGroups[3].Weapons)
assert.Equal(t, g.ShipGroups[2].Shields, g.ShipGroups[3].Shields)
assert.Equal(t, g.ShipGroups[2].Cargo, g.ShipGroups[3].Cargo)
assert.Equal(t, g.ShipGroups[2].Destination, g.ShipGroups[3].Destination)
assert.Equal(t, g.ShipGroups[2].Origin, g.ShipGroups[3].Origin)
assert.Equal(t, g.ShipGroups[2].Range, g.ShipGroups[3].Range)
assert.Equal(t, g.ShipGroups[3].OwnerID, g.Race[Race_1_idx].ID)
assert.Equal(t, g.ShipGroups[3].TypeID, g.Race[Race_1_idx].ShipTypes[sti].ID)
assert.Equal(t, g.ShipGroups[3].Number, uint(11))
assert.Nil(t, g.ShipGroups[3].FleetID)
assert.NoError(t, g.GiveawayGroup(Race_1.Name, Race_0.Name, 2, 11))
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 3)
assert.Len(t, slices.Collect(g.ListShipGroups(Race_1_idx)), 1)
}
func TestLoadCargo(t *testing.T) {
g := copyGame()
// 1: idx = 0 / Ready to load
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
// 2: idx = 1 / Has no cargo bay
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 1))
// 3: idx = 2 / In_Space
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 7))
g.ShipGroups[2].Origin = &R0_Planet_2_num
rng := 31.337
g.ShipGroups[2].Range = &rng
g.ShipGroups[2].State = "In_Space"
// 4: idx = 3 / loaded with COL
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
g.ShipGroups[3].CargoType = game.CargoColonist.Ref()
g.ShipGroups[3].Load = 1.234
// 5: idx = 4 / on foreign planet
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
g.ShipGroups[4].Destination = R1_Planet_1_num
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 5)
// tests
assert.ErrorContains(t,
g.LoadCargo("UnknownRace", 1, game.CargoMaterial.String(), 0, 0),
e.GenericErrorText(e.ErrInputUnknownRace))
assert.ErrorContains(t,
g.LoadCargo(Race_0.Name, 1, "GOLD", 0, 0),
e.GenericErrorText(e.ErrInputCargoTypeInvalid))
assert.ErrorContains(t,
g.LoadCargo(Race_0.Name, 555, game.CargoMaterial.String(), 0, 0),
e.GenericErrorText(e.ErrInputEntityNotExists))
assert.ErrorContains(t,
g.LoadCargo(Race_0.Name, 3, game.CargoMaterial.String(), 0, 0),
e.GenericErrorText(e.ErrShipsBusy))
assert.ErrorContains(t,
g.LoadCargo(Race_0.Name, 5, game.CargoMaterial.String(), 0, 0),
e.GenericErrorText(e.ErrInputEntityNotOwned))
assert.ErrorContains(t,
g.LoadCargo(Race_0.Name, 2, game.CargoMaterial.String(), 0, 0),
e.GenericErrorText(e.ErrInputNoCargoBay))
assert.ErrorContains(t,
g.LoadCargo(Race_0.Name, 4, game.CargoMaterial.String(), 0, 0),
e.GenericErrorText(e.ErrInputCargoLoadNotEqual))
// initial planet is empty
assert.ErrorContains(t,
g.LoadCargo(Race_0.Name, 1, game.CargoMaterial.String(), 0, 0),
e.GenericErrorText(e.ErrInputCargoLoadNotEnough))
// add cargo to planet
g.Map.Planet[0].Material = 100
// not enough on the planet
assert.ErrorContains(t,
g.LoadCargo(Race_0.Name, 1, game.CargoMaterial.String(), 11, 101),
e.GenericErrorText(e.ErrInputCargoLoadNotEnough))
// quantity > ships
assert.ErrorContains(t,
g.LoadCargo(Race_0.Name, 1, game.CargoMaterial.String(), 0, 1),
e.GenericErrorText(e.ErrInputCargoQuantityWithoutGroupBreak))
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 5)
// break group and load maximum
assert.NoError(t, g.LoadCargo(Race_0.Name, 1, game.CargoMaterial.String(), 2, 0))
assert.Equal(t, 58.0, g.Map.Planet[0].Material)
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 6)
assert.Nil(t, g.ShipGroups[0].CargoType)
assert.Equal(t, game.CargoMaterial.Ref(), g.ShipGroups[5].CargoType)
assert.Equal(t, uint(9), g.ShipGroups[0].Number)
assert.Equal(t, 0.0, g.ShipGroups[0].Load)
assert.Equal(t, uint(2), g.ShipGroups[5].Number)
assert.Equal(t, 42.0, g.ShipGroups[5].Load)
// break group and load limited
assert.NoError(t, g.LoadCargo(Race_0.Name, 1, game.CargoMaterial.String(), 2, 18))
assert.Equal(t, 40.0, g.Map.Planet[0].Material)
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 7)
assert.Nil(t, g.ShipGroups[0].CargoType)
assert.Equal(t, game.CargoMaterial.Ref(), g.ShipGroups[6].CargoType)
assert.Equal(t, uint(7), g.ShipGroups[0].Number)
assert.Equal(t, 0.0, g.ShipGroups[0].Load)
assert.Equal(t, uint(2), g.ShipGroups[6].Number)
assert.Equal(t, 18.0, g.ShipGroups[6].Load)
// add cargo to planet
g.Map.Planet[0].Material = 100
// loading all available cargo
assert.NoError(t, g.LoadCargo(Race_0.Name, 1, game.CargoMaterial.String(), 0, 0))
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 7)
assert.Equal(t, 0.0, g.Map.Planet[0].Material)
assert.Equal(t, 100.0, g.ShipGroups[0].Load) // free: 131.0
assert.Equal(t, game.CargoMaterial.Ref(), g.ShipGroups[0].CargoType)
// add cargo to planet
g.Map.Planet[0].Material = 200
assert.NoError(t, g.LoadCargo(Race_0.Name, 1, game.CargoMaterial.String(), 11, 31))
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 7)
assert.Equal(t, 169.0, g.Map.Planet[0].Material)
assert.Equal(t, 131.0, g.ShipGroups[0].Load) // free: 100.0
assert.Equal(t, game.CargoMaterial.Ref(), g.ShipGroups[0].CargoType)
// load to maximum cargo space left
assert.NoError(t, g.LoadCargo(Race_0.Name, 1, game.CargoMaterial.String(), 11, 0))
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 7)
assert.Equal(t, 153.0, g.Map.Planet[0].Material)
assert.Equal(t, 147.0, g.ShipGroups[0].Load) // free: 0.0
assert.Equal(t, game.CargoMaterial.Ref(), g.ShipGroups[0].CargoType)
// ship group is full
assert.ErrorContains(t,
g.LoadCargo(Race_0.Name, 1, game.CargoMaterial.String(), 0, 0),
e.GenericErrorText(e.ErrInputCargoLoadNoSpaceLeft))
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 7)
}
func TestUnloadCargo(t *testing.T) {
g := copyGame()
// 1: idx = 0 / Ready to unload COL
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 10))
// 2: idx = 1 / Has no cargo bay
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 1))
// 3: idx = 2 / In_Space
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 7))
g.ShipGroups[2].Origin = &R0_Planet_2_num
rng := 31.337
g.ShipGroups[2].Range = &rng
g.ShipGroups[2].State = "In_Space"
// 4: idx = 3 / loaded with COL
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
g.ShipGroups[3].CargoType = game.CargoColonist.Ref()
g.ShipGroups[3].Load = 1.234
// 5: idx = 4 / on foreign planet / loaded with COL
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
g.ShipGroups[4].Destination = R1_Planet_1_num
g.ShipGroups[4].CargoType = game.CargoColonist.Ref()
g.ShipGroups[4].Load = 1.234
// 6: idx = 5 / on foreign planet / loaded with MAT
assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
g.ShipGroups[5].Destination = R1_Planet_1_num
g.ShipGroups[5].CargoType = game.CargoMaterial.Ref()
g.ShipGroups[5].Load = 100.0
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 6)
// tests
assert.ErrorContains(t,
g.UnloadCargo("UnknownRace", 1, 0, 0),
e.GenericErrorText(e.ErrInputUnknownRace))
assert.ErrorContains(t,
g.UnloadCargo(Race_0.Name, 555, 0, 0),
e.GenericErrorText(e.ErrInputEntityNotExists))
assert.ErrorContains(t,
g.UnloadCargo(Race_0.Name, 3, 0, 0),
e.GenericErrorText(e.ErrShipsBusy))
assert.ErrorContains(t,
g.UnloadCargo(Race_0.Name, 2, 0, 0),
e.GenericErrorText(e.ErrInputNoCargoBay))
assert.ErrorContains(t,
g.UnloadCargo(Race_0.Name, 1, 0, 0),
e.GenericErrorText(e.ErrInputCargoUnloadEmpty))
assert.ErrorContains(t,
g.UnloadCargo(Race_0.Name, 5, 0, 0),
e.GenericErrorText(e.ErrInputEntityNotOwned))
g.ShipGroups[0].CargoType = game.CargoColonist.Ref()
g.ShipGroups[0].Load = 100
assert.ErrorContains(t,
g.UnloadCargo(Race_0.Name, 1, 11, 101),
e.GenericErrorText(e.ErrInputCargoUnoadNotEnough))
assert.ErrorContains(t,
g.UnloadCargo(Race_0.Name, 1, 0, 1),
e.GenericErrorText(e.ErrInputCargoQuantityWithoutGroupBreak))
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 6)
// unload MAT on foreign planet / break group
assert.NoError(t, g.UnloadCargo(Race_0.Name, 6, 3, 0))
assert.Equal(t, 27.273, number.Fixed3(g.Map.Planet[1].Material))
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 7)
assert.Equal(t, uint(3), g.ShipGroups[6].Number)
assert.Nil(t, g.ShipGroups[6].CargoType)
assert.Equal(t, 0.0, g.ShipGroups[6].Load)
assert.Equal(t, game.CargoMaterial.Ref(), g.ShipGroups[5].CargoType)
assert.Equal(t, uint(8), g.ShipGroups[5].Number)
assert.Equal(t, 72.727, number.Fixed3(g.ShipGroups[5].Load))
// unload MAT on foreign planet / break group / limited MAT
assert.NoError(t, g.UnloadCargo(Race_0.Name, 6, 3, 20.0))
assert.Equal(t, 47.273, number.Fixed3(g.Map.Planet[1].Material))
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 8)
assert.Equal(t, uint(3), g.ShipGroups[7].Number)
assert.Equal(t, game.CargoMaterial.Ref(), g.ShipGroups[7].CargoType)
assert.Equal(t, 7.273, number.Fixed3(g.ShipGroups[7].Load))
assert.Equal(t, game.CargoMaterial.Ref(), g.ShipGroups[5].CargoType)
assert.Equal(t, uint(5), g.ShipGroups[5].Number)
assert.Equal(t, 45.455, number.Fixed3(g.ShipGroups[5].Load))
// unload ALL
assert.NoError(t, g.UnloadCargo(Race_0.Name, 1, 0, 0))
assert.Equal(t, 100.0, number.Fixed3(g.Map.Planet[0].Colonists))
assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 8)
assert.Equal(t, uint(10), g.ShipGroups[0].Number)
assert.Nil(t, g.ShipGroups[0].CargoType)
assert.Equal(t, 0.0, number.Fixed3(g.ShipGroups[0].Load))
}