cmd: disassebmle group

internal/model/game/group_test.go

@@ -605,7 +605,7 @@ func TestLoadCargo(t *testing.T) {
 func TestUnloadCargo(t *testing.T) {
 	g := copyGame()
 
-	// 1: idx = 0 / Ready to unload COL
+	// 1: idx = 0 / empty
 	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 10))
 
 	// 2: idx = 1 / Has no cargo bay
@@ -697,3 +697,87 @@ func TestUnloadCargo(t *testing.T) {
 	assert.Nil(t, g.ShipGroups[0].CargoType)
 	assert.Equal(t, 0.0, number.Fixed3(g.ShipGroups[0].Load))
 }
+
+func TestDisassembleGroup(t *testing.T) {
+	g := copyGame()
+
+	// 1: idx = 0 / empty
+	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 10))
+
+	// 2: idx = 1 / In_Space
+	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 7))
+	g.ShipGroups[1].Origin = &R0_Planet_2_num
+	rng := 31.337
+	g.ShipGroups[1].Range = &rng
+	g.ShipGroups[1].State = "In_Space"
+
+	// 3: idx = 2 / loaded with COL
+	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 10))
+	g.ShipGroups[2].CargoType = game.CargoColonist.Ref()
+	g.ShipGroups[2].Load = 80.0
+
+	// 4: idx = 3 / on foreign planet / loaded with MAT
+	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
+	g.ShipGroups[3].Destination = R1_Planet_1_num
+	g.ShipGroups[3].CargoType = game.CargoMaterial.Ref()
+	g.ShipGroups[3].Load = 100.0
+
+	// 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 = 2.345
+
+	assert.Len(t, slices.Collect(g.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 := g.ShipGroups[4].EmptyMass(&g.Race[Race_0_idx].ShipTypes[Race_0_Freighter_idx])
+	// groupLoadCOL := g.ShipGroups[4].Load
+	planetMAT := g.Map.Planet[1].Material
+	planetCOL := g.Map.Planet[1].Colonists
+
+	assert.NoError(t, g.DisassembleGroup(Race_0.Name, 5, 0))
+	assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 4)
+	assert.Equal(t, planetMAT+groupEmptyMass, g.Map.Planet[1].Material)
+	assert.Equal(t, planetCOL, g.Map.Planet[1].Colonists)
+
+	groupEmptyMass = g.ShipGroups[3].EmptyMass(&g.Race[Race_0_idx].ShipTypes[Race_0_Freighter_idx])
+	groupLoadMAT := g.ShipGroups[3].Load
+	planetMAT = g.Map.Planet[1].Material
+	assert.NoError(t, g.DisassembleGroup(Race_0.Name, 4, 0))
+	assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 3)
+	assert.Equal(t, planetMAT+groupEmptyMass+groupLoadMAT, g.Map.Planet[1].Material)
+
+	groupEmptyMass = g.ShipGroups[2].EmptyMass(&g.Race[Race_0_idx].ShipTypes[Race_0_Freighter_idx])
+	planetMAT = g.Map.Planet[0].Material
+	planetCOL = g.Map.Planet[0].Colonists
+	planetPOP := 90.0
+	g.Map.Planet[0].Population = planetPOP
+	var shipsDisassembling uint = 3
+	groupEmptyMass = groupEmptyMass / float64(g.ShipGroups[2].Number) * float64(shipsDisassembling)
+	newGroupUnloadedCOL := g.ShipGroups[2].Load / float64(g.ShipGroups[2].Number) * float64(shipsDisassembling)
+	expectPOPIncrease := newGroupUnloadedCOL * 8
+	freePOPLeft := g.Map.Planet[0].Size - g.Map.Planet[0].Population
+	expectAddedCOL := (expectPOPIncrease - freePOPLeft) / 8
+	expectAddedPOP := freePOPLeft
+	assert.NoError(t, g.DisassembleGroup(Race_0.Name, 3, shipsDisassembling))
+	assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 3)
+	assert.Equal(t, planetCOL+expectAddedCOL, g.Map.Planet[0].Colonists)
+	assert.Equal(t, planetPOP+expectAddedPOP, g.Map.Planet[0].Population)
+	assert.Equal(t, planetMAT+groupEmptyMass, g.Map.Planet[0].Material)
+	assert.Equal(t, uint(7), g.ShipGroups[2].Number)
+	assert.Equal(t, 56.0, g.ShipGroups[2].Load)
+}