mirror of
https://github.com/jcreek/CosmicClash.git
synced 2026-09-10 16:04:04 +00:00
Josh Creek
70 lines
3.2 KiB
Go
70 lines
3.2 KiB
Go
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)
|
|
}
|
|
}
|