test: cover provider allocation reconciliation SQL

This commit is contained in:
Josh Creek
2026-09-01 09:52:00 +01:00
parent 0023bdab6e
commit 55e07648cf
+5 -3
View File
@@ -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) {