From 55e07648cf99c12a7e37ceef95c8d1922f584d78 Mon Sep 17 00:00:00 2001
From: Josh Creek <8179928+jcreek@users.noreply.github.com>
Date: Tue, 1 Sep 2026 09:52:00 +0100
Subject: [PATCH] test: cover provider allocation reconciliation SQL

---
 server/store/allocator_sql_test.go | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/server/store/allocator_sql_test.go b/server/store/allocator_sql_test.go
index f9814c51..da7f54e1 100644
--- a/server/store/allocator_sql_test.go
+++ b/server/store/allocator_sql_test.go
@@ -9,9 +9,11 @@ import (
 
 func TestAllocatorSQLClaimsAndAuditsCompatibleReadyServers(t *testing.T) {
 	for query, fragments := range map[string][]string{
-		RegisterReadyServerSQL: {"game_servers", "ON CONFLICT", "state = 'READY'"},
-		ClaimReadyServerSQL:    {"state = 'READY'", "region = $1", "protocol_version = $3", "FOR UPDATE SKIP LOCKED", "ORDER BY server_id"},
-		InsertAllocationSQL:    {"allocations", "request_digest", "state", "ALLOCATED"},
+		RegisterReadyServerSQL:      {"game_servers", "ON CONFLICT", "state = 'READY'"},
+		ClaimReadyServerSQL:         {"state = 'READY'", "region = $1", "protocol_version = $3", "FOR UPDATE SKIP LOCKED", "ORDER BY server_id"},
+		InsertAllocationSQL:         {"allocations", "request_digest", "state", "ALLOCATED"},
+		ProviderServerClaimSQL:      {"state = 'READY'", "region = $2", "protocol_version = $4", "RETURNING"},
+		ServerAllocationConflictSQL: {"server_id = $1", "FOR UPDATE"},
 	} {
 		for _, fragment := range fragments {
 			if !contains(query, fragment) {