Files
x0gp/server/internal/stats/stats_test.go
T
x0gp 978d36c505 feat(server): initial implementation
This commit captures the full server code accumulated across
several development sessions. It includes the following layers,
applied in roughly this order during development:

1. Base infrastructure
   - cmd/poc-server HTTP+WebSocket server, /health, /stats
   - internal/transport wire types (Envelope + all message types)
   - internal/catalog, internal/realtime, internal/stats
   - internal/storage/postgres with migrations 001-004
   - .env, .env.example, docker-compose, Dockerfile, scripts/

2. Swagger documentation
   - go get github.com/swaggo/http-swagger
   - swag init produces docs/docs.go
   - /swagger/index.html UI, /swagger/doc.json spec
   - annotations on health/stats/ws and catalog handlers

3. Races API
   - internal/races/{store,service,keyset,types}.go
   - migration 005_races.sql: finished_races, race_plans, race_queue
   - GET /api/races with keyset pagination, status filter
   - GET /api/races/upcoming
   - POST /api/races/queue/join, GET, DELETE
   - POST/GET /api/races/plans, DELETE /{id}
   - lobby.RaceMeta simplification (no host_id/host_name)

4. Races seeder
   - internal/races/seed with deterministic generator
   - --seed-races / --reset CLI flags in main.go
   - 30 finished + 5 live + 5 plans + 4 queue entries

5. Drivers and clans
   - internal/drivers, internal/clans (CRUD, validation)
   - migration 008_drivers_clans.sql
   - /api/clans and /api/drivers endpoints
   - 3-letter uppercase nickname/tag validation
   - lobby.DriverMeta: Nickname, AvatarURL, ClanID, ClanTag

6. Podium
   - internal/transport.RacePodiumEntry
   - lobby.SetDriverProfile for in-memory metadata sync
   - migration 007_podium.sql (podium JSONB column)
   - seeder populates top-3 per finished race

7. Live races persistence
   - migration 009_live_persistence.sql: live_races,
     live_race_drivers, lobby_drivers
   - internal/races/live_store.go: LiveStore with write-side
     mirror for lobby.Service mutations
   - Service.collectLive and Upcoming read from Postgres
   - main.go RestoreFromDB rehydrates lobby on startup

8. Unify live + finished into one races table
   - migration 010_unify_races.sql: rename finished_races to
     races, expand status CHECK, merge live rows
   - PgStore now hosts both write paths; LiveStore is a
     thin facade implementing lobby.Persistence
   - seeder resetAll drops only finished/cancelled rows and
     race_plans / race_queue / lobby_drivers

Each layer is consistent on its own; cross-layer changes are
visible in the file history. A future refactor may split this
commit into the per-stage boundaries listed above.
2026-06-22 22:01:09 +04:00

109 lines
2.8 KiB
Go

package stats
import (
"testing"
"time"
)
func TestConnectDisconnectCounts(t *testing.T) {
c := NewCollector()
c.Start()
c.OnConnect("d1", "Alice")
c.OnConnect("d2", "Bob")
c.OnDisconnect("d1", 5000)
s := c.Server()
if s.CurrentClients != 1 {
t.Errorf("CurrentClients: got %d", s.CurrentClients)
}
if s.PeakClients != 2 {
t.Errorf("PeakClients: got %d", s.PeakClients)
}
if s.TotalConnections != 2 {
t.Errorf("TotalConnections: got %d", s.TotalConnections)
}
if s.TotalDisconnects != 1 {
t.Errorf("TotalDisconnects: got %d", s.TotalDisconnects)
}
if s.UptimeMs < 0 {
t.Errorf("UptimeMs negative: %d", s.UptimeMs)
}
}
func TestLapAndDriverProfile(t *testing.T) {
c := NewCollector()
c.Start()
c.OnConnect("d1", "Alice")
c.OnLapRecorded(LapRecord{RaceID: "r1", DriverID: "d1", LapMs: 12_345, TSMs: time.Now().UnixMilli()})
c.OnLapRecorded(LapRecord{RaceID: "r1", DriverID: "d1", LapMs: 11_500, TSMs: time.Now().UnixMilli()})
c.OnLapRecorded(LapRecord{RaceID: "r1", DriverID: "d1", LapMs: 13_000, TSMs: time.Now().UnixMilli()})
d := c.Driver("d1", "Alice")
if d.TotalLaps != 3 {
t.Errorf("TotalLaps: got %d", d.TotalLaps)
}
if d.BestLapMs != 11_500 {
t.Errorf("BestLapMs: got %d", d.BestLapMs)
}
if d.LastLapMs != 13_000 {
t.Errorf("LastLapMs: got %d", d.LastLapMs)
}
}
func TestRaceFinishedUpdatesProfile(t *testing.T) {
c := NewCollector()
c.Start()
c.OnConnect("d1", "Alice")
c.OnConnect("d2", "Bob")
c.OnRaceStarted([]string{"d1", "d2"})
c.OnRaceFinished(RaceResult{RaceID: "r1", DriverID: "d1", Position: 1, TotalLaps: 5, BestLapMs: 11_000, TotalTimeMs: 60_000})
c.OnRaceFinished(RaceResult{RaceID: "r1", DriverID: "d2", Position: 2, TotalLaps: 5, BestLapMs: 11_500, TotalTimeMs: 60_500})
d1 := c.Driver("d1", "")
if d1.Wins != 1 {
t.Errorf("Wins: got %d", d1.Wins)
}
if d1.BestLapMs != 11_000 {
t.Errorf("BestLapMs: got %d", d1.BestLapMs)
}
d2 := c.Driver("d2", "")
if d2.Wins != 0 || d2.Podiums != 1 {
t.Errorf("Podiums: got %d", d2.Podiums)
}
s := c.Server()
if s.TotalRacesStarted != 1 || s.TotalRacesFinished != 2 {
t.Errorf("counters: started=%d finished=%d", s.TotalRacesStarted, s.TotalRacesFinished)
}
}
func TestRTTWindowAverage(t *testing.T) {
c := NewCollector()
c.Start()
for _, v := range []int64{10, 20, 30, 40, 50} {
c.OnRTT(v)
}
s := c.Server()
if s.AvgRTTMs != 30 {
t.Errorf("AvgRTTMs: got %v want 30", s.AvgRTTMs)
}
if len(s.RecentRTTs) != 5 {
t.Errorf("RecentRTTs len: got %d", len(s.RecentRTTs))
}
}
func TestRecentLapsOrder(t *testing.T) {
c := NewCollector()
c.Start()
for i := 0; i < 5; i++ {
c.OnLapRecorded(LapRecord{RaceID: "r1", DriverID: "d1", LapMs: int64(10_000 + i), TSMs: int64(i)})
}
laps := c.RecentLaps(3)
if len(laps) != 3 {
t.Fatalf("got %d laps", len(laps))
}
if laps[0].TSMs < laps[1].TSMs {
t.Errorf("expected newest first, got order: %v", laps)
}
}