wip: refactor controller
This commit is contained in:
Ilia Denisov
@@ -0,0 +1,539 @@
|
||||
package controller_test
|
||||
|
||||
import (
|
||||
"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 TestCreateShips(t *testing.T) {
|
||||
c, _ := newCache()
|
||||
|
||||
assert.ErrorContains(t,
|
||||
c.CreateShips(Race_0_idx, "Unknown_Ship_Type", R0_Planet_0_num, 2),
|
||||
e.GenericErrorText(e.ErrInputEntityNotExists))
|
||||
assert.ErrorContains(t,
|
||||
c.CreateShips(Race_0_idx, Race_0_Gunship, R1_Planet_1_num, 2),
|
||||
e.GenericErrorText(e.ErrInputEntityNotOwned))
|
||||
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 1))
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 1)
|
||||
|
||||
assert.NoError(t, c.CreateShips(Race_1_idx, Race_1_Freighter, R1_Planet_1_num, 1))
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(1)), 1)
|
||||
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 6))
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 2)
|
||||
|
||||
assert.NoError(t, c.CreateShips(Race_1_idx, Race_1_Gunship, R1_Planet_1_num, 1))
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(1)), 2)
|
||||
}
|
||||
|
||||
func TestJoinEqualGroups(t *testing.T) {
|
||||
c, _ := newCache()
|
||||
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 1)) // 1 -> 2
|
||||
assert.NoError(t, c.CreateShips(Race_1_idx, Race_1_Freighter, R1_Planet_1_num, 1))
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 6)) // (2)
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 2)) // (3)
|
||||
assert.NoError(t, c.CreateShips(Race_1_idx, Race_1_Gunship, R1_Planet_1_num, 1))
|
||||
|
||||
// g.Race[Race_0_idx].SetTechLevel(game.TechDrive, 1.5)
|
||||
c.RacetTechLevel(Race_0_idx, game.TechDrive, 1.5)
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 9)) // 4 -> 6
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 7)) // 5 -> 7
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 4)) // (6)
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 4)) // (7)
|
||||
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 7)
|
||||
|
||||
// g.Race[Race_1_idx].SetTechLevel(game.TechShields, 2.0)
|
||||
c.RacetTechLevel(Race_1_idx, game.TechShields, 2.0)
|
||||
assert.NoError(t, c.CreateShips(1, Race_1_Freighter, R1_Planet_1_num, 1))
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_1_idx)), 3)
|
||||
|
||||
c.JoinEqualGroups(Race_0_idx)
|
||||
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_1_idx)), 3)
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 4)
|
||||
|
||||
shipTypeID := func(ri int, name string) uuid.UUID {
|
||||
class, _, ok := c.ShipClass(ri, name)
|
||||
if !ok {
|
||||
t.Fatalf("ShipType not found: %s", name)
|
||||
return uuid.Nil
|
||||
}
|
||||
return class.ID
|
||||
}
|
||||
|
||||
for sg := range c.ListShipGroups(Race_0_idx) {
|
||||
switch {
|
||||
case sg.TypeID == shipTypeID(Race_0_idx, Race_0_Freighter) && sg.TechLevel(game.TechDrive) == 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.TechLevel(game.TechDrive) == 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.TechLevel(game.TechDrive) == 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.TechLevel(game.TechDrive) == 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) {
|
||||
c, g := newCache()
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 13)) // group #1 (0)
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 7)) // group #2 (1) - In_Space
|
||||
c.ShipGroup(1).StateInSpace = &game.InSpace{
|
||||
Origin: 1,
|
||||
Range: 1,
|
||||
}
|
||||
|
||||
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(c.ListShipGroups(Race_0_idx)), 2)
|
||||
assert.Len(t, slices.Collect(c.ListFleets(Race_0_idx)), 1)
|
||||
|
||||
// group #1 -> group #3 (5 new, 8 left)
|
||||
assert.NoError(t, c.BreakGroup(Race_0_idx, 1, 5)) // group #3 (2)
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 3)
|
||||
assert.Equal(t, uint(8), c.ShipGroup(0).Number)
|
||||
assert.NotNil(t, c.ShipGroup(0).FleetID)
|
||||
assert.Equal(t, uint(5), c.ShipGroup(2).Number)
|
||||
assert.Equal(t, uint(3), c.ShipGroup(2).Index)
|
||||
assert.Nil(t, c.ShipGroup(2).FleetID)
|
||||
assert.Nil(t, c.ShipGroup(2).CargoType)
|
||||
|
||||
// group #1 -> group #4 (2 new, 6 left)
|
||||
c.ShipGroup(0).CargoType = game.CargoColonist.Ref()
|
||||
c.ShipGroup(0).Load = 32.8 // 8 ships
|
||||
assert.NoError(t, c.BreakGroup(Race_0_idx, 1, 2)) // group #4 (3)
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 4)
|
||||
assert.Equal(t, uint(6), c.ShipGroup(0).Number)
|
||||
assert.NotNil(t, c.ShipGroup(0).FleetID)
|
||||
assert.Equal(t, uint(2), c.ShipGroup(3).Number)
|
||||
assert.Equal(t, uint(4), c.ShipGroup(3).Index)
|
||||
assert.Nil(t, c.ShipGroup(3).FleetID)
|
||||
assert.NoError(t, c.JoinShipGroupToFleet(Race_0_idx, fleet, 4, 0))
|
||||
assert.NotNil(t, c.ShipGroup(3).FleetID)
|
||||
|
||||
assert.Equal(t, game.CargoColonist.Ref(), c.ShipGroup(0).CargoType)
|
||||
assert.Equal(t, 24.6, number.Fixed3(c.ShipGroup(0).Load))
|
||||
assert.Equal(t, game.CargoColonist.Ref(), c.ShipGroup(3).CargoType)
|
||||
assert.Equal(t, 8.2, number.Fixed3(c.ShipGroup(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(c.ListShipGroups(Race_0_idx)), 4)
|
||||
assert.Equal(t, uint(6), c.ShipGroup(0).Number)
|
||||
assert.Nil(t, c.ShipGroup(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(c.ListShipGroups(Race_0_idx)), 4)
|
||||
assert.Equal(t, uint(2), c.ShipGroup(3).Number)
|
||||
assert.Nil(t, c.ShipGroup(3).FleetID)
|
||||
}
|
||||
|
||||
func TestGiveawayGroup(t *testing.T) {
|
||||
c, g := newCache()
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, ShipType_Cruiser, R0_Planet_0_num, 11)) // group #1 (0)
|
||||
assert.NoError(t, c.CreateShips(Race_1_idx, ShipType_Cruiser, R1_Planet_1_num, 23)) // group #1 (1)
|
||||
|
||||
assert.NoError(t, c.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, c.ShipGroup(2).FleetID)
|
||||
c.ShipGroup(2).StateInSpace = &game.InSpace{
|
||||
Origin: 2,
|
||||
Range: 31.337,
|
||||
}
|
||||
c.ShipGroup(2).CargoType = game.CargoMaterial.Ref()
|
||||
c.ShipGroup(2).Load = 1.234
|
||||
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 2)
|
||||
assert.Len(t, slices.Collect(c.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(c.ListShipGroups(Race_0_idx)), 2)
|
||||
assert.Len(t, slices.Collect(c.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 })
|
||||
|
||||
c.MustShipClass(Race_0_idx, Race_0_Gunship)
|
||||
assert.Equal(t, c.MustShipClass(Race_1_idx, Race_0_Gunship).Name, c.MustShipClass(Race_0_idx, Race_0_Gunship).Name)
|
||||
assert.Equal(t, c.MustShipClass(Race_1_idx, Race_0_Gunship).Drive, c.MustShipClass(Race_0_idx, Race_0_Gunship).Drive)
|
||||
assert.Equal(t, c.MustShipClass(Race_1_idx, Race_0_Gunship).Weapons, c.MustShipClass(Race_0_idx, Race_0_Gunship).Weapons)
|
||||
assert.Equal(t, c.MustShipClass(Race_1_idx, Race_0_Gunship).Shields, c.MustShipClass(Race_0_idx, Race_0_Gunship).Shields)
|
||||
assert.Equal(t, c.MustShipClass(Race_1_idx, Race_0_Gunship).Cargo, c.MustShipClass(Race_0_idx, Race_0_Gunship).Cargo)
|
||||
assert.Equal(t, c.MustShipClass(Race_1_idx, Race_0_Gunship).Armament, c.MustShipClass(Race_0_idx, Race_0_Gunship).Armament)
|
||||
assert.Equal(t, c.ShipGroup(2).State(), c.ShipGroup(3).State())
|
||||
assert.Equal(t, c.ShipGroup(2).CargoType, c.ShipGroup(3).CargoType)
|
||||
assert.Equal(t, c.ShipGroup(2).Load, c.ShipGroup(3).Load)
|
||||
assert.Equal(t, c.ShipGroup(2).TechLevel(game.TechDrive), c.ShipGroup(3).TechLevel(game.TechDrive))
|
||||
assert.Equal(t, c.ShipGroup(2).TechLevel(game.TechWeapons), c.ShipGroup(3).TechLevel(game.TechWeapons))
|
||||
assert.Equal(t, c.ShipGroup(2).TechLevel(game.TechShields), c.ShipGroup(3).TechLevel(game.TechShields))
|
||||
assert.Equal(t, c.ShipGroup(2).TechLevel(game.TechCargo), c.ShipGroup(3).TechLevel(game.TechCargo))
|
||||
assert.Equal(t, c.ShipGroup(2).Destination, c.ShipGroup(3).Destination)
|
||||
assert.Equal(t, c.ShipGroup(2).StateInSpace, c.ShipGroup(3).StateInSpace)
|
||||
assert.Equal(t, c.ShipGroup(2).StateUpgrade, c.ShipGroup(3).StateUpgrade)
|
||||
assert.Equal(t, c.ShipGroup(3).OwnerID, Race_1_ID)
|
||||
assert.Equal(t, c.ShipGroup(3).TypeID, c.MustShipClass(Race_1_idx, Race_0_Gunship).ID)
|
||||
assert.Equal(t, c.ShipGroup(3).Number, uint(11))
|
||||
assert.Nil(t, c.ShipGroup(3).FleetID)
|
||||
|
||||
assert.NoError(t, g.GiveawayGroup(Race_1.Name, Race_0.Name, 2, 11))
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 3)
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_1_idx)), 1)
|
||||
}
|
||||
|
||||
func TestLoadCargo(t *testing.T) {
|
||||
c, g := newCache()
|
||||
|
||||
// 1: idx = 0 / Ready to load
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
|
||||
|
||||
// 2: idx = 1 / Has no cargo bay
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 1))
|
||||
|
||||
// 3: idx = 2 / In_Space
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 7))
|
||||
c.ShipGroup(2).StateInSpace = &game.InSpace{
|
||||
Origin: 2,
|
||||
Range: 31.337,
|
||||
}
|
||||
|
||||
// 4: idx = 3 / loaded with COL
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
|
||||
c.ShipGroup(3).CargoType = game.CargoColonist.Ref()
|
||||
c.ShipGroup(3).Load = 1.234
|
||||
|
||||
// 5: idx = 4 / on foreign planet
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
|
||||
c.ShipGroup(4).Destination = R1_Planet_1_num
|
||||
|
||||
assert.Len(t, slices.Collect(c.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
|
||||
c.PutMaterial(R0_Planet_0_num, 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(c.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, c.MustPlanet(R0_Planet_0_num).Material)
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 6)
|
||||
assert.Nil(t, c.ShipGroup(0).CargoType)
|
||||
assert.Equal(t, game.CargoMaterial.Ref(), c.ShipGroup(5).CargoType)
|
||||
assert.Equal(t, uint(9), c.ShipGroup(0).Number)
|
||||
assert.Equal(t, 0.0, c.ShipGroup(0).Load)
|
||||
assert.Equal(t, uint(2), c.ShipGroup(5).Number)
|
||||
assert.Equal(t, 42.0, c.ShipGroup(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, c.MustPlanet(R0_Planet_0_num).Material)
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 7)
|
||||
assert.Nil(t, c.ShipGroup(0).CargoType)
|
||||
assert.Equal(t, game.CargoMaterial.Ref(), c.ShipGroup(6).CargoType)
|
||||
assert.Equal(t, uint(7), c.ShipGroup(0).Number)
|
||||
assert.Equal(t, 0.0, c.ShipGroup(0).Load)
|
||||
assert.Equal(t, uint(2), c.ShipGroup(6).Number)
|
||||
assert.Equal(t, 18.0, c.ShipGroup(6).Load)
|
||||
|
||||
// add cargo to planet
|
||||
// g.Map.Planet[0].Material = 100
|
||||
c.PutMaterial(R0_Planet_0_num, 100)
|
||||
// loading all available cargo
|
||||
assert.NoError(t, g.LoadCargo(Race_0.Name, 1, game.CargoMaterial.String(), 0, 0))
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 7)
|
||||
assert.Equal(t, 0.0, c.MustPlanet(R0_Planet_0_num).Material)
|
||||
assert.Equal(t, 100.0, c.ShipGroup(0).Load) // free: 131.0
|
||||
assert.Equal(t, game.CargoMaterial.Ref(), c.ShipGroup(0).CargoType)
|
||||
|
||||
// add cargo to planet
|
||||
// g.Map.Planet[0].Material = 200
|
||||
c.PutMaterial(R0_Planet_0_num, 200)
|
||||
assert.NoError(t, g.LoadCargo(Race_0.Name, 1, game.CargoMaterial.String(), 11, 31))
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 7)
|
||||
assert.Equal(t, 169.0, c.MustPlanet(R0_Planet_0_num).Material)
|
||||
assert.Equal(t, 131.0, c.ShipGroup(0).Load) // free: 100.0
|
||||
assert.Equal(t, game.CargoMaterial.Ref(), c.ShipGroup(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(c.ListShipGroups(Race_0_idx)), 7)
|
||||
assert.Equal(t, 153.0, c.MustPlanet(R0_Planet_0_num).Material)
|
||||
assert.Equal(t, 147.0, c.ShipGroup(0).Load) // free: 0.0
|
||||
assert.Equal(t, game.CargoMaterial.Ref(), c.ShipGroup(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(c.ListShipGroups(Race_0_idx)), 7)
|
||||
}
|
||||
|
||||
func TestUnloadCargo(t *testing.T) {
|
||||
c, g := newCache()
|
||||
|
||||
// 1: idx = 0 / empty
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 10))
|
||||
|
||||
// 2: idx = 1 / Has no cargo bay
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 1))
|
||||
|
||||
// 3: idx = 2 / In_Space
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 7))
|
||||
c.ShipGroup(2).StateInSpace = &game.InSpace{
|
||||
Origin: 2,
|
||||
Range: 31.337,
|
||||
}
|
||||
|
||||
// 4: idx = 3 / loaded with COL
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
|
||||
c.ShipGroup(3).CargoType = game.CargoColonist.Ref()
|
||||
c.ShipGroup(3).Load = 1.234
|
||||
|
||||
// 5: idx = 4 / on foreign planet / loaded with COL
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
|
||||
c.ShipGroup(4).Destination = R1_Planet_1_num
|
||||
c.ShipGroup(4).CargoType = game.CargoColonist.Ref()
|
||||
c.ShipGroup(4).Load = 1.234
|
||||
|
||||
// 6: idx = 5 / on foreign planet / loaded with MAT
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
|
||||
c.ShipGroup(5).Destination = R1_Planet_1_num
|
||||
c.ShipGroup(5).CargoType = game.CargoMaterial.Ref()
|
||||
c.ShipGroup(5).Load = 100.0
|
||||
|
||||
assert.Len(t, slices.Collect(c.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))
|
||||
c.ShipGroup(0).CargoType = game.CargoColonist.Ref()
|
||||
c.ShipGroup(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(c.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(c.MustPlanet(R1_Planet_1_num).Material))
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 7)
|
||||
assert.Equal(t, uint(3), c.ShipGroup(6).Number)
|
||||
assert.Nil(t, c.ShipGroup(6).CargoType)
|
||||
assert.Equal(t, 0.0, c.ShipGroup(6).Load)
|
||||
assert.Equal(t, game.CargoMaterial.Ref(), c.ShipGroup(5).CargoType)
|
||||
assert.Equal(t, uint(8), c.ShipGroup(5).Number)
|
||||
assert.Equal(t, 72.727, number.Fixed3(c.ShipGroup(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(c.MustPlanet(R1_Planet_1_num).Material))
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 8)
|
||||
assert.Equal(t, uint(3), c.ShipGroup(7).Number)
|
||||
assert.Equal(t, game.CargoMaterial.Ref(), c.ShipGroup(7).CargoType)
|
||||
assert.Equal(t, 7.273, number.Fixed3(c.ShipGroup(7).Load))
|
||||
assert.Equal(t, game.CargoMaterial.Ref(), c.ShipGroup(5).CargoType)
|
||||
assert.Equal(t, uint(5), c.ShipGroup(5).Number)
|
||||
assert.Equal(t, 45.455, number.Fixed3(c.ShipGroup(5).Load))
|
||||
|
||||
// unload ALL
|
||||
assert.NoError(t, g.UnloadCargo(Race_0.Name, 1, 0, 0))
|
||||
assert.Equal(t, 100.0, number.Fixed3(c.MustPlanet(R0_Planet_0_num).Colonists))
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 8)
|
||||
assert.Equal(t, uint(10), c.ShipGroup(0).Number)
|
||||
assert.Nil(t, c.ShipGroup(0).CargoType)
|
||||
assert.Equal(t, 0.0, number.Fixed3(c.ShipGroup(0).Load))
|
||||
}
|
||||
|
||||
func TestDisassembleGroup(t *testing.T) {
|
||||
c, g := newCache()
|
||||
|
||||
// 1: idx = 0 / empty
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 10))
|
||||
|
||||
// 2: idx = 1 / In_Space
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 7))
|
||||
c.ShipGroup(1).StateInSpace = &game.InSpace{
|
||||
Origin: 2,
|
||||
Range: 31.337,
|
||||
}
|
||||
|
||||
// 3: idx = 2 / loaded with COL
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 10))
|
||||
c.ShipGroup(2).CargoType = game.CargoColonist.Ref()
|
||||
c.ShipGroup(2).Load = 80.0
|
||||
|
||||
// 4: idx = 3 / on foreign planet / loaded with MAT
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
|
||||
c.ShipGroup(3).Destination = R1_Planet_1_num
|
||||
c.ShipGroup(3).CargoType = game.CargoMaterial.Ref()
|
||||
c.ShipGroup(3).Load = 100.0
|
||||
|
||||
// 5: idx = 4 / on foreign planet / loaded with COL
|
||||
assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
|
||||
c.ShipGroup(4).Destination = R1_Planet_1_num
|
||||
c.ShipGroup(4).CargoType = game.CargoColonist.Ref()
|
||||
c.ShipGroup(4).Load = 2.345
|
||||
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 5)
|
||||
|
||||
// tests
|
||||
assert.ErrorContains(t,
|
||||
g.DisassembleGroup("UnknownRace", 1, 0),
|
||||
e.GenericErrorText(e.ErrInputUnknownRace))
|
||||
assert.ErrorContains(t,
|
||||
g.DisassembleGroup(Race_0.Name, 555, 0),
|
||||
e.GenericErrorText(e.ErrInputEntityNotExists))
|
||||
assert.ErrorContains(t,
|
||||
g.DisassembleGroup(Race_0.Name, 2, 0),
|
||||
e.GenericErrorText(e.ErrShipsBusy))
|
||||
assert.ErrorContains(t,
|
||||
g.DisassembleGroup(Race_0.Name, 3, 12),
|
||||
e.GenericErrorText(e.ErrBeakGroupNumberNotEnough))
|
||||
|
||||
groupEmptyMass := c.ShipGroup(4).EmptyMass(c.MustShipClass(Race_0_idx, Race_0_Freighter))
|
||||
// groupEmptyMass := c.ShipGroup(4).EmptyMass(&g.Race[Race_0_idx].ShipTypes[Race_0_Freighter_idx])
|
||||
// groupLoadCOL := c.ShipGroup(4).Load
|
||||
planetMAT := c.MustPlanet(R1_Planet_1_num).Material
|
||||
planetCOL := c.MustPlanet(R1_Planet_1_num).Colonists
|
||||
|
||||
assert.NoError(t, g.DisassembleGroup(Race_0.Name, 5, 0))
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 4)
|
||||
assert.Equal(t, planetMAT+groupEmptyMass, c.MustPlanet(R1_Planet_1_num).Material)
|
||||
assert.Equal(t, planetCOL, c.MustPlanet(R1_Planet_1_num).Colonists)
|
||||
|
||||
groupEmptyMass = c.ShipGroup(3).EmptyMass(c.MustShipClass(Race_0_idx, Race_0_Freighter))
|
||||
// groupEmptyMass = c.ShipGroup(3).EmptyMass(&g.Race[Race_0_idx].ShipTypes[Race_0_Freighter_idx])
|
||||
groupLoadMAT := c.ShipGroup(3).Load
|
||||
planetMAT = c.MustPlanet(R1_Planet_1_num).Material
|
||||
assert.NoError(t, g.DisassembleGroup(Race_0.Name, 4, 0))
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 3)
|
||||
assert.Equal(t, planetMAT+groupEmptyMass+groupLoadMAT, c.MustPlanet(R1_Planet_1_num).Material)
|
||||
|
||||
groupEmptyMass = c.ShipGroup(2).EmptyMass(c.MustShipClass(Race_0_idx, Race_0_Freighter))
|
||||
// groupEmptyMass = c.ShipGroup(2).EmptyMass(&g.Race[Race_0_idx].ShipTypes[Race_0_Freighter_idx])
|
||||
planetMAT = c.MustPlanet(R0_Planet_0_num).Material
|
||||
planetCOL = c.MustPlanet(R0_Planet_0_num).Colonists
|
||||
planetPOP := 90.0
|
||||
|
||||
// g.Map.Planet[0].Population = planetPOP
|
||||
c.PutPopulation(R0_Planet_0_num, planetPOP)
|
||||
var shipsDisassembling uint = 3
|
||||
groupEmptyMass = groupEmptyMass / float64(c.ShipGroup(2).Number) * float64(shipsDisassembling)
|
||||
newGroupUnloadedCOL := c.ShipGroup(2).Load / float64(c.ShipGroup(2).Number) * float64(shipsDisassembling)
|
||||
expectPOPIncrease := newGroupUnloadedCOL * 8
|
||||
freePOPLeft := c.MustPlanet(R0_Planet_0_num).Size - c.MustPlanet(R0_Planet_0_num).Population
|
||||
expectAddedCOL := (expectPOPIncrease - freePOPLeft) / 8
|
||||
expectAddedPOP := freePOPLeft
|
||||
assert.NoError(t, g.DisassembleGroup(Race_0.Name, 3, shipsDisassembling))
|
||||
assert.Len(t, slices.Collect(c.ListShipGroups(Race_0_idx)), 3)
|
||||
assert.Equal(t, planetCOL+expectAddedCOL, c.MustPlanet(R0_Planet_0_num).Colonists)
|
||||
assert.Equal(t, planetPOP+expectAddedPOP, c.MustPlanet(R0_Planet_0_num).Population)
|
||||
assert.Equal(t, planetMAT+groupEmptyMass, c.MustPlanet(R0_Planet_0_num).Material)
|
||||
assert.Equal(t, uint(7), c.ShipGroup(2).Number)
|
||||
assert.Equal(t, 56.0, c.ShipGroup(2).Load)
|
||||
}
|
||||
Reference in New Issue
Block a user