cmd: load cargo

internal/model/game/group_test.go

@@ -8,6 +8,7 @@ import (
 	"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"
 )
 
@@ -380,8 +381,11 @@ func TestBreakGroup(t *testing.T) {
 	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)
@@ -392,6 +396,11 @@ func TestBreakGroup(t *testing.T) {
 	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)
@@ -472,3 +481,87 @@ func TestGiveawayGroup(t *testing.T) {
 	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))
+}