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 // tests assert.ErrorContains(t, g.LoadCargo("UnknownRace", 1, game.CargoMaterial.String(), 0), e.GenericErrorText(e.ErrInputUnknownRace)) assert.ErrorContains(t, g.LoadCargo(Race_0.Name, 1, "GOLD", 0), e.GenericErrorText(e.ErrInputCargoTypeInvalid)) assert.ErrorContains(t, g.LoadCargo(Race_0.Name, 555, game.CargoMaterial.String(), 0), e.GenericErrorText(e.ErrInputEntityNotExists)) assert.ErrorContains(t, g.LoadCargo(Race_0.Name, 3, game.CargoMaterial.String(), 0), e.GenericErrorText(e.ErrShipsBusy)) assert.ErrorContains(t, g.LoadCargo(Race_0.Name, 5, game.CargoMaterial.String(), 0), e.GenericErrorText(e.ErrInputEntityNotOwned)) assert.ErrorContains(t, g.LoadCargo(Race_0.Name, 2, game.CargoMaterial.String(), 0), e.GenericErrorText(e.ErrInputCargoLoadNoCargoBay)) assert.ErrorContains(t, g.LoadCargo(Race_0.Name, 4, game.CargoMaterial.String(), 0), e.GenericErrorText(e.ErrInputCargoLoadNotEqual)) // initial planet is empty assert.ErrorContains(t, g.LoadCargo(Race_0.Name, 1, game.CargoMaterial.String(), 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(), 101), e.GenericErrorText(e.ErrInputCargoLoadNotEnough)) // loading all available cargo assert.NoError(t, g.LoadCargo(Race_0.Name, 1, game.CargoMaterial.String(), 0)) 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(), 31)) 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(), 0)) assert.Equal(t, 69.0, g.Map.Planet[0].Material) assert.Equal(t, 231.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), e.GenericErrorText(e.ErrInputCargoLoadNoSpaceLeft)) }