package game_test

import (
	"math/rand/v2"
	"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 TestCargoCapacity(t *testing.T) {
	test := func(cargoSize float64, expectCapacity float64) {
		ship := game.ShipType{
			ShipTypeReport: game.ShipTypeReport{
				Drive:    1,
				Armament: 1,
				Weapons:  1,
				Shields:  1,
				Cargo:    cargoSize,
			},
		}
		sg := game.ShipGroup{
			Number: 1,
			Tech: map[game.Tech]float64{
				game.TechDrive:   1.5,
				game.TechWeapons: 1.1,
				game.TechShields: 2.0,
				game.TechCargo:   1.0,
			},
		}
		assert.Equal(t, expectCapacity, sg.CargoCapacity(&ship))
	}
	test(1, 1.05)
	test(5, 6.25)
	test(10, 15)
	test(50, 175)
	test(100, 600)
}

func TestCarryingAndFullMass(t *testing.T) {
	Freighter := &game.ShipType{
		ShipTypeReport: game.ShipTypeReport{
			Name:     "Freighter",
			Drive:    8,
			Armament: 0,
			Weapons:  0,
			Shields:  2,
			Cargo:    10,
		},
	}
	sg := &game.ShipGroup{
		Number: 1,
		Tech: map[game.Tech]float64{
			game.TechDrive:   1.0,
			game.TechWeapons: 1.0,
			game.TechShields: 1.0,
			game.TechCargo:   1.0,
		},
		Load: 0.0,
	}
	em := Freighter.EmptyMass()
	assert.Equal(t, 0.0, sg.CarryingMass())
	assert.Equal(t, em, sg.FullMass(Freighter))

	sg.Load = 10.0
	assert.Equal(t, 10.0, sg.CarryingMass())
	assert.Equal(t, em+10.0, sg.FullMass(Freighter))

	sg.SetTechLevel(game.TechCargo, 2.5)
	assert.Equal(t, 4.0, sg.CarryingMass())
	assert.Equal(t, em+4.0, sg.FullMass(Freighter))
}

func TestSpeed(t *testing.T) {
	Freighter := &game.ShipType{
		ShipTypeReport: game.ShipTypeReport{
			Name:     "Freighter",
			Drive:    8,
			Armament: 0,
			Weapons:  0,
			Shields:  2,
			Cargo:    10,
		},
	}
	sg := &game.ShipGroup{
		Number: 1,
		Tech: map[game.Tech]float64{
			game.TechDrive:   1.0,
			game.TechWeapons: 1.0,
			game.TechShields: 1.0,
			game.TechCargo:   1.0,
		},
		Load: 0.0,
	}
	assert.Equal(t, 8.0, sg.Speed(Freighter))
	sg.Load = 5.0
	assert.Equal(t, 6.4, sg.Speed(Freighter))
	sg.SetTechLevel(game.TechDrive, 1.5)
	assert.Equal(t, 9.6, sg.Speed(Freighter))
	sg.Load = 10
	sg.SetTechLevel(game.TechCargo, 1.5)
	assert.Equal(t, 9.0, sg.Speed(Freighter))
}

func TestBombingPower(t *testing.T) {
	Gunship := game.ShipType{
		ShipTypeReport: game.ShipTypeReport{
			Drive:    60.0,
			Armament: 3,
			Weapons:  30.0,
			Shields:  100.0,
			Cargo:    0.0,
		},
	}
	sg := game.ShipGroup{
		Number: 1,
		Tech: map[game.Tech]float64{
			game.TechDrive:   1.0,
			game.TechWeapons: 1.0,
			game.TechShields: 1.0,
			game.TechCargo:   1.0,
		},
	}
	expectedBombingPower := 139.295
	result := sg.BombingPower(&Gunship)
	assert.Equal(t, expectedBombingPower, result)
}

func TestDriveEffective(t *testing.T) {
	tc := []struct {
		driveShipType        float64
		driveTech            float64
		expectDriveEffective float64
	}{
		{1, 1, 1},
		{1, 2, 2},
		{2, 1, 2},
		{0, 1, 0},
		{0, 1.5, 0},
		{0, 10, 0},
		{1.5, 1.5, 2.25},
	}
	for i := range tc {
		someShip := game.ShipType{
			ShipTypeReport: game.ShipTypeReport{
				Drive:    tc[i].driveShipType,
				Armament: rand.UintN(30) + 1,
				Weapons:  rand.Float64()*30 + 1,
				Shields:  rand.Float64()*100 + 1,
				Cargo:    rand.Float64()*20 + 1,
			},
		}
		sg := game.ShipGroup{
			Number: rand.UintN(4) + 1,
			Tech: map[game.Tech]float64{
				game.TechDrive:   tc[i].driveTech,
				game.TechWeapons: rand.Float64()*5 + 1,
				game.TechShields: rand.Float64()*5 + 1,
				game.TechCargo:   rand.Float64()*5 + 1,
			},
		}
		assert.Equal(t, tc[i].expectDriveEffective, sg.DriveEffective(&someShip))
	}
}

func TestShipGroupEqual(t *testing.T) {
	fleetId := uuid.New()
	someUUID := uuid.New()
	mat := game.CargoMaterial
	cap := game.CargoCapital
	left := &game.ShipGroup{
		Index:  1,
		Number: 1,

		OwnerID:   uuid.New(),
		TypeID:    uuid.New(),
		FleetID:   &fleetId,
		CargoType: &mat,
		Load:      123.45,
		Tech: map[game.Tech]float64{
			game.TechDrive:   1.0,
			game.TechWeapons: 1.0,
			game.TechShields: 1.0,
			game.TechCargo:   1.0,
		},
	}

	// essential properties
	right := *left
	assert.True(t, left.Equal(right))

	left.OwnerID = someUUID
	assert.False(t, left.Equal(right))

	right = *left
	left.TypeID = someUUID
	assert.False(t, left.Equal(right))

	right = *left
	left.FleetID = &someUUID
	assert.False(t, left.Equal(right))

	right = *left
	left.FleetID = nil
	assert.False(t, left.Equal(right))

	right = *left
	left.StateInSpace = &game.InSpace{
		Origin: 1,
		Range:  1,
	}
	assert.False(t, left.Equal(right))

	right = *left
	left.CargoType = &cap
	assert.False(t, left.Equal(right))

	right = *left
	left.CargoType = nil
	assert.False(t, left.Equal(right))

	right = *left
	left.Load = 45.123
	assert.False(t, left.Equal(right))

	right = *left
	left.SetTechLevel(game.TechDrive, 1.1)
	assert.Equal(t, 1.1, left.TechLevel(game.TechDrive))
	assert.False(t, left.Equal(right))

	right = *left
	left.SetTechLevel(game.TechWeapons, 1.1)
	assert.Equal(t, 1.1, left.TechLevel(game.TechWeapons))
	assert.False(t, left.Equal(right))

	right = *left
	left.SetTechLevel(game.TechShields, 1.1)
	assert.Equal(t, 1.1, left.TechLevel(game.TechShields))
	assert.False(t, left.Equal(right))

	right = *left
	left.SetTechLevel(game.TechCargo, 1.1)
	assert.Equal(t, 1.1, left.TechLevel(game.TechCargo))
	assert.False(t, left.Equal(right))

	// non-essential properties
	right = *left

	left.Index = 2
	assert.True(t, left.Equal(right))
	left.Number = 5
	assert.True(t, left.Equal(right))
}

func TestCreateShips(t *testing.T) {
	g := newGame()

	assert.ErrorContains(t,
		g.CreateShips(Race_0_idx, "Unknown_Ship_Type", R0_Planet_0_num, 2),
		e.GenericErrorText(e.ErrInputEntityNotExists))
	assert.ErrorContains(t,
		g.CreateShips(Race_0_idx, Race_0_Gunship, R1_Planet_1_num, 2),
		e.GenericErrorText(e.ErrInputEntityNotOwned))

	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 1))
	assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 1)

	assert.NoError(t, g.CreateShips(Race_1_idx, Race_1_Freighter, R1_Planet_1_num, 1))
	assert.Len(t, slices.Collect(g.ListShipGroups(1)), 1)

	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 6))
	assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 2)

	assert.NoError(t, g.CreateShips(Race_1_idx, Race_1_Gunship, R1_Planet_1_num, 1))
	assert.Len(t, slices.Collect(g.ListShipGroups(1)), 2)
}

func TestJoinEqualGroups(t *testing.T) {
	g := newGame()

	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 1)) // 1 -> 2
	assert.NoError(t, g.CreateShips(Race_1_idx, Race_1_Freighter, R1_Planet_1_num, 1))
	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 6)) // (2)
	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 2))   // (3)
	assert.NoError(t, g.CreateShips(Race_1_idx, Race_1_Gunship, R1_Planet_1_num, 1))

	g.Race[Race_0_idx].SetTechLevel(game.TechDrive, 1.5)
	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 9))   // 4 -> 6
	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 7)) // 5 -> 7
	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 4))   // (6)
	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 4)) // (7)

	assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 7)

	g.Race[Race_1_idx].SetTechLevel(game.TechShields, 2.0)
	assert.NoError(t, g.CreateShips(1, Race_1_Freighter, R1_Planet_1_num, 1))
	assert.Len(t, slices.Collect(g.ListShipGroups(Race_1_idx)), 3)

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

	assert.Len(t, slices.Collect(g.ListShipGroups(Race_1_idx)), 3)
	assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 4)

	shipTypeID := func(ri int, name string) uuid.UUID {
		st := slices.IndexFunc(g.Race[ri].ShipTypes, func(v game.ShipType) bool { return v.Name == name })
		if st < 0 {
			t.Fatalf("ShipType not found: %s", name)
			return uuid.Nil
		}
		return g.Race[ri].ShipTypes[st].ID
	}

	for sg := range g.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) {
	g := newGame()
	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 13)) // group #1 (0)
	assert.NoError(t, g.CreateShips(Race_0_idx, Race_0_Gunship, R0_Planet_0_num, 7))    // group #2 (1) - In_Space
	g.ShipGroups[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(g.ListShipGroups(Race_0_idx)), 2)
	assert.Len(t, slices.Collect(g.ListFleets(Race_0_idx)), 1)

	// group #1 -> group #3 (5 new, 8 left)
	assert.NoError(t, g.BreakGroup(Race_0.Name, 1, 5)) // group #3 (2)
	assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 3)
	assert.Equal(t, uint(8), g.ShipGroups[0].Number)
	assert.NotNil(t, g.ShipGroups[0].FleetID)
	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)
	assert.NotNil(t, g.ShipGroups[0].FleetID)
	assert.Equal(t, uint(2), g.ShipGroups[3].Number)
	assert.Equal(t, uint(4), g.ShipGroups[3].Index)
	assert.Nil(t, g.ShipGroups[3].FleetID)
	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)
	assert.Equal(t, uint(6), g.ShipGroups[0].Number)
	assert.Nil(t, g.ShipGroups[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(g.ListShipGroups(Race_0_idx)), 4)
	assert.Equal(t, uint(2), g.ShipGroups[3].Number)
	assert.Nil(t, g.ShipGroups[3].FleetID)
}

func TestGiveawayGroup(t *testing.T) {
	g := newGame()
	assert.NoError(t, g.CreateShips(Race_0_idx, ShipType_Cruiser, R0_Planet_0_num, 11)) // group #1 (0)
	assert.NoError(t, g.CreateShips(Race_1_idx, ShipType_Cruiser, R1_Planet_1_num, 23)) // group #1 (1)

	assert.NoError(t, g.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, g.ShipGroups[2].FleetID)
	g.ShipGroups[2].StateInSpace = &game.InSpace{
		Origin: 2,
		Range:  31.337,
	}
	g.ShipGroups[2].CargoType = game.CargoMaterial.Ref()
	g.ShipGroups[2].Load = 1.234

	assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 2)
	assert.Len(t, slices.Collect(g.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(g.ListShipGroups(Race_0_idx)), 2)
	assert.Len(t, slices.Collect(g.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 })
	assert.Equal(t, g.Race[Race_1_idx].ShipTypes[sti].Name, g.Race[Race_0_idx].ShipTypes[sto].Name)
	assert.Equal(t, g.Race[Race_1_idx].ShipTypes[sti].Drive, g.Race[Race_0_idx].ShipTypes[sto].Drive)
	assert.Equal(t, g.Race[Race_1_idx].ShipTypes[sti].Weapons, g.Race[Race_0_idx].ShipTypes[sto].Weapons)
	assert.Equal(t, g.Race[Race_1_idx].ShipTypes[sti].Shields, g.Race[Race_0_idx].ShipTypes[sto].Shields)
	assert.Equal(t, g.Race[Race_1_idx].ShipTypes[sti].Cargo, g.Race[Race_0_idx].ShipTypes[sto].Cargo)
	assert.Equal(t, g.Race[Race_1_idx].ShipTypes[sti].Armament, g.Race[Race_0_idx].ShipTypes[sto].Armament)
	assert.Equal(t, g.ShipGroups[2].State(), g.ShipGroups[3].State())
	assert.Equal(t, g.ShipGroups[2].CargoType, g.ShipGroups[3].CargoType)
	assert.Equal(t, g.ShipGroups[2].Load, g.ShipGroups[3].Load)
	assert.Equal(t, g.ShipGroups[2].TechLevel(game.TechDrive), g.ShipGroups[3].TechLevel(game.TechDrive))
	assert.Equal(t, g.ShipGroups[2].TechLevel(game.TechWeapons), g.ShipGroups[3].TechLevel(game.TechWeapons))
	assert.Equal(t, g.ShipGroups[2].TechLevel(game.TechShields), g.ShipGroups[3].TechLevel(game.TechShields))
	assert.Equal(t, g.ShipGroups[2].TechLevel(game.TechCargo), g.ShipGroups[3].TechLevel(game.TechCargo))
	assert.Equal(t, g.ShipGroups[2].Destination, g.ShipGroups[3].Destination)
	assert.Equal(t, g.ShipGroups[2].StateInSpace, g.ShipGroups[3].StateInSpace)
	assert.Equal(t, g.ShipGroups[2].StateUpgrade, g.ShipGroups[3].StateUpgrade)
	assert.Equal(t, g.ShipGroups[3].OwnerID, g.Race[Race_1_idx].ID)
	assert.Equal(t, g.ShipGroups[3].TypeID, g.Race[Race_1_idx].ShipTypes[sti].ID)
	assert.Equal(t, g.ShipGroups[3].Number, uint(11))
	assert.Nil(t, g.ShipGroups[3].FleetID)

	assert.NoError(t, g.GiveawayGroup(Race_1.Name, Race_0.Name, 2, 11))
	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 := newGame()

	// 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].StateInSpace = &game.InSpace{
		Origin: 2,
		Range:  31.337,
	}

	// 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

	assert.Len(t, slices.Collect(g.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
	// 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(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, 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(), 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(), 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, 0),
		e.GenericErrorText(e.ErrInputCargoLoadNoSpaceLeft))
	assert.Len(t, slices.Collect(g.ListShipGroups(Race_0_idx)), 7)
}

func TestUnloadCargo(t *testing.T) {
	g := newGame()

	// 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
	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].StateInSpace = &game.InSpace{
		Origin: 2,
		Range:  31.337,
	}

	// 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))
}

func TestDisassembleGroup(t *testing.T) {
	g := newGame()

	// 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].StateInSpace = &game.InSpace{
		Origin: 2,
		Range:  31.337,
	}

	// 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)
}