test: expand PostgreSQL adapter coverage

2026-09-11 08:23:45 +00:00 · 2026-09-01 08:59:39 +01:00
parent 5234390931
commit 414df530f0
2 changed files with 73 additions and 2 deletions
@@ -88,6 +88,77 @@ func TestPostgreSQLQueueAdapterAgainstRealDatabase(t *testing.T) {
 	}
 }

+func TestPostgreSQLQueueHeartbeatAndCancelAreRevisionFenced(t *testing.T) {
+	db := openIntegrationPostgres(t)
+	applyIntegrationMigrations(t, db)
+
+	now := time.Now().UTC().Truncate(time.Microsecond)
+	ctx := context.Background()
+	if _, err := db.ExecContext(ctx, `INSERT INTO identities (player_id, steam_id) VALUES ('heartbeat-player', 'heartbeat-steam')`); err != nil {
+		t.Fatal(err)
+	}
+	spec := domain.QueueSpec{Playlist: domain.Ranked, ClientBuild: "integration-build", ProtocolVersion: 1}
+	if _, err := CreateQueueTicket(ctx, db, "heartbeat-ticket", "heartbeat-player", "heartbeat-create-0001", spec, now); err != nil {
+		t.Fatal(err)
+	}
+	heartbeat, err := HeartbeatQueueTicket(ctx, db, "heartbeat-player", "heartbeat-ticket", "heartbeat-op-0000001", 0, now.Add(5*time.Second))
+	if err != nil {
+		t.Fatalf("heartbeat: %v", err)
+	}
+	if heartbeat.Revision != 1 || !heartbeat.ExpiresAt.Equal(now.Add(35*time.Second)) {
+		t.Fatalf("unexpected heartbeat result: %+v", heartbeat)
+	}
+	if _, err := HeartbeatQueueTicket(ctx, db, "heartbeat-player", "heartbeat-ticket", "heartbeat-op-0000002", 0, now.Add(6*time.Second)); err == nil {
+		t.Fatal("stale heartbeat revision was accepted")
+	}
+	cancelled, err := CancelQueueTicket(ctx, db, "heartbeat-player", "heartbeat-ticket", "heartbeat-op-0000003", 1, now.Add(7*time.Second))
+	if err != nil {
+		t.Fatalf("cancel: %v", err)
+	}
+	if cancelled.State != domain.Cancelled || cancelled.Revision != 2 {
+		t.Fatalf("unexpected cancellation result: %+v", cancelled)
+	}
+}
+
+func TestPostgreSQLAssignmentPersistenceIsPlayerScopedAndExpiryBound(t *testing.T) {
+	db := openIntegrationPostgres(t)
+	applyIntegrationMigrations(t, db)
+
+	now := time.Now().UTC().Truncate(time.Microsecond)
+	ctx := context.Background()
+	for _, player := range []string{"assignment-player", "assignment-other"} {
+		if _, err := db.ExecContext(ctx, `INSERT INTO identities (player_id, steam_id) VALUES ($1, $1)`, player); err != nil {
+			t.Fatal(err)
+		}
+	}
+	if _, err := db.ExecContext(ctx, `INSERT INTO queue_tickets (ticket_id, player_id, playlist, state, client_build, protocol_version, enqueued_at, expires_at) VALUES ('assignment-ticket', 'assignment-player', 'casual', 'ASSIGNED', 'integration-build', 1, $1, $2)`, now, now.Add(time.Minute)); err != nil {
+		t.Fatal(err)
+	}
+	if _, err := db.ExecContext(ctx, `INSERT INTO matches (match_id, playlist, state, region, protocol_version, server_id) VALUES ('assignment-match', 'casual', 'ASSIGNED', 'EU', 1, 'assignment-server')`); err != nil {
+		t.Fatal(err)
+	}
+	if _, err := db.ExecContext(ctx, `INSERT INTO match_participants (match_id, player_id, ticket_id, slot, team) VALUES ('assignment-match', 'assignment-player', 'assignment-ticket', 0, 0)`); err != nil {
+		t.Fatal(err)
+	}
+	assignment := DurableAssignment{MatchID: "assignment-match", PlayerID: "assignment-player", AllocationID: "allocation-1", ServerID: "assignment-server", Slot: 0, Region: "EU", ClientBuild: "integration-build", ProtocolVersion: 1, Transport: "enet", Endpoint: "127.0.0.1:7777", JoinAuthorisation: "join-token", ManifestDigest: []byte("manifest"), ExpiresAt: now.Add(time.Minute), Revision: 1}
+	if err := SaveAssignment(ctx, db, assignment); err != nil {
+		t.Fatalf("save assignment: %v", err)
+	}
+	got, err := GetAssignment(ctx, db, assignment.PlayerID, assignment.MatchID, now)
+	if err != nil {
+		t.Fatalf("recover assignment: %v", err)
+	}
+	if got.JoinAuthorisation != assignment.JoinAuthorisation || got.Slot != assignment.Slot {
+		t.Fatalf("assignment changed on round trip: %+v", got)
+	}
+	if _, err := GetAssignment(ctx, db, "assignment-other", assignment.MatchID, now); err == nil {
+		t.Fatal("non-owner recovered assignment")
+	}
+	if _, err := GetAssignment(ctx, db, assignment.PlayerID, assignment.MatchID, now.Add(2*time.Minute)); err == nil {
+		t.Fatal("expired assignment was recovered")
+	}
+}
+
 func TestPostgreSQLMigrationsAreForwardExecutable(t *testing.T) {
 	db := openIntegrationPostgres(t)
 	applyIntegrationMigrations(t, db)