package controller_test

import (
	"fmt"
	"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.RaceShipGroups(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.RaceShipGroups(1)), 1)

	assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 6))
	assert.Len(t, slices.Collect(c.RaceShipGroups(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.RaceShipGroups(1)), 2)
}

func TestShipGroupMerge(t *testing.T) {
	c, g := 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))

	c.RaceTechLevel(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.RaceShipGroups(Race_0_idx)), 7)

	c.RaceTechLevel(Race_1_idx, game.TechShields, 2.0)
	assert.Equal(t, 2.0, c.Race(Race_1_idx).Tech[game.TechShields].F())
	assert.NoError(t, c.CreateShips(1, Race_1_Freighter, R1_Planet_1_num, 1))
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_1_idx)), 3)

	assert.ErrorContains(t,
		g.ShipGroupMerge(UnknownRace),
		e.GenericErrorText(e.ErrInputUnknownRace))
	assert.ErrorContains(t,
		g.ShipGroupMerge(Race_Extinct.Name),
		e.GenericErrorText(e.ErrRaceExinct))

	assert.NoError(t, g.ShipGroupMerge(Race_0.Name))

	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_1_idx)), 3)
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 4)

	shipTypeID := func(ri int, name string) uuid.UUID {
		class, _, ok := c.ShipClass(ri, name)
		if !ok {
			t.Fatalf("ship_class not found: %s", name)
			return uuid.Nil
		}
		return class.ID
	}

	for sg := range c.RaceShipGroups(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(1), 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(4), 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(2), 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(3), sg.Index)
		default:
			t.Error("not all ship groups covered")
		}
	}
}

func TestShipGroupBreak(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 = &InSpace

	fleet := "R0_Fleet"
	assert.NoError(t, g.ShipGroupJoinFleet(Race_0.Name, fleet, c.ShipGroup(0).ID, 0))

	assert.ErrorContains(t,
		g.ShipGroupBreak(UnknownRace, c.ShipGroup(0).ID, 0),
		e.GenericErrorText(e.ErrInputUnknownRace))
	assert.ErrorContains(t,
		g.ShipGroupBreak(Race_Extinct.Name, c.ShipGroup(0).ID, 0),
		e.GenericErrorText(e.ErrRaceExinct))
	assert.ErrorContains(t,
		g.ShipGroupBreak(Race_0.Name, uuid.New(), 0),
		e.GenericErrorText(e.ErrInputEntityNotExists))
	assert.ErrorContains(t,
		g.ShipGroupBreak(Race_0.Name, c.ShipGroup(0).ID, 17),
		e.GenericErrorText(e.ErrBeakGroupNumberNotEnough))
	assert.ErrorContains(t,
		g.ShipGroupBreak(Race_0.Name, c.ShipGroup(1).ID, 0),
		e.GenericErrorText(e.ErrShipsBusy))

	assert.Len(t, slices.Collect(c.RaceShipGroups(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.ShipGroupBreak(Race_0_idx, c.ShipGroup(0).ID, 5)) // group #3 (2)
	assert.Len(t, slices.Collect(c.RaceShipGroups(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.ShipGroupBreak(Race_0_idx, c.ShipGroup(0).ID, 2)) // group #4 (3)
	assert.Len(t, slices.Collect(c.RaceShipGroups(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.ShipGroupJoinFleet(Race_0_idx, fleet, c.ShipGroup(3).ID, 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.F()))
	assert.Equal(t, game.CargoColonist.Ref(), c.ShipGroup(3).CargoType)
	assert.Equal(t, 8.2, number.Fixed3(c.ShipGroup(3).Load.F()))

	// group #1 -> MAX 6 off the fleet
	assert.NoError(t, g.ShipGroupBreak(Race_0.Name, c.ShipGroup(0).ID, 6)) // group #1 (0)
	assert.Len(t, slices.Collect(c.RaceShipGroups(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.ShipGroupBreak(Race_0.Name, c.ShipGroup(3).ID, 0)) // group #1 (0)
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 4)
	assert.Equal(t, uint(2), c.ShipGroup(3).Number)
	assert.Nil(t, c.ShipGroup(3).FleetID)
}

func TestShipGroupTransfer(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 #2 (1)

	assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 17)) // group #3 (2) - In_Space
	assert.NoError(t, g.ShipGroupJoinFleet(Race_0.Name, "R0_Fleet", c.ShipGroup(2).ID, 0))
	assert.NotNil(t, c.ShipGroup(2).FleetID)
	c.ShipGroup(2).StateInSpace = &InSpace
	c.ShipGroup(2).CargoType = game.CargoMaterial.Ref()
	c.ShipGroup(2).Load = 1.234

	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 2)
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_1_idx)), 1)

	assert.ErrorContains(t,
		g.ShipGroupTransfer(UnknownRace, Race_1.Name, c.ShipGroup(1).ID, 0),
		e.GenericErrorText(e.ErrInputUnknownRace))
	assert.ErrorContains(t,
		g.ShipGroupTransfer(Race_0.Name, UnknownRace, c.ShipGroup(1).ID, 0),
		e.GenericErrorText(e.ErrInputUnknownRace))
	assert.ErrorContains(t,
		g.ShipGroupTransfer(Race_0.Name, Race_Extinct.Name, c.ShipGroup(1).ID, 0),
		e.GenericErrorText(e.ErrRaceExinct))
	assert.ErrorContains(t,
		g.ShipGroupTransfer(Race_Extinct.Name, Race_1.Name, c.ShipGroup(1).ID, 0),
		e.GenericErrorText(e.ErrRaceExinct))
	assert.ErrorContains(t,
		g.ShipGroupTransfer(Race_0.Name, Race_0.Name, c.ShipGroup(1).ID, 0),
		e.GenericErrorText(e.ErrInputSameRace))
	assert.ErrorContains(t,
		g.ShipGroupTransfer(Race_0.Name, Race_1.Name, uuid.New(), 0),
		e.GenericErrorText(e.ErrInputEntityNotExists))
	assert.ErrorContains(t,
		g.ShipGroupTransfer(Race_0.Name, Race_1.Name, c.ShipGroup(2).ID, 18),
		e.GenericErrorText(e.ErrBeakGroupNumberNotEnough))
	assert.ErrorContains(t,
		g.ShipGroupTransfer(Race_0.Name, Race_1.Name, c.ShipGroup(0).ID, 0),
		e.GenericErrorText(e.ErrGiveawayGroupShipsTypeNotEqual))

	assert.NoError(t, g.ShipGroupTransfer(Race_0.Name, Race_1.Name, c.ShipGroup(2).ID, 11)) // group #2 (3)
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 2)
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_1_idx)), 2)

	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(2).StateTransfer, c.ShipGroup(3).StateTransfer)
	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.ShipGroupTransfer(Race_1.Name, Race_0.Name, c.ShipGroup(3).ID, 11))
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 3)
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_1_idx)), 1)

	assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 1))
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 4)
	assert.Equal(t, game.StateInOrbit, c.ShipGroup(4).State())
	assert.NoError(t, g.ShipGroupSend(Race_0.Name, c.ShipGroup(4).ID, R0_Planet_2_num, 0))
	assert.Equal(t, game.StateLaunched, c.ShipGroup(4).State())
	assert.Equal(t, c.ShipGroup(4).OwnerID, Race_0_ID)

	assert.NoError(t, g.ShipGroupTransfer(Race_0.Name, Race_1.Name, c.ShipGroup(4).ID, 0))
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 3)
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_1_idx)), 2)
	assert.Equal(t, game.StateTransfer, c.ShipGroup(4).State())
	assert.Equal(t, c.ShipGroup(4).OwnerID, Race_1_ID)
	assert.ErrorContains(t,
		g.ShipGroupTransfer(Race_1.Name, Race_0.Name, c.ShipGroup(4).ID, 0),
		e.GenericErrorText(e.ErrShipsBusy))
}

func TestShipGroupLoad(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 = &InSpace

	// 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.RaceShipGroups(Race_0_idx)), 5)

	// tests
	assert.ErrorContains(t,
		g.ShipGroupLoad(UnknownRace, c.ShipGroup(0).ID, game.CargoMaterial.String(), 0, 0),
		e.GenericErrorText(e.ErrInputUnknownRace))
	assert.ErrorContains(t,
		g.ShipGroupLoad(Race_Extinct.Name, c.ShipGroup(0).ID, game.CargoMaterial.String(), 0, 0),
		e.GenericErrorText(e.ErrRaceExinct))
	assert.ErrorContains(t,
		g.ShipGroupLoad(Race_0.Name, c.ShipGroup(0).ID, "GOLD", 0, 0),
		e.GenericErrorText(e.ErrInputCargoTypeInvalid))
	assert.ErrorContains(t,
		g.ShipGroupLoad(Race_0.Name, uuid.New(), game.CargoMaterial.String(), 0, 0),
		e.GenericErrorText(e.ErrInputEntityNotExists))
	assert.ErrorContains(t,
		g.ShipGroupLoad(Race_0.Name, c.ShipGroup(2).ID, game.CargoMaterial.String(), 0, 0),
		e.GenericErrorText(e.ErrShipsBusy))
	assert.ErrorContains(t,
		g.ShipGroupLoad(Race_0.Name, c.ShipGroup(4).ID, game.CargoMaterial.String(), 0, 0),
		e.GenericErrorText(e.ErrInputEntityNotOwned))
	assert.ErrorContains(t,
		g.ShipGroupLoad(Race_0.Name, c.ShipGroup(1).ID, game.CargoMaterial.String(), 0, 0),
		e.GenericErrorText(e.ErrInputNoCargoBay))
	assert.ErrorContains(t,
		g.ShipGroupLoad(Race_0.Name, c.ShipGroup(3).ID, game.CargoMaterial.String(), 0, 0),
		e.GenericErrorText(e.ErrInputCargoLoadNotEqual))

	// initial planet is empty
	assert.ErrorContains(t,
		g.ShipGroupLoad(Race_0.Name, c.ShipGroup(0).ID, game.CargoMaterial.String(), 0, 0),
		e.GenericErrorText(e.ErrInputCargoLoadNotEnough))
	// add cargo to planet
	c.PutMaterial(R0_Planet_0_num, 100)
	// not enough on the planet
	assert.ErrorContains(t,
		g.ShipGroupLoad(Race_0.Name, c.ShipGroup(0).ID, game.CargoMaterial.String(), 11, 101),
		e.GenericErrorText(e.ErrInputCargoLoadNotEnough))
	// quantity > ships
	assert.ErrorContains(t,
		g.ShipGroupLoad(Race_0.Name, c.ShipGroup(0).ID, game.CargoMaterial.String(), 0, 1),
		e.GenericErrorText(e.ErrInputCargoQuantityWithoutGroupBreak))

	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 5)

	// break group and load maximum
	assert.NoError(t, g.ShipGroupLoad(Race_0.Name, c.ShipGroup(0).ID, game.CargoMaterial.String(), 2, 0))
	assert.Equal(t, 58.0, c.MustPlanet(R0_Planet_0_num).Material.F())
	assert.Len(t, slices.Collect(c.RaceShipGroups(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.F())
	assert.Equal(t, uint(2), c.ShipGroup(5).Number)
	assert.Equal(t, 42.0, c.ShipGroup(5).Load.F())

	// break group and load limited
	assert.NoError(t, g.ShipGroupLoad(Race_0.Name, c.ShipGroup(0).ID, game.CargoMaterial.String(), 2, 18))
	assert.Equal(t, 40.0, c.MustPlanet(R0_Planet_0_num).Material.F())
	assert.Len(t, slices.Collect(c.RaceShipGroups(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.F())
	assert.Equal(t, uint(2), c.ShipGroup(6).Number)
	assert.Equal(t, 18.0, c.ShipGroup(6).Load.F())

	// add cargo to planet
	c.PutMaterial(R0_Planet_0_num, 100)
	// loading all available cargo
	assert.NoError(t, g.ShipGroupLoad(Race_0.Name, c.ShipGroup(0).ID, game.CargoMaterial.String(), 0, 0))
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 7)
	assert.Equal(t, 0.0, c.MustPlanet(R0_Planet_0_num).Material.F())
	assert.Equal(t, 100.0, c.ShipGroup(0).Load.F()) // free: 131.0
	assert.Equal(t, game.CargoMaterial.Ref(), c.ShipGroup(0).CargoType)

	// add cargo to planet
	c.PutMaterial(R0_Planet_0_num, 200)
	assert.NoError(t, g.ShipGroupLoad(Race_0.Name, c.ShipGroup(0).ID, game.CargoMaterial.String(), 11, 31))
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 7)
	assert.Equal(t, 169.0, c.MustPlanet(R0_Planet_0_num).Material.F())
	assert.Equal(t, 131.0, c.ShipGroup(0).Load.F()) // free: 100.0
	assert.Equal(t, game.CargoMaterial.Ref(), c.ShipGroup(0).CargoType)

	// load to maximum cargo space left
	assert.NoError(t, g.ShipGroupLoad(Race_0.Name, c.ShipGroup(0).ID, game.CargoMaterial.String(), 11, 0))
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 7)
	assert.Equal(t, 153.0, c.MustPlanet(R0_Planet_0_num).Material.F())
	assert.Equal(t, 147.0, c.ShipGroup(0).Load.F()) // free: 0.0
	assert.Equal(t, game.CargoMaterial.Ref(), c.ShipGroup(0).CargoType)

	// ship group is full
	assert.ErrorContains(t,
		g.ShipGroupLoad(Race_0.Name, c.ShipGroup(0).ID, game.CargoMaterial.String(), 0, 0),
		e.GenericErrorText(e.ErrInputCargoLoadNoSpaceLeft))
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 7)
}

func TestShipGroupUnload(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 = &InSpace

	// 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.RaceShipGroups(Race_0_idx)), 6)

	// tests
	assert.ErrorContains(t,
		g.ShipGroupUnload(UnknownRace, c.ShipGroup(0).ID, 0, 0),
		e.GenericErrorText(e.ErrInputUnknownRace))
	assert.ErrorContains(t,
		g.ShipGroupUnload(Race_Extinct.Name, c.ShipGroup(0).ID, 0, 0),
		e.GenericErrorText(e.ErrRaceExinct))
	assert.ErrorContains(t,
		g.ShipGroupUnload(Race_0.Name, uuid.New(), 0, 0),
		e.GenericErrorText(e.ErrInputEntityNotExists))
	assert.ErrorContains(t,
		g.ShipGroupUnload(Race_0.Name, c.ShipGroup(2).ID, 0, 0),
		e.GenericErrorText(e.ErrShipsBusy))
	assert.ErrorContains(t,
		g.ShipGroupUnload(Race_0.Name, c.ShipGroup(1).ID, 0, 0),
		e.GenericErrorText(e.ErrInputNoCargoBay))
	assert.ErrorContains(t,
		g.ShipGroupUnload(Race_0.Name, c.ShipGroup(0).ID, 0, 0),
		e.GenericErrorText(e.ErrInputCargoUnloadEmpty))
	assert.ErrorContains(t,
		g.ShipGroupUnload(Race_0.Name, c.ShipGroup(4).ID, 0, 0),
		e.GenericErrorText(e.ErrInputEntityNotOwned))
	c.ShipGroup(0).CargoType = game.CargoColonist.Ref()
	c.ShipGroup(0).Load = 100
	assert.ErrorContains(t,
		g.ShipGroupUnload(Race_0.Name, c.ShipGroup(0).ID, 11, 101),
		e.GenericErrorText(e.ErrInputCargoUnoadNotEnough))
	assert.ErrorContains(t,
		g.ShipGroupUnload(Race_0.Name, c.ShipGroup(0).ID, 0, 1),
		e.GenericErrorText(e.ErrInputCargoQuantityWithoutGroupBreak))

	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 6)

	// unload MAT on foreign planet / break group
	assert.NoError(t, g.ShipGroupUnload(Race_0.Name, c.ShipGroup(5).ID, 3, 0))
	assert.Equal(t, 27.273, number.Fixed3(c.MustPlanet(R1_Planet_1_num).Material.F()))
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 7)
	assert.Equal(t, uint(3), c.ShipGroup(6).Number)
	assert.Equal(t, 0., c.ShipGroup(6).Load.F())
	assert.Nil(t, c.ShipGroup(6).CargoType)
	assert.Equal(t, 0.0, c.ShipGroup(6).Load.F())
	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.F()))

	// unload MAT on foreign planet / break group / limited MAT
	assert.NoError(t, g.ShipGroupUnload(Race_0.Name, c.ShipGroup(5).ID, 3, 20.0))
	assert.Equal(t, 47.273, number.Fixed3(c.MustPlanet(R1_Planet_1_num).Material.F()))
	assert.Len(t, slices.Collect(c.RaceShipGroups(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.F()))
	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.F()))

	// unload ALL
	assert.NoError(t, g.ShipGroupUnload(Race_0.Name, c.ShipGroup(0).ID, 0, 0))
	assert.Equal(t, 100.0, number.Fixed3(c.MustPlanet(R0_Planet_0_num).Colonists.F()))
	assert.Len(t, slices.Collect(c.RaceShipGroups(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.F()))
}

func TestShipGroupDismantle(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 = &InSpace

	// 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.RaceShipGroups(Race_0_idx)), 5)

	// tests
	assert.ErrorContains(t,
		g.ShipGroupDismantle(UnknownRace, c.ShipGroup(0).ID, 0),
		e.GenericErrorText(e.ErrInputUnknownRace))
	assert.ErrorContains(t,
		g.ShipGroupDismantle(Race_Extinct.Name, c.ShipGroup(0).ID, 0),
		e.GenericErrorText(e.ErrRaceExinct))
	assert.ErrorContains(t,
		g.ShipGroupDismantle(Race_0.Name, uuid.New(), 0),
		e.GenericErrorText(e.ErrInputEntityNotExists))
	assert.ErrorContains(t,
		g.ShipGroupDismantle(Race_0.Name, c.ShipGroup(1).ID, 0),
		e.GenericErrorText(e.ErrShipsBusy))
	assert.ErrorContains(t,
		g.ShipGroupDismantle(Race_0.Name, c.ShipGroup(2).ID, 12),
		e.GenericErrorText(e.ErrBeakGroupNumberNotEnough))

	groupEmptyMass := c.ShipGroup(4).EmptyMass(c.MustShipClass(Race_0_idx, Race_0_Freighter))
	planetMAT := c.MustPlanet(R1_Planet_1_num).Material.F()
	planetCOL := c.MustPlanet(R1_Planet_1_num).Colonists.F()

	assert.NoError(t, g.ShipGroupDismantle(Race_0.Name, c.ShipGroup(4).ID, 0))
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 4)
	assert.Equal(t, planetMAT+groupEmptyMass, c.MustPlanet(R1_Planet_1_num).Material.F())
	assert.Equal(t, planetCOL, c.MustPlanet(R1_Planet_1_num).Colonists.F())

	groupEmptyMass = c.ShipGroup(3).EmptyMass(c.MustShipClass(Race_0_idx, Race_0_Freighter))
	groupLoadMAT := c.ShipGroup(3).Load.F()
	planetMAT = c.MustPlanet(R1_Planet_1_num).Material.F()
	assert.NoError(t, g.ShipGroupDismantle(Race_0.Name, c.ShipGroup(3).ID, 0))
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 3)
	assert.Equal(t, planetMAT+groupEmptyMass+groupLoadMAT, c.MustPlanet(R1_Planet_1_num).Material.F())

	groupEmptyMass = c.ShipGroup(2).EmptyMass(c.MustShipClass(Race_0_idx, Race_0_Freighter))
	planetMAT = c.MustPlanet(R0_Planet_0_num).Material.F()
	planetCOL = c.MustPlanet(R0_Planet_0_num).Colonists.F()
	planetPOP := 90.0

	c.PutPopulation(R0_Planet_0_num, planetPOP)
	assert.Equal(t, planetPOP, c.MustPlanet(R0_Planet_0_num).Population.F())
	var quantity uint = 3
	groupEmptyMass = groupEmptyMass / float64(c.ShipGroup(2).Number) * float64(quantity)
	newGroupUnloadedCOL := c.ShipGroup(2).Load.F() / float64(c.ShipGroup(2).Number) * float64(quantity)
	expectPOPIncrease := newGroupUnloadedCOL * 8
	freePOPLeft := c.MustPlanet(R0_Planet_0_num).Size.F() - c.MustPlanet(R0_Planet_0_num).Population.F()
	expectAddedCOL := (expectPOPIncrease - freePOPLeft) / 8
	expectAddedPOP := freePOPLeft
	assert.NoError(t, g.ShipGroupDismantle(Race_0.Name, c.ShipGroup(2).ID, quantity))
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 3)
	assert.Equal(t, planetCOL+expectAddedCOL, c.MustPlanet(R0_Planet_0_num).Colonists.F())
	assert.Equal(t, planetPOP+expectAddedPOP, c.MustPlanet(R0_Planet_0_num).Population.F())
	assert.Equal(t, planetMAT+groupEmptyMass, c.MustPlanet(R0_Planet_0_num).Material.F())
	assert.Equal(t, uint(7), c.ShipGroup(2).Number)
	assert.Equal(t, 56.0, c.ShipGroup(2).Load.F())
}

func TestShipGroupDestroyItem(t *testing.T) {
	c, _ := newCache()

	assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 10))
	c.ShipGroup(0).CargoType = game.CargoColonist.Ref()
	c.ShipGroup(0).Load = 100.0

	for c.ShipGroup(0).Number > 0 {
		c.ShipGroupDestroyItem(0)
		assert.Equal(t, float64(c.ShipGroup(0).Number)*10, c.ShipGroup(0).Load.F())
	}
}

func TestState(t *testing.T) {
	assert.Equal(t, "In_Orbit", fmt.Sprintf("%s", game.StateInOrbit))
}