fs storage

2026-03-13 21:07:23 +02:00
parent 43039a79bf
commit 9ade76e21d
117 changed files with 1734 additions and 176 deletions
@@ -0,0 +1,107 @@
+package generator_test
+
+import (
+	"fmt"
+	"testing"
+
+	"galaxy/game/internal/generator"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGenerator(t *testing.T) {
+	maxPlayers := 30
+	for players := 10; players <= maxPlayers; players++ {
+		t.Run(fmt.Sprintf("%d_players", players), func(t *testing.T) {
+			var s generator.MapSetting
+			m, err := generator.Generate(func(ms *generator.MapSetting) {
+				ms.Players = uint32(players)
+				s = *ms
+			})
+			if err != nil {
+				t.Fatalf("generate: %s", err)
+				return
+			}
+			assert.Equal(t, players, len(m.HomePlanets), "hw-s count")
+			for hw := range m.HomePlanets {
+				assert.Equal(t, s.HWSize, m.HomePlanets[hw].HW.Size, "hw #%d: size", hw)
+				assert.Equal(t, s.HWResources, m.HomePlanets[hw].HW.Resources, "hw #%d: resources", hw)
+				assert.Equal(t, int(s.DWCount), len(m.HomePlanets[hw].DW), "hw #%d: dw-s count", hw)
+				for dw := range m.HomePlanets[hw].DW {
+					assert.Equal(t, s.DWSize, m.HomePlanets[hw].DW[dw].Size, "hw #%d dw #%d: size", hw, dw)
+					assert.Equal(t, s.DWResources, m.HomePlanets[hw].DW[dw].Resources, "hw #%d dw #%d: resources", hw, dw)
+					d := m.ShortDistance(m.HomePlanets[hw].HW.Position, m.HomePlanets[hw].DW[dw].Position)
+					assert.LessOrEqualf(t, float64(s.DWMinDistance), d, "distance: HW[%.04f,%04f] <-> DW[%.04f,%04f]",
+						m.HomePlanets[hw].HW.Position.X, m.HomePlanets[hw].HW.Position.Y,
+						m.HomePlanets[hw].DW[dw].Position.X, m.HomePlanets[hw].DW[dw].Position.Y)
+					assert.GreaterOrEqualf(t, float64(s.DWMaxDistance), d, "distance: HW[%.04f,%04f] <-> DW[%.04f,%04f]",
+						m.HomePlanets[hw].HW.Position.X, m.HomePlanets[hw].HW.Position.Y,
+						m.HomePlanets[hw].DW[dw].Position.X, m.HomePlanets[hw].DW[dw].Position.Y)
+				}
+			}
+			assert.LessOrEqualf(t, int(s.NobodysPlanets()), len(m.FreePlanets), "free planets clount")
+			freePlanetCount := make(map[generator.PlanetClass]int)
+			for fp := range m.FreePlanets {
+				ps := planetSettings(t, m.FreePlanets[fp].PlanetClass, s)
+				testPlanetParameters(t, ps, m.FreePlanets[fp])
+				if v, ok := freePlanetCount[m.FreePlanets[fp].PlanetClass]; !ok {
+					freePlanetCount[m.FreePlanets[fp].PlanetClass] = 1
+				} else {
+					freePlanetCount[m.FreePlanets[fp].PlanetClass] = v + 1
+				}
+				if ps.MinDistanceHW > 0 {
+					for hw := range m.HomePlanets {
+						d := m.ShortDistance(m.HomePlanets[hw].HW.Position, m.FreePlanets[fp].Position)
+						assert.LessOrEqualf(t, float64(ps.MinDistanceHW), d, "distance: HW[%.04f,%04f] <-> planet_class=%v[%.04f,%04f]",
+							m.HomePlanets[hw].HW.Position.X, m.HomePlanets[hw].HW.Position.Y,
+							m.FreePlanets[fp].PlanetClass,
+							m.FreePlanets[fp].Position.X, m.FreePlanets[fp].Position.Y)
+					}
+				}
+			}
+			for pc, num := range freePlanetCount {
+				ps := planetSettings(t, pc, s)
+				maxNum := ps.Number(s.NobodysPlanets())
+				assert.Equalf(t, num, maxNum, "planet_class=%v ratio=%f of total %d", pc, ps.Ratio, s.NobodysPlanets())
+			}
+		})
+	}
+}
+
+func testPlanetParameters(t *testing.T, s generator.PlanetSetting, p generator.Planet) {
+	assert.LessOrEqualf(t, s.MinResource, p.Resources, "planet class=%s min resources", p.PlanetClass)
+	assert.GreaterOrEqualf(t, s.MaxResource, p.Resources, "planet class=%s max resources", p.PlanetClass)
+	assert.LessOrEqualf(t, s.MinSize, p.Size, "planet class=%s min size", p.PlanetClass)
+	assert.GreaterOrEqualf(t, s.MaxSize, p.Size, "planet class=%s max size", p.PlanetClass)
+}
+
+func planetSettings(t *testing.T, pc generator.PlanetClass, s generator.MapSetting) generator.PlanetSetting {
+	switch pc {
+	case generator.PlanetClassGiant:
+		return s.GiantPlanets
+	case generator.PlanetClassBig:
+		return s.BigPlanets
+	case generator.PlanetClassNormal:
+		return s.NormalPlanets
+	case generator.PlanetClassRich:
+		return s.RichPlanets
+	case generator.PlanetClassAsterioid:
+		return s.Asterioids
+	default:
+		assert.FailNow(t, "unexpected planet class: %s", pc)
+		return generator.PlanetSetting{}
+	}
+}
+
+func BenchmarkGenerator(b *testing.B) {
+	i := 0
+	for b.Loop() {
+		i++
+		b.Run(fmt.Sprintf("instance #%02d", i), func(b *testing.B) {
+			_, err := generator.Generate()
+			if err != nil {
+				b.Error(err)
+			}
+		})
+	}
+}