From 3c4968ab8f6aac824dfbb7648887d1918b157799 Mon Sep 17 00:00:00 2001 From: Ilia Denisov Date: Fri, 6 Feb 2026 20:08:39 +0200 Subject: [PATCH] test: PlanetProductionDisplayName --- internal/controller/planet.go | 1 - internal/controller/planet_test.go | 22 ++++++++++++++++------ 2 files changed, 16 insertions(+), 7 deletions(-) diff --git a/internal/controller/planet.go b/internal/controller/planet.go index 2220a42..cacbdf3 100644 --- a/internal/controller/planet.go +++ b/internal/controller/planet.go @@ -125,7 +125,6 @@ func (c *Cache) PlanetProduction(ri int, number int, prod game.ProductionType, s return nil } -// TODO: test func (c *Cache) PlanetProductionDisplayName(pn uint) string { p := c.MustPlanet(pn) if !p.Owned() { diff --git a/internal/controller/planet_test.go b/internal/controller/planet_test.go index b3768bf..8696489 100644 --- a/internal/controller/planet_test.go +++ b/internal/controller/planet_test.go @@ -36,54 +36,64 @@ func TestRenamePlanet(t *testing.T) { 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)) + scienceName := "Drive_Shields" + assert.NoError(t, g.CreateScience(Race_0.Name, scienceName, 0.4, 0, 0.6, 0)) assert.Len(t, c.RaceScience(Race_0_idx), 1) scID := c.RaceScience(Race_0_idx)[0].ID + assert.Equal(t, "-", c.PlanetProductionDisplayName(3)) + 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.Equal(t, "Material", c.PlanetProductionDisplayName(R0_Planet_0_num)) 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.Equal(t, "Capital", c.PlanetProductionDisplayName(R0_Planet_0_num)) 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.Equal(t, "Weapons", c.PlanetProductionDisplayName(R0_Planet_0_num)) 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.Equal(t, "Cargo", c.PlanetProductionDisplayName(R0_Planet_0_num)) - assert.NoError(t, g.PlanetProduction(Race_0.Name, pn, "SHIELDS", first)) + assert.NoError(t, g.PlanetProduction(Race_0.Name, pn, "SHIELDS", scienceName)) 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.Equal(t, "Shields", c.PlanetProductionDisplayName(R0_Planet_0_num)) 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.Equal(t, "Drive", c.PlanetProductionDisplayName(R0_Planet_0_num)) - assert.NoError(t, g.PlanetProduction(Race_0.Name, pn, "Science", first)) + assert.NoError(t, g.PlanetProduction(Race_0.Name, pn, "Science", scienceName)) 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.Equal(t, scienceName, c.PlanetProductionDisplayName(R0_Planet_0_num)) 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) + st := c.MustShipClass(Race_0_idx, Race_0_Gunship) + assert.Equal(t, st.ID, *c.MustPlanet(R0_Planet_0_num).Production.SubjectID) + assert.Equal(t, st.Name, c.PlanetProductionDisplayName(R0_Planet_0_num)) pn = int(R0_Planet_2_num) assert.ErrorContains(t,