cmd: unload cargo

2025-12-11 23:00:31 +03:00
parent 4447d125ac
commit d0d5bc3c88
7 changed files with 322 additions and 39 deletions
@@ -507,61 +507,193 @@ func TestLoadCargo(t *testing.T) {
 	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),
+		g.LoadCargo("UnknownRace", 1, game.CargoMaterial.String(), 0, 0),
 		e.GenericErrorText(e.ErrInputUnknownRace))
 	assert.ErrorContains(t,
-		g.LoadCargo(Race_0.Name, 1, "GOLD", 0),
+		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),
+		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),
+		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),
+		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),
-		e.GenericErrorText(e.ErrInputCargoLoadNoCargoBay))
+		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),
+		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),
+		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(), 101),
+		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))
+	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(), 31))
+	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(), 0))
-	assert.Equal(t, 69.0, g.Map.Planet[0].Material)
-	assert.Equal(t, 231.0, g.ShipGroups[0].Load) // free: 0.0
+	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),
+		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))
 }