feat: runtime manager

2026-04-28 20:39:18 +02:00
parent e0a99b346b
commit a7cee15115
289 changed files with 45660 additions and 2207 deletions
@@ -0,0 +1,157 @@
+package healthsnapshotstore_test
+
+import (
+	"context"
+	"encoding/json"
+	"testing"
+	"time"
+
+	"galaxy/rtmanager/internal/adapters/postgres/healthsnapshotstore"
+	"galaxy/rtmanager/internal/adapters/postgres/internal/pgtest"
+	"galaxy/rtmanager/internal/domain/health"
+	"galaxy/rtmanager/internal/domain/runtime"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestMain(m *testing.M) { pgtest.RunMain(m) }
+
+func newStore(t *testing.T) *healthsnapshotstore.Store {
+	t.Helper()
+	pgtest.TruncateAll(t)
+	store, err := healthsnapshotstore.New(healthsnapshotstore.Config{
+		DB:               pgtest.Ensure(t).Pool(),
+		OperationTimeout: pgtest.OperationTimeout,
+	})
+	require.NoError(t, err)
+	return store
+}
+
+func probeFailedSnapshot(gameID string, observedAt time.Time) health.HealthSnapshot {
+	return health.HealthSnapshot{
+		GameID:      gameID,
+		ContainerID: "container-1",
+		Status:      health.SnapshotStatusProbeFailed,
+		Source:      health.SnapshotSourceProbe,
+		Details:     json.RawMessage(`{"consecutive_failures":3,"last_status":503,"last_error":"timeout"}`),
+		ObservedAt:  observedAt,
+	}
+}
+
+func TestUpsertAndGetRoundTrip(t *testing.T) {
+	ctx := context.Background()
+	store := newStore(t)
+
+	snapshot := probeFailedSnapshot("game-001",
+		time.Date(2026, 4, 27, 12, 0, 0, 0, time.UTC))
+	require.NoError(t, store.Upsert(ctx, snapshot))
+
+	got, err := store.Get(ctx, "game-001")
+	require.NoError(t, err)
+	assert.Equal(t, snapshot.GameID, got.GameID)
+	assert.Equal(t, snapshot.ContainerID, got.ContainerID)
+	assert.Equal(t, snapshot.Status, got.Status)
+	assert.Equal(t, snapshot.Source, got.Source)
+	assert.JSONEq(t, string(snapshot.Details), string(got.Details))
+	assert.True(t, snapshot.ObservedAt.Equal(got.ObservedAt))
+	assert.Equal(t, time.UTC, got.ObservedAt.Location())
+}
+
+func TestUpsertOverwritesPriorSnapshot(t *testing.T) {
+	ctx := context.Background()
+	store := newStore(t)
+
+	first := probeFailedSnapshot("game-001",
+		time.Date(2026, 4, 27, 12, 0, 0, 0, time.UTC))
+	require.NoError(t, store.Upsert(ctx, first))
+
+	second := health.HealthSnapshot{
+		GameID:      "game-001",
+		ContainerID: "container-2",
+		Status:      health.SnapshotStatusHealthy,
+		Source:      health.SnapshotSourceInspect,
+		Details:     json.RawMessage(`{"restart_count":0,"state":"running"}`),
+		ObservedAt:  first.ObservedAt.Add(time.Minute),
+	}
+	require.NoError(t, store.Upsert(ctx, second))
+
+	got, err := store.Get(ctx, "game-001")
+	require.NoError(t, err)
+	assert.Equal(t, "container-2", got.ContainerID)
+	assert.Equal(t, health.SnapshotStatusHealthy, got.Status)
+	assert.Equal(t, health.SnapshotSourceInspect, got.Source)
+	assert.JSONEq(t, string(second.Details), string(got.Details))
+	assert.True(t, second.ObservedAt.Equal(got.ObservedAt))
+}
+
+func TestGetReturnsNotFound(t *testing.T) {
+	ctx := context.Background()
+	store := newStore(t)
+
+	_, err := store.Get(ctx, "game-missing")
+	require.ErrorIs(t, err, runtime.ErrNotFound)
+}
+
+func TestUpsertEmptyDetailsRoundTripsAsEmptyObject(t *testing.T) {
+	ctx := context.Background()
+	store := newStore(t)
+
+	snapshot := probeFailedSnapshot("game-001",
+		time.Date(2026, 4, 27, 12, 0, 0, 0, time.UTC))
+	snapshot.Details = nil
+	require.NoError(t, store.Upsert(ctx, snapshot))
+
+	got, err := store.Get(ctx, "game-001")
+	require.NoError(t, err)
+	assert.JSONEq(t, "{}", string(got.Details),
+		"empty json.RawMessage must round-trip as the SQL default {}, got %q",
+		string(got.Details))
+}
+
+func TestUpsertValidatesSnapshot(t *testing.T) {
+	ctx := context.Background()
+	store := newStore(t)
+
+	tests := []struct {
+		name   string
+		mutate func(*health.HealthSnapshot)
+	}{
+		{"empty game id", func(s *health.HealthSnapshot) { s.GameID = "" }},
+		{"unknown status", func(s *health.HealthSnapshot) { s.Status = "exotic" }},
+		{"unknown source", func(s *health.HealthSnapshot) { s.Source = "exotic" }},
+		{"zero observed at", func(s *health.HealthSnapshot) { s.ObservedAt = time.Time{} }},
+		{"invalid json details", func(s *health.HealthSnapshot) {
+			s.Details = json.RawMessage("not json")
+		}},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			snapshot := probeFailedSnapshot("game-001",
+				time.Date(2026, 4, 27, 12, 0, 0, 0, time.UTC))
+			tt.mutate(&snapshot)
+			err := store.Upsert(ctx, snapshot)
+			require.Error(t, err)
+		})
+	}
+}
+
+func TestGetRejectsEmptyGameID(t *testing.T) {
+	ctx := context.Background()
+	store := newStore(t)
+
+	_, err := store.Get(ctx, "")
+	require.Error(t, err)
+}
+
+func TestNewRejectsNilDB(t *testing.T) {
+	_, err := healthsnapshotstore.New(healthsnapshotstore.Config{OperationTimeout: time.Second})
+	require.Error(t, err)
+}
+
+func TestNewRejectsNonPositiveTimeout(t *testing.T) {
+	_, err := healthsnapshotstore.New(healthsnapshotstore.Config{
+		DB: pgtest.Ensure(t).Pool(),
+	})
+	require.Error(t, err)
+}