feat(multiplayer): run leased allocator worker

2026-09-10 16:04:04 +00:00 · 2026-09-01 10:38:45 +01:00
parent 6f7d61eafb
commit 55c46f56ec
6 changed files with 255 additions and 6 deletions
@@ -0,0 +1,69 @@
+package allocator
+
+import (
+	"context"
+	"errors"
+	"reflect"
+	"testing"
+	"time"
+
+	"github.com/cosmic-clash/cosmic-clash/server/agones"
+	"github.com/cosmic-clash/cosmic-clash/server/domain"
+)
+
+type matchClaimSpy struct {
+	request domain.AllocationRequest
+	found   bool
+	err     error
+	bound   domain.Allocation
+	bindErr error
+}
+
+func (s *matchClaimSpy) ClaimAllocatingMatch(_ context.Context, _ time.Time) (domain.AllocationRequest, bool, error) {
+	return s.request, s.found, s.err
+}
+
+func (s *matchClaimSpy) BindAllocatedMatch(_ context.Context, allocation domain.Allocation) error {
+	s.bound = allocation
+	return s.bindErr
+}
+
+func TestWorkerClaimsAllocatesAndBindsDurably(t *testing.T) {
+	request := domain.AllocationRequest{AllocationID: "allocation-1", MatchID: "match-1", Region: "EU", Build: "build-1", Protocol: 1, Transport: "enet"}
+	claims := &matchClaimSpy{request: request, found: true}
+	provider := &providerSpy{result: agones.AllocatedServer{Allocation: domain.Allocation{AllocationID: request.AllocationID, MatchID: request.MatchID, ServerID: "server-1", Region: request.Region, Build: request.Build, Protocol: request.Protocol, Transport: request.Transport, State: domain.ServerAllocated}, Endpoint: "127.0.0.1:7777"}}
+	durable := &durableSpy{}
+	worker := Worker{Claims: claims, Service: Service{Provider: provider, Durable: durable, Now: func() time.Time { return time.Unix(1_000, 0) }}, Now: func() time.Time { return time.Unix(1_000, 0) }}
+	processed, err := worker.RunOnce(context.Background())
+	if err != nil || !processed || provider.calls != 1 || durable.calls != 1 || claims.bound.ServerID != "server-1" {
+		t.Fatalf("processed=%t err=%v provider=%d durable=%d bound=%+v", processed, err, provider.calls, durable.calls, claims.bound)
+	}
+}
+
+func TestWorkerRetainsClaimWhenProviderOutcomeIsAmbiguous(t *testing.T) {
+	request := domain.AllocationRequest{AllocationID: "allocation-1", MatchID: "match-1", Region: "EU", Build: "build-1", Protocol: 1, Transport: "enet"}
+	claims := &matchClaimSpy{request: request, found: true}
+	provider := &providerSpy{err: errors.New("provider timeout")}
+	worker := Worker{Claims: claims, Service: Service{Provider: provider, Durable: &durableSpy{}, Now: func() time.Time { return time.Unix(1_000, 0) }}, Now: func() time.Time { return time.Unix(1_000, 0) }}
+	processed, err := worker.RunOnce(context.Background())
+	if err == nil || !processed || claims.bound != (domain.Allocation{}) {
+		t.Fatalf("processed=%t err=%v bound=%+v", processed, err, claims.bound)
+	}
+}
+
+func TestWorkerDoesNothingWhenNoDurableMatchIsAvailable(t *testing.T) {
+	claims := &matchClaimSpy{}
+	worker := Worker{Claims: claims, Now: func() time.Time { return time.Unix(1_000, 0) }}
+	processed, err := worker.RunOnce(context.Background())
+	if err != nil || processed {
+		t.Fatalf("processed=%t err=%v", processed, err)
+	}
+}
+
+func TestAllocationLabelsMirrorFleetCompatibilityTuple(t *testing.T) {
+	got := AllocationLabels(domain.AllocationRequest{Region: "NA", Build: "build-4", Protocol: 12, Transport: "steam_sdr"})
+	want := map[string]string{"cosmic-clash.io/region": "NA", "cosmic-clash.io/build": "build-4", "cosmic-clash.io/protocol": "12", "cosmic-clash.io/transport": "steam_sdr"}
+	if !reflect.DeepEqual(got, want) {
+		t.Fatalf("labels=%v want=%v", got, want)
+	}
+}