refactor: game funcs moved to controller

2026-01-15 14:42:04 +02:00
parent fe8a8d4150
commit 16aba8435d
47 changed files with 1023 additions and 3093 deletions
@@ -0,0 +1,120 @@
+package controller_test
+
+import (
+	"testing"
+
+	e "github.com/iliadenisov/galaxy/internal/error"
+	"github.com/iliadenisov/galaxy/internal/model/game"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestRenamePlanet(t *testing.T) {
+	c, g := newCache()
+
+	assert.Equal(t, "Planet_0", c.MustPlanet(R0_Planet_0_num).Name)
+	assert.NoError(t, g.RenamePlanet(Race_0.Name, int(R0_Planet_0_num), "Home_World"))
+	assert.Equal(t, "Home_World", c.MustPlanet(R0_Planet_0_num).Name)
+
+	assert.ErrorContains(t,
+		g.RenamePlanet("UnknownRace", int(R0_Planet_0_num), "Home_World"),
+		e.GenericErrorText(e.ErrInputUnknownRace))
+	assert.ErrorContains(t,
+		g.RenamePlanet(Race_0.Name, -1, "Home_World"),
+		e.GenericErrorText(e.ErrInputPlanetNumber))
+	assert.ErrorContains(t,
+		g.RenamePlanet(Race_0.Name, 500, "Home_World"),
+		e.GenericErrorText(e.ErrInputEntityNotExists))
+	assert.ErrorContains(t,
+		g.RenamePlanet(Race_0.Name, int(R1_Planet_1_num), "Home_World"),
+		e.GenericErrorText(e.ErrInputEntityNotOwned))
+}
+
+func TestPlanetProduction(t *testing.T) {
+	c, g := newCache()
+
+	first := "Drive_Shields"
+	assert.NoError(t, g.CreateScience(Race_0.Name, first, 0.4, 0, 0.6, 0))
+	assert.Len(t, c.RaceScience(Race_0_idx), 1)
+	scID := c.RaceScience(Race_0_idx)[0].ID
+
+	pn := int(R0_Planet_0_num)
+	assert.NoError(t, g.PlanetProduction(Race_0.Name, pn, "MAT", ""))
+	assert.Equal(t, game.ProductionMaterial, c.MustPlanet(R0_Planet_0_num).Production.Type)
+	assert.Nil(t, c.MustPlanet(R0_Planet_0_num).Production.SubjectID)
+	assert.Nil(t, c.MustPlanet(R0_Planet_0_num).Production.Progress)
+
+	assert.NoError(t, g.PlanetProduction(Race_0.Name, pn, "CAP", ""))
+	assert.Equal(t, game.ProductionCapital, c.MustPlanet(R0_Planet_0_num).Production.Type)
+	assert.Nil(t, c.MustPlanet(R0_Planet_0_num).Production.SubjectID)
+	assert.Nil(t, c.MustPlanet(R0_Planet_0_num).Production.Progress)
+
+	assert.NoError(t, g.PlanetProduction(Race_0.Name, pn, "Weapons", "500"))
+	assert.Equal(t, game.ResearchWeapons, c.MustPlanet(R0_Planet_0_num).Production.Type)
+	assert.Nil(t, c.MustPlanet(R0_Planet_0_num).Production.SubjectID)
+	assert.Nil(t, c.MustPlanet(R0_Planet_0_num).Production.Progress)
+
+	assert.NoError(t, g.PlanetProduction(Race_0.Name, pn, "cargo", ""))
+	assert.Equal(t, game.ResearchCargo, c.MustPlanet(R0_Planet_0_num).Production.Type)
+	assert.Nil(t, c.MustPlanet(R0_Planet_0_num).Production.SubjectID)
+	assert.Nil(t, c.MustPlanet(R0_Planet_0_num).Production.Progress)
+
+	assert.NoError(t, g.PlanetProduction(Race_0.Name, pn, "SHIELDS", first))
+	assert.Equal(t, game.ResearchShields, c.MustPlanet(R0_Planet_0_num).Production.Type)
+	assert.Nil(t, c.MustPlanet(R0_Planet_0_num).Production.SubjectID)
+	assert.Nil(t, c.MustPlanet(R0_Planet_0_num).Production.Progress)
+
+	assert.NoError(t, g.PlanetProduction(Race_0.Name, pn, "DrivE", ""))
+	assert.Equal(t, game.ResearchDrive, c.MustPlanet(R0_Planet_0_num).Production.Type)
+	assert.Nil(t, c.MustPlanet(R0_Planet_0_num).Production.SubjectID)
+	assert.Nil(t, c.MustPlanet(R0_Planet_0_num).Production.Progress)
+
+	assert.NoError(t, g.PlanetProduction(Race_0.Name, pn, "Science", first))
+	assert.Equal(t, game.ResearchScience, c.MustPlanet(R0_Planet_0_num).Production.Type)
+	assert.Nil(t, c.MustPlanet(R0_Planet_0_num).Production.Progress)
+	assert.NotNil(t, c.MustPlanet(R0_Planet_0_num).Production.SubjectID)
+	assert.Equal(t, scID, *c.MustPlanet(R0_Planet_0_num).Production.SubjectID)
+
+	assert.NoError(t, g.PlanetProduction(Race_0.Name, pn, "SHIP", Race_0_Gunship))
+	assert.Equal(t, game.ProductionShip, c.MustPlanet(R0_Planet_0_num).Production.Type)
+	assert.NotNil(t, c.MustPlanet(R0_Planet_0_num).Production.Progress)
+	assert.NotNil(t, c.MustPlanet(R0_Planet_0_num).Production.SubjectID)
+	stID := c.MustShipClass(Race_0_idx, Race_0_Gunship).ID
+	assert.Equal(t, stID, *c.MustPlanet(R0_Planet_0_num).Production.SubjectID)
+
+	pn = int(R0_Planet_2_num)
+	assert.ErrorContains(t,
+		g.PlanetProduction("UnknownRace", pn, "DRIVE", ""),
+		e.GenericErrorText(e.ErrInputUnknownRace))
+	assert.ErrorContains(t,
+		g.PlanetProduction(Race_0.Name, pn, "Hyperdrive", ""),
+		e.GenericErrorText(e.ErrInputProductionInvalid))
+	assert.ErrorContains(t,
+		g.PlanetProduction(Race_0.Name, -1, "DRIVE", ""),
+		e.GenericErrorText(e.ErrInputPlanetNumber))
+	assert.ErrorContains(t,
+		g.PlanetProduction(Race_0.Name, 500, "DRIVE", ""),
+		e.GenericErrorText(e.ErrInputEntityNotExists))
+	assert.ErrorContains(t,
+		g.PlanetProduction(Race_0.Name, int(R1_Planet_1_num), "DRIVE", ""),
+		e.GenericErrorText(e.ErrInputEntityNotOwned))
+	assert.ErrorContains(t,
+		g.PlanetProduction(Race_0.Name, pn, "Science", ""),
+		e.GenericErrorText(e.ErrInputEntityTypeNameInvalid))
+	assert.ErrorContains(t,
+		g.PlanetProduction(Race_0.Name, pn, "SHIP", ""),
+		e.GenericErrorText(e.ErrInputEntityTypeNameInvalid))
+	assert.ErrorContains(t,
+		g.PlanetProduction(Race_0.Name, pn, "Science", "Winning"),
+		e.GenericErrorText(e.ErrInputEntityNotExists))
+	assert.ErrorContains(t,
+		g.PlanetProduction(Race_0.Name, pn, "SHIP", "Drone"),
+		e.GenericErrorText(e.ErrInputEntityNotExists))
+}
+
+func TestPlanetProductionCapacity(t *testing.T) {
+	c, _ := newCache()
+	assert.NoError(t, c.CreateShips(Race_0_idx, ShipType_Cruiser, R0_Planet_0_num, 1))
+	assert.Equal(t, 100., c.PlanetProductionCapacity(R0_Planet_0_num))
+	c.UpgradeShipGroup(0, game.TechDrive, 1.6)
+	assert.Equal(t, 53.125, c.PlanetProductionCapacity(R0_Planet_0_num))
+}