tests: produce on planets, unload on routes

2026-01-23 00:28:23 +02:00
parent 9825e05c0e
commit 812e0d4afd
15 changed files with 522 additions and 103 deletions
@@ -4,6 +4,7 @@ import (
 	"slices"
 	"testing"

+	"github.com/iliadenisov/galaxy/internal/controller"
 	e "github.com/iliadenisov/galaxy/internal/error"
 	"github.com/iliadenisov/galaxy/internal/model/game"

@@ -91,7 +92,7 @@ func TestRemoveRoute(t *testing.T) {
 		e.GenericErrorText(e.ErrInputEntityNotOwned))
 }

-func TestListRouteEligibleGroupIds(t *testing.T) {
+func TestListRoutedSendGroupIds(t *testing.T) {
 	c, g := newCache()

 	// 1: idx = 0 / Ready to load
@@ -120,7 +121,7 @@ func TestListRouteEligibleGroupIds(t *testing.T) {
 	assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 10))
 	assert.NoError(t, g.JoinShipGroupToFleet(Race_0.Name, "Fleet", 5, 0))

-	planet_0_groups := slices.Collect(c.ListRouteEligibleGroupIds(0))
+	planet_0_groups := slices.Collect(c.ListRoutedSendGroupIds(0))
 	assert.Len(t, planet_0_groups, 1)
 	for _, i := range planet_0_groups {
 		sg := c.ShipGroup(i)
@@ -254,3 +255,164 @@ func TestEnrouteGroups_LaunchOrder(t *testing.T) {
 	assert.Nil(t, c.ShipGroup(sgi).CargoType)
 	assert.Equal(t, uint(1), c.ShipGroup(sgi).Number)
 }
+
+func TestListRoutedUnloadShipGroupIds(t *testing.T) {
+	c, g := newCache()
+
+	// 1: idx = 0 / Empty cargo
+	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 = 0.
+
+	// 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 = &game.InSpace{
+		Origin: 2,
+		Range:  31.337,
+	}
+
+	// 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 / Part of the Fleet
+	assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 10))
+	assert.NoError(t, g.JoinShipGroupToFleet(Race_0.Name, "Fleet", 5, 0))
+
+	assert.NoError(t, g.SetRoute(Race_0.Name, "COL", R0_Planet_0_num, R0_Planet_2_num))
+	for _, rt := range game.RouteTypeSet {
+		groups := slices.Collect(c.ListRoutedUnloadShipGroupIds(R0_Planet_2_num, rt))
+		assert.Len(t, groups, 0, "route: %v", rt)
+		groups = slices.Collect(c.ListRoutedUnloadShipGroupIds(R0_Planet_0_num, rt))
+		assert.Len(t, groups, 0, "route: %v", rt)
+	}
+
+	// double route from different planets - must not double group ids
+	assert.NoError(t, g.SetRoute(Race_0.Name, "COL", R0_Planet_2_num, R0_Planet_0_num))
+	assert.NoError(t, g.SetRoute(Race_1.Name, "COL", R1_Planet_1_num, R0_Planet_0_num))
+
+	// 6: idx = 5 / loaded with CAP
+	assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 11))
+	c.ShipGroup(5).CargoType = game.CargoCapital.Ref()
+	c.ShipGroup(5).Load = 1.234
+
+	groups := slices.Collect(c.ListRoutedUnloadShipGroupIds(R0_Planet_0_num, game.RouteColonist))
+	assert.Len(t, groups, 1)
+	for _, sgi := range groups {
+		assert.Greater(t, c.ShipGroup(sgi).Load, 0.)
+		assert.Equal(t, game.StateInOrbit, c.ShipGroup(sgi).State())
+		assert.Equal(t, R0_Planet_0_num, c.ShipGroup(sgi).Destination)
+	}
+
+	groups = slices.Collect(c.ListRoutedUnloadShipGroupIds(R0_Planet_0_num, game.RouteMaterial))
+	assert.Len(t, groups, 0)
+
+	groups = slices.Collect(c.ListRoutedUnloadShipGroupIds(R0_Planet_0_num, game.RouteCapital))
+	assert.Len(t, groups, 0)
+}
+
+func TestMaxOrRandomLoadId(t *testing.T) {
+	IDtoLoad := make(map[int]float64)
+
+	assert.Panics(t, func() { controller.MaxOrRandomLoadId(IDtoLoad) })
+	IDtoLoad[1] = 100.
+	assert.Panics(t, func() { controller.MaxOrRandomLoadId(IDtoLoad) })
+
+	IDtoLoad[5] = 100.001
+	assert.Equal(t, 5, controller.MaxOrRandomLoadId(IDtoLoad))
+
+	IDtoLoad[3] = 100.
+	assert.NotContains(t, []int{1, 3}, controller.MaxOrRandomLoadId(IDtoLoad))
+
+	IDtoLoad[7] = 100.001
+	rndCount := make(map[int]int)
+	for range 100 {
+		id := controller.MaxOrRandomLoadId(IDtoLoad)
+		assert.NotContains(t, []int{1, 3}, id)
+		assert.Contains(t, []int{5, 7}, id)
+		rndCount[id]++
+	}
+	assert.Greater(t, rndCount[5], 10)
+	assert.Greater(t, rndCount[7], 10)
+}
+
+func TestSelectColUnloadGroup(t *testing.T) {
+	c, g := newCache()
+
+	assert.NoError(t, g.SetRoute(Race_0.Name, "COL", R0_Planet_2_num, R0_Planet_0_num))
+	assert.NoError(t, g.SetRoute(Race_1.Name, "COL", R1_Planet_1_num, R0_Planet_0_num))
+
+	// 1: idx = 0 / Loaded COL
+	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 = 7.
+
+	// 2: idx = 1 / Loaded COL
+	assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 5))
+	c.ShipGroup(1).CargoType = game.CargoColonist.Ref()
+	c.ShipGroup(1).Load = 5.
+
+	groups := slices.Collect(c.ListRoutedUnloadShipGroupIds(R0_Planet_0_num, game.RouteColonist))
+	assert.Len(t, groups, 2)
+	unloadGroups := slices.Collect(c.SelectColUnloadGroup(groups))
+	assert.ElementsMatch(t, groups, unloadGroups)
+
+	// 3: idx = 2 / Loaded COL - another race, winner
+	assert.NoError(t, c.CreateShips(Race_1_idx, Race_1_Freighter, R1_Planet_1_num, 10))
+	c.ShipGroup(2).Destination = R0_Planet_0_num
+	c.ShipGroup(2).CargoType = game.CargoColonist.Ref()
+	c.ShipGroup(2).Load = 12.1
+
+	groups = slices.Collect(c.ListRoutedUnloadShipGroupIds(R0_Planet_0_num, game.RouteColonist))
+	assert.Len(t, groups, 3)
+	unloadGroups = slices.Collect(c.SelectColUnloadGroup(groups))
+	assert.Equal(t, 2, unloadGroups[0])
+}
+
+func TestTurnUnloadEnroutedGroups(t *testing.T) {
+	c, g := newCache()
+
+	assert.NoError(t, g.SetRoute(Race_0.Name, "MAT", R0_Planet_2_num, R0_Planet_0_num))
+	assert.NoError(t, g.SetRoute(Race_0.Name, "CAP", R0_Planet_2_num, R0_Planet_0_num))
+
+	assert.NoError(t, g.SetRoute(Race_1.Name, "COL", R1_Planet_1_num, R0_Planet_0_num))
+	assert.NoError(t, g.SetRoute(Race_1.Name, "COL", R1_Planet_1_num, Uninhabited_Planet_4_num))
+
+	// 1: idx = 0 / Loaded MAT
+	assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 10))
+	c.ShipGroup(0).CargoType = game.CargoMaterial.Ref()
+	c.ShipGroup(0).Load = 222.
+
+	// 2: idx = 1 / Loaded CAP
+	assert.NoError(t, c.CreateShips(Race_0_idx, Race_0_Freighter, R0_Planet_0_num, 10))
+	c.ShipGroup(1).CargoType = game.CargoCapital.Ref()
+	c.ShipGroup(1).Load = 11.
+
+	// 3: idx = 2 / Loaded COL - on empty planet
+	assert.NoError(t, c.CreateShips(Race_1_idx, Race_1_Freighter, R1_Planet_1_num, 10))
+	c.ShipGroup(2).Destination = Uninhabited_Planet_4_num
+	c.ShipGroup(2).CargoType = game.CargoColonist.Ref()
+	c.ShipGroup(2).Load = 12.1
+
+	// 4: idx = 3 / Loaded COL - on inhabited planet
+	assert.NoError(t, c.CreateShips(Race_1_idx, Race_1_Freighter, R1_Planet_1_num, 10))
+	c.ShipGroup(3).Destination = R0_Planet_0_num
+	c.ShipGroup(3).CargoType = game.CargoColonist.Ref()
+	c.ShipGroup(3).Load = 17.3
+
+	c.TurnUnloadEnroutedGroups()
+
+	assert.Equal(t, 0., c.ShipGroup(0).Load)
+	assert.Equal(t, 222., c.MustPlanet(R0_Planet_0_num).Material)
+	assert.Equal(t, 0., c.ShipGroup(1).Load)
+	assert.Equal(t, 11., c.MustPlanet(R0_Planet_0_num).Capital)
+	assert.Equal(t, 0., c.ShipGroup(2).Load)
+	assert.Equal(t, 96.8, c.MustPlanet(Uninhabited_Planet_4_num).Population)
+	assert.Equal(t, Race_1_ID, c.MustPlanet(Uninhabited_Planet_4_num).Owner)
+	assert.Equal(t, game.ProductionCapital, c.MustPlanet(Uninhabited_Planet_4_num).Production.Type)
+	assert.Equal(t, 17.3, c.ShipGroup(3).Load)
+}