feat: game lobby service
This commit is contained in:
@@ -0,0 +1,635 @@
|
||||
// Package lobbynotification_test exercises Lobby's notification-intent
|
||||
// publication boundary by booting Lobby + the real User Service against a
|
||||
// Redis container and asserting on the contents of `notification:intents`.
|
||||
// The Notification Service is intentionally NOT booted: the boundary under
|
||||
// test is "Lobby produces correct intent envelopes onto the stream",
|
||||
// independent of how the Notification Service consumes them.
|
||||
package lobbynotification_test
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"maps"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"slices"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync/atomic"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"galaxy/integration/internal/harness"
|
||||
|
||||
"github.com/redis/go-redis/v9"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
const (
|
||||
notificationIntentsStream = "notification:intents"
|
||||
userLifecycleStream = "user:lifecycle_events"
|
||||
runtimeJobResultsStream = "runtime:job_results"
|
||||
gmLobbyEventsStream = "gm:lobby_events"
|
||||
intentTypeApplicationSubmitted = "lobby.application.submitted"
|
||||
intentTypeMembershipApproved = "lobby.membership.approved"
|
||||
intentTypeMembershipRejected = "lobby.membership.rejected"
|
||||
intentTypeMembershipBlocked = "lobby.membership.blocked"
|
||||
intentTypeInviteCreated = "lobby.invite.created"
|
||||
intentTypeInviteRedeemed = "lobby.invite.redeemed"
|
||||
intentTypeInviteExpired = "lobby.invite.expired"
|
||||
intentTypeRuntimePausedAfter = "lobby.runtime_paused_after_start"
|
||||
expectedProducer = "game_lobby"
|
||||
)
|
||||
|
||||
func TestApplicationFlowPublishesSubmittedApprovedRejected(t *testing.T) {
|
||||
h := newLobbyNotificationHarness(t, gmAlwaysOK)
|
||||
|
||||
applicantA := h.ensureUser(t, "applicantA@example.com")
|
||||
applicantB := h.ensureUser(t, "applicantB@example.com")
|
||||
|
||||
gameID := h.adminCreatePublicGame(t, "Application Galaxy", time.Now().Add(48*time.Hour).Unix())
|
||||
h.openEnrollment(t, gameID)
|
||||
|
||||
appA := h.submitApplication(t, applicantA.UserID, gameID, "PilotAlpha")
|
||||
h.adminApproveApplication(t, gameID, appA["application_id"].(string))
|
||||
|
||||
appB := h.submitApplication(t, applicantB.UserID, gameID, "PilotBeta")
|
||||
h.adminRejectApplication(t, gameID, appB["application_id"].(string))
|
||||
|
||||
h.requireIntents(t,
|
||||
expect(intentTypeApplicationSubmitted, "admin"),
|
||||
expect(intentTypeApplicationSubmitted, "admin"),
|
||||
expect(intentTypeMembershipApproved, applicantA.UserID),
|
||||
expect(intentTypeMembershipRejected, applicantB.UserID),
|
||||
)
|
||||
}
|
||||
|
||||
func TestPrivateInviteLifecyclePublishesCreatedRedeemedExpired(t *testing.T) {
|
||||
h := newLobbyNotificationHarness(t, gmAlwaysOK)
|
||||
|
||||
owner := h.ensureUser(t, "owner@example.com")
|
||||
inviteeA := h.ensureUser(t, "inviteeA@example.com")
|
||||
inviteeB := h.ensureUser(t, "inviteeB@example.com")
|
||||
|
||||
gameID := h.userCreatePrivateGame(t, owner.UserID, "Private Invite Galaxy",
|
||||
time.Now().Add(48*time.Hour).Unix())
|
||||
h.userOpenEnrollment(t, owner.UserID, gameID)
|
||||
|
||||
h.userCreateInvite(t, owner.UserID, gameID, inviteeA.UserID)
|
||||
inviteB := h.userCreateInvite(t, owner.UserID, gameID, inviteeB.UserID)
|
||||
_ = inviteB
|
||||
|
||||
// Read invitee A's invite ID by listing their invites.
|
||||
inviteAID := h.firstCreatedInviteID(t, inviteeA.UserID, gameID)
|
||||
h.userRedeemInvite(t, inviteeA.UserID, gameID, inviteAID, "PilotPrivateA")
|
||||
|
||||
// Close enrollment (min_players=1 satisfied by inviteeA's redeem).
|
||||
// Invite B is still in `created` and must transition to `expired`.
|
||||
h.userReadyToStart(t, owner.UserID, gameID)
|
||||
|
||||
h.requireIntents(t,
|
||||
expect(intentTypeInviteCreated, inviteeA.UserID),
|
||||
expect(intentTypeInviteCreated, inviteeB.UserID),
|
||||
expect(intentTypeInviteRedeemed, owner.UserID),
|
||||
expect(intentTypeInviteExpired, owner.UserID),
|
||||
)
|
||||
}
|
||||
|
||||
func TestCascadeMembershipBlockedPublishesIntent(t *testing.T) {
|
||||
h := newLobbyNotificationHarness(t, gmAlwaysOK)
|
||||
|
||||
owner := h.ensureUser(t, "cascade-owner@example.com")
|
||||
invitee := h.ensureUser(t, "cascade-invitee@example.com")
|
||||
|
||||
gameID := h.userCreatePrivateGame(t, owner.UserID, "Cascade Galaxy",
|
||||
time.Now().Add(48*time.Hour).Unix())
|
||||
h.userOpenEnrollment(t, owner.UserID, gameID)
|
||||
h.userCreateInvite(t, owner.UserID, gameID, invitee.UserID)
|
||||
|
||||
inviteID := h.firstCreatedInviteID(t, invitee.UserID, gameID)
|
||||
h.userRedeemInvite(t, invitee.UserID, gameID, inviteID, "PilotCascade")
|
||||
|
||||
h.publishUserLifecycleEvent(t, "user.lifecycle.permanent_blocked", invitee.UserID)
|
||||
|
||||
h.requireIntents(t,
|
||||
expect(intentTypeInviteCreated, invitee.UserID),
|
||||
expect(intentTypeInviteRedeemed, owner.UserID),
|
||||
expect(intentTypeMembershipBlocked, owner.UserID),
|
||||
)
|
||||
}
|
||||
|
||||
func TestRuntimePausedAfterStartPublishesAdminIntent(t *testing.T) {
|
||||
gmRegisterFails := func(w http.ResponseWriter, r *http.Request) {
|
||||
if strings.Contains(r.URL.Path, "/register-runtime") {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
_, _ = w.Write([]byte(`{"error":"forced GM unavailability"}`))
|
||||
return
|
||||
}
|
||||
w.WriteHeader(http.StatusOK)
|
||||
_, _ = w.Write([]byte(`{}`))
|
||||
}
|
||||
|
||||
h := newLobbyNotificationHarness(t, gmRegisterFails)
|
||||
|
||||
applicant := h.ensureUser(t, "starter@example.com")
|
||||
|
||||
gameID := h.adminCreatePublicGame(t, "Runtime Pause Galaxy",
|
||||
time.Now().Add(48*time.Hour).Unix())
|
||||
h.openEnrollment(t, gameID)
|
||||
|
||||
app := h.submitApplication(t, applicant.UserID, gameID, "PilotPause")
|
||||
h.adminApproveApplication(t, gameID, app["application_id"].(string))
|
||||
|
||||
h.adminReadyToStart(t, gameID)
|
||||
h.adminStartGame(t, gameID)
|
||||
|
||||
h.publishRuntimeJobSuccess(t, gameID)
|
||||
|
||||
h.requireIntents(t,
|
||||
expect(intentTypeApplicationSubmitted, "admin"),
|
||||
expect(intentTypeMembershipApproved, applicant.UserID),
|
||||
expect(intentTypeRuntimePausedAfter, "admin"),
|
||||
)
|
||||
}
|
||||
|
||||
type lobbyNotificationHarness struct {
|
||||
redis *redis.Client
|
||||
|
||||
userServiceURL string
|
||||
lobbyPublicURL string
|
||||
lobbyAdminURL string
|
||||
|
||||
intentsStream string
|
||||
lifecycleStream string
|
||||
jobResultsStream string
|
||||
gmEventsStream string
|
||||
|
||||
gmStub *httptest.Server
|
||||
|
||||
userServiceProcess *harness.Process
|
||||
lobbyProcess *harness.Process
|
||||
}
|
||||
|
||||
type ensureByEmailResponse struct {
|
||||
Outcome string `json:"outcome"`
|
||||
UserID string `json:"user_id"`
|
||||
}
|
||||
|
||||
type expectedIntent struct {
|
||||
NotificationType string
|
||||
Recipient string // user_id, or "admin" for admin_email audience
|
||||
}
|
||||
|
||||
func expect(notificationType, recipient string) expectedIntent {
|
||||
return expectedIntent{NotificationType: notificationType, Recipient: recipient}
|
||||
}
|
||||
|
||||
func gmAlwaysOK(w http.ResponseWriter, _ *http.Request) {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
_, _ = w.Write([]byte(`{}`))
|
||||
}
|
||||
|
||||
var harnessSeq atomic.Int64
|
||||
|
||||
func newLobbyNotificationHarness(t *testing.T, gmHandler http.HandlerFunc) *lobbyNotificationHarness {
|
||||
t.Helper()
|
||||
|
||||
redisRuntime := harness.StartRedisContainer(t)
|
||||
redisClient := redis.NewClient(&redis.Options{
|
||||
Addr: redisRuntime.Addr,
|
||||
Protocol: 2,
|
||||
DisableIdentity: true,
|
||||
})
|
||||
t.Cleanup(func() {
|
||||
require.NoError(t, redisClient.Close())
|
||||
})
|
||||
|
||||
gmStub := httptest.NewServer(http.HandlerFunc(gmHandler))
|
||||
t.Cleanup(gmStub.Close)
|
||||
|
||||
userServiceAddr := harness.FreeTCPAddress(t)
|
||||
lobbyPublicAddr := harness.FreeTCPAddress(t)
|
||||
lobbyInternalAddr := harness.FreeTCPAddress(t)
|
||||
|
||||
userServiceBinary := harness.BuildBinary(t, "userservice", "./user/cmd/userservice")
|
||||
lobbyBinary := harness.BuildBinary(t, "lobby", "./lobby/cmd/lobby")
|
||||
|
||||
userServiceProcess := harness.StartProcess(t, "userservice", userServiceBinary, map[string]string{
|
||||
"USERSERVICE_LOG_LEVEL": "info",
|
||||
"USERSERVICE_INTERNAL_HTTP_ADDR": userServiceAddr,
|
||||
"USERSERVICE_REDIS_ADDR": redisRuntime.Addr,
|
||||
"OTEL_TRACES_EXPORTER": "none",
|
||||
"OTEL_METRICS_EXPORTER": "none",
|
||||
})
|
||||
waitForUserServiceReady(t, userServiceProcess, "http://"+userServiceAddr)
|
||||
|
||||
// Use unique stream prefixes per test so concurrent runs do not bleed.
|
||||
suffix := strconv.FormatInt(harnessSeq.Add(1), 10)
|
||||
intentsStream := notificationIntentsStream + ":" + suffix
|
||||
lifecycleStream := userLifecycleStream + ":" + suffix
|
||||
jobResultsStream := runtimeJobResultsStream + ":" + suffix
|
||||
gmEventsStream := gmLobbyEventsStream + ":" + suffix
|
||||
|
||||
lobbyProcess := harness.StartProcess(t, "lobby", lobbyBinary, map[string]string{
|
||||
"LOBBY_LOG_LEVEL": "info",
|
||||
"LOBBY_PUBLIC_HTTP_ADDR": lobbyPublicAddr,
|
||||
"LOBBY_INTERNAL_HTTP_ADDR": lobbyInternalAddr,
|
||||
"LOBBY_REDIS_ADDR": redisRuntime.Addr,
|
||||
"LOBBY_USER_SERVICE_BASE_URL": "http://" + userServiceAddr,
|
||||
"LOBBY_GM_BASE_URL": gmStub.URL,
|
||||
"LOBBY_NOTIFICATION_INTENTS_STREAM": intentsStream,
|
||||
"LOBBY_USER_LIFECYCLE_STREAM": lifecycleStream,
|
||||
"LOBBY_RUNTIME_JOB_RESULTS_STREAM": jobResultsStream,
|
||||
"LOBBY_GM_EVENTS_STREAM": gmEventsStream,
|
||||
"LOBBY_RUNTIME_JOB_RESULTS_READ_BLOCK_TIMEOUT": "200ms",
|
||||
"LOBBY_USER_LIFECYCLE_READ_BLOCK_TIMEOUT": "200ms",
|
||||
"LOBBY_GM_EVENTS_READ_BLOCK_TIMEOUT": "200ms",
|
||||
"OTEL_TRACES_EXPORTER": "none",
|
||||
"OTEL_METRICS_EXPORTER": "none",
|
||||
})
|
||||
harness.WaitForHTTPStatus(t, lobbyProcess, "http://"+lobbyInternalAddr+"/readyz", http.StatusOK)
|
||||
|
||||
return &lobbyNotificationHarness{
|
||||
redis: redisClient,
|
||||
userServiceURL: "http://" + userServiceAddr,
|
||||
lobbyPublicURL: "http://" + lobbyPublicAddr,
|
||||
lobbyAdminURL: "http://" + lobbyInternalAddr,
|
||||
intentsStream: intentsStream,
|
||||
lifecycleStream: lifecycleStream,
|
||||
jobResultsStream: jobResultsStream,
|
||||
gmEventsStream: gmEventsStream,
|
||||
gmStub: gmStub,
|
||||
userServiceProcess: userServiceProcess,
|
||||
lobbyProcess: lobbyProcess,
|
||||
}
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) ensureUser(t *testing.T, email string) ensureByEmailResponse {
|
||||
t.Helper()
|
||||
|
||||
resp := postJSON(t, h.userServiceURL+"/api/v1/internal/users/ensure-by-email", map[string]any{
|
||||
"email": email,
|
||||
"registration_context": map[string]string{
|
||||
"preferred_language": "en",
|
||||
"time_zone": "Europe/Kaliningrad",
|
||||
},
|
||||
}, nil)
|
||||
var out ensureByEmailResponse
|
||||
requireJSONStatus(t, resp, http.StatusOK, &out)
|
||||
require.Equal(t, "created", out.Outcome)
|
||||
require.NotEmpty(t, out.UserID)
|
||||
return out
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) adminCreatePublicGame(t *testing.T, name string, enrollmentEndsAt int64) string {
|
||||
t.Helper()
|
||||
return h.createGame(t, h.lobbyAdminURL+"/api/v1/lobby/games", "public", name, enrollmentEndsAt, nil)
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) userCreatePrivateGame(t *testing.T, ownerUserID, name string, enrollmentEndsAt int64) string {
|
||||
t.Helper()
|
||||
return h.createGame(t, h.lobbyPublicURL+"/api/v1/lobby/games", "private", name, enrollmentEndsAt,
|
||||
http.Header{"X-User-Id": []string{ownerUserID}})
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) createGame(t *testing.T, url, gameType, name string, enrollmentEndsAt int64, header http.Header) string {
|
||||
t.Helper()
|
||||
|
||||
resp := postJSON(t, url, map[string]any{
|
||||
"game_name": name,
|
||||
"game_type": gameType,
|
||||
"min_players": 1,
|
||||
"max_players": 4,
|
||||
"start_gap_hours": 6,
|
||||
"start_gap_players": 1,
|
||||
"enrollment_ends_at": enrollmentEndsAt,
|
||||
"turn_schedule": "0 18 * * *",
|
||||
"target_engine_version": "1.0.0",
|
||||
}, header)
|
||||
require.Equalf(t, http.StatusCreated, resp.StatusCode, "create %s game: %s", gameType, resp.Body)
|
||||
|
||||
var record map[string]any
|
||||
require.NoError(t, json.Unmarshal([]byte(resp.Body), &record))
|
||||
gameID, ok := record["game_id"].(string)
|
||||
require.Truef(t, ok, "game_id missing: %s", resp.Body)
|
||||
return gameID
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) openEnrollment(t *testing.T, gameID string) {
|
||||
t.Helper()
|
||||
resp := postJSON(t, h.lobbyAdminURL+"/api/v1/lobby/games/"+gameID+"/open-enrollment", nil, nil)
|
||||
require.Equalf(t, http.StatusOK, resp.StatusCode, "admin open enrollment: %s", resp.Body)
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) userOpenEnrollment(t *testing.T, ownerUserID, gameID string) {
|
||||
t.Helper()
|
||||
resp := postJSON(t, h.lobbyPublicURL+"/api/v1/lobby/games/"+gameID+"/open-enrollment", nil,
|
||||
http.Header{"X-User-Id": []string{ownerUserID}})
|
||||
require.Equalf(t, http.StatusOK, resp.StatusCode, "user open enrollment: %s", resp.Body)
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) submitApplication(t *testing.T, userID, gameID, raceName string) map[string]any {
|
||||
t.Helper()
|
||||
resp := postJSON(t, h.lobbyPublicURL+"/api/v1/lobby/games/"+gameID+"/applications",
|
||||
map[string]any{"race_name": raceName},
|
||||
http.Header{"X-User-Id": []string{userID}})
|
||||
require.Equalf(t, http.StatusCreated, resp.StatusCode, "submit application: %s", resp.Body)
|
||||
var body map[string]any
|
||||
require.NoError(t, json.Unmarshal([]byte(resp.Body), &body))
|
||||
return body
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) adminApproveApplication(t *testing.T, gameID, applicationID string) {
|
||||
t.Helper()
|
||||
resp := postJSON(t,
|
||||
h.lobbyAdminURL+"/api/v1/lobby/games/"+gameID+"/applications/"+applicationID+"/approve",
|
||||
nil, nil)
|
||||
require.Equalf(t, http.StatusOK, resp.StatusCode, "admin approve: %s", resp.Body)
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) adminRejectApplication(t *testing.T, gameID, applicationID string) {
|
||||
t.Helper()
|
||||
resp := postJSON(t,
|
||||
h.lobbyAdminURL+"/api/v1/lobby/games/"+gameID+"/applications/"+applicationID+"/reject",
|
||||
nil, nil)
|
||||
require.Equalf(t, http.StatusOK, resp.StatusCode, "admin reject: %s", resp.Body)
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) userCreateInvite(t *testing.T, ownerUserID, gameID, inviteeUserID string) map[string]any {
|
||||
t.Helper()
|
||||
resp := postJSON(t, h.lobbyPublicURL+"/api/v1/lobby/games/"+gameID+"/invites",
|
||||
map[string]any{"invitee_user_id": inviteeUserID},
|
||||
http.Header{"X-User-Id": []string{ownerUserID}})
|
||||
require.Equalf(t, http.StatusCreated, resp.StatusCode, "create invite: %s", resp.Body)
|
||||
var body map[string]any
|
||||
require.NoError(t, json.Unmarshal([]byte(resp.Body), &body))
|
||||
return body
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) firstCreatedInviteID(t *testing.T, inviteeUserID, gameID string) string {
|
||||
t.Helper()
|
||||
req, err := http.NewRequest(http.MethodGet, h.lobbyPublicURL+"/api/v1/lobby/my/invites?status=created", nil)
|
||||
require.NoError(t, err)
|
||||
req.Header.Set("X-User-Id", inviteeUserID)
|
||||
resp := doRequest(t, req)
|
||||
require.Equalf(t, http.StatusOK, resp.StatusCode, "list my invites: %s", resp.Body)
|
||||
|
||||
var body struct {
|
||||
Items []struct {
|
||||
InviteID string `json:"invite_id"`
|
||||
GameID string `json:"game_id"`
|
||||
} `json:"items"`
|
||||
}
|
||||
require.NoError(t, json.Unmarshal([]byte(resp.Body), &body))
|
||||
for _, item := range body.Items {
|
||||
if item.GameID == gameID {
|
||||
return item.InviteID
|
||||
}
|
||||
}
|
||||
t.Fatalf("no invite found for invitee %s on game %s; body=%s", inviteeUserID, gameID, resp.Body)
|
||||
return ""
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) userRedeemInvite(t *testing.T, inviteeUserID, gameID, inviteID, raceName string) {
|
||||
t.Helper()
|
||||
resp := postJSON(t,
|
||||
h.lobbyPublicURL+"/api/v1/lobby/games/"+gameID+"/invites/"+inviteID+"/redeem",
|
||||
map[string]any{"race_name": raceName},
|
||||
http.Header{"X-User-Id": []string{inviteeUserID}})
|
||||
require.Equalf(t, http.StatusOK, resp.StatusCode, "redeem invite: %s", resp.Body)
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) userReadyToStart(t *testing.T, ownerUserID, gameID string) {
|
||||
t.Helper()
|
||||
resp := postJSON(t,
|
||||
h.lobbyPublicURL+"/api/v1/lobby/games/"+gameID+"/ready-to-start",
|
||||
nil,
|
||||
http.Header{"X-User-Id": []string{ownerUserID}})
|
||||
require.Equalf(t, http.StatusOK, resp.StatusCode, "user ready-to-start: %s", resp.Body)
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) adminReadyToStart(t *testing.T, gameID string) {
|
||||
t.Helper()
|
||||
resp := postJSON(t, h.lobbyAdminURL+"/api/v1/lobby/games/"+gameID+"/ready-to-start", nil, nil)
|
||||
require.Equalf(t, http.StatusOK, resp.StatusCode, "admin ready-to-start: %s", resp.Body)
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) adminStartGame(t *testing.T, gameID string) {
|
||||
t.Helper()
|
||||
resp := postJSON(t, h.lobbyAdminURL+"/api/v1/lobby/games/"+gameID+"/start", nil, nil)
|
||||
require.Equalf(t, http.StatusOK, resp.StatusCode, "admin start game: %s", resp.Body)
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) publishUserLifecycleEvent(t *testing.T, eventType, userID string) {
|
||||
t.Helper()
|
||||
_, err := h.redis.XAdd(context.Background(), &redis.XAddArgs{
|
||||
Stream: h.lifecycleStream,
|
||||
Values: map[string]any{
|
||||
"event_type": eventType,
|
||||
"user_id": userID,
|
||||
"occurred_at_ms": strconv.FormatInt(time.Now().UnixMilli(), 10),
|
||||
"source": "user_admin",
|
||||
"actor_type": "admin",
|
||||
"actor_id": "admin-1",
|
||||
"reason_code": "terminal_policy_violation",
|
||||
},
|
||||
}).Result()
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) publishRuntimeJobSuccess(t *testing.T, gameID string) {
|
||||
t.Helper()
|
||||
_, err := h.redis.XAdd(context.Background(), &redis.XAddArgs{
|
||||
Stream: h.jobResultsStream,
|
||||
Values: map[string]any{
|
||||
"game_id": gameID,
|
||||
"outcome": "success",
|
||||
"container_id": "container-" + gameID,
|
||||
"engine_endpoint": "127.0.0.1:0",
|
||||
},
|
||||
}).Result()
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
func (h *lobbyNotificationHarness) requireIntents(t *testing.T, want ...expectedIntent) {
|
||||
t.Helper()
|
||||
|
||||
want = append([]expectedIntent(nil), want...)
|
||||
|
||||
require.Eventuallyf(t, func() bool {
|
||||
entries, err := h.redis.XRange(context.Background(), h.intentsStream, "-", "+").Result()
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
published := decodePublishedIntents(t, entries)
|
||||
return matchesAll(published, want)
|
||||
}, 15*time.Second, 100*time.Millisecond,
|
||||
"expected intents %+v not all observed on stream %s", want, h.intentsStream)
|
||||
|
||||
entries, err := h.redis.XRange(context.Background(), h.intentsStream, "-", "+").Result()
|
||||
require.NoError(t, err)
|
||||
published := decodePublishedIntents(t, entries)
|
||||
for _, p := range published {
|
||||
require.Equal(t, expectedProducer, p.Producer,
|
||||
"every published intent must declare producer=%q", expectedProducer)
|
||||
}
|
||||
}
|
||||
|
||||
type publishedIntent struct {
|
||||
NotificationType string
|
||||
Producer string
|
||||
AudienceKind string
|
||||
RecipientUserIDs []string
|
||||
}
|
||||
|
||||
func decodePublishedIntents(t *testing.T, entries []redis.XMessage) []publishedIntent {
|
||||
t.Helper()
|
||||
|
||||
out := make([]publishedIntent, 0, len(entries))
|
||||
for _, entry := range entries {
|
||||
notificationType, _ := entry.Values["notification_type"].(string)
|
||||
producer, _ := entry.Values["producer"].(string)
|
||||
audienceKind, _ := entry.Values["audience_kind"].(string)
|
||||
recipientsJSON, _ := entry.Values["recipient_user_ids_json"].(string)
|
||||
|
||||
var recipients []string
|
||||
if recipientsJSON != "" {
|
||||
require.NoError(t, json.Unmarshal([]byte(recipientsJSON), &recipients))
|
||||
}
|
||||
|
||||
out = append(out, publishedIntent{
|
||||
NotificationType: notificationType,
|
||||
Producer: producer,
|
||||
AudienceKind: audienceKind,
|
||||
RecipientUserIDs: recipients,
|
||||
})
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func matchesAll(published []publishedIntent, want []expectedIntent) bool {
|
||||
used := make([]bool, len(published))
|
||||
for _, w := range want {
|
||||
matched := -1
|
||||
for i, p := range published {
|
||||
if used[i] {
|
||||
continue
|
||||
}
|
||||
if p.NotificationType != w.NotificationType {
|
||||
continue
|
||||
}
|
||||
if w.Recipient == "admin" {
|
||||
if p.AudienceKind == "admin_email" {
|
||||
matched = i
|
||||
break
|
||||
}
|
||||
continue
|
||||
}
|
||||
if slices.Contains(p.RecipientUserIDs, w.Recipient) {
|
||||
matched = i
|
||||
break
|
||||
}
|
||||
}
|
||||
if matched < 0 {
|
||||
return false
|
||||
}
|
||||
used[matched] = true
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func waitForUserServiceReady(t *testing.T, process *harness.Process, baseURL string) {
|
||||
t.Helper()
|
||||
client := &http.Client{Timeout: 250 * time.Millisecond}
|
||||
t.Cleanup(client.CloseIdleConnections)
|
||||
|
||||
deadline := time.Now().Add(10 * time.Second)
|
||||
for time.Now().Before(deadline) {
|
||||
req, err := http.NewRequest(http.MethodGet, baseURL+"/api/v1/internal/users/user-readiness-probe/exists", nil)
|
||||
require.NoError(t, err)
|
||||
response, err := client.Do(req)
|
||||
if err == nil {
|
||||
_, _ = io.Copy(io.Discard, response.Body)
|
||||
response.Body.Close()
|
||||
if response.StatusCode == http.StatusOK {
|
||||
return
|
||||
}
|
||||
}
|
||||
time.Sleep(25 * time.Millisecond)
|
||||
}
|
||||
t.Fatalf("wait for userservice readiness: timeout\n%s", process.Logs())
|
||||
}
|
||||
|
||||
type httpResponse struct {
|
||||
StatusCode int
|
||||
Body string
|
||||
Header http.Header
|
||||
}
|
||||
|
||||
func postJSON(t *testing.T, url string, body any, header http.Header) httpResponse {
|
||||
t.Helper()
|
||||
var reader io.Reader
|
||||
if body != nil {
|
||||
payload, err := json.Marshal(body)
|
||||
require.NoError(t, err)
|
||||
reader = bytes.NewReader(payload)
|
||||
}
|
||||
req, err := http.NewRequest(http.MethodPost, url, reader)
|
||||
require.NoError(t, err)
|
||||
if body != nil {
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
}
|
||||
maps.Copy(req.Header, header)
|
||||
return doRequest(t, req)
|
||||
}
|
||||
|
||||
func doRequest(t *testing.T, request *http.Request) httpResponse {
|
||||
t.Helper()
|
||||
client := &http.Client{
|
||||
Timeout: 5 * time.Second,
|
||||
Transport: &http.Transport{DisableKeepAlives: true},
|
||||
}
|
||||
t.Cleanup(client.CloseIdleConnections)
|
||||
|
||||
response, err := client.Do(request)
|
||||
require.NoError(t, err)
|
||||
defer response.Body.Close()
|
||||
|
||||
payload, err := io.ReadAll(response.Body)
|
||||
require.NoError(t, err)
|
||||
return httpResponse{
|
||||
StatusCode: response.StatusCode,
|
||||
Body: string(payload),
|
||||
Header: response.Header.Clone(),
|
||||
}
|
||||
}
|
||||
|
||||
func requireJSONStatus(t *testing.T, response httpResponse, wantStatus int, target any) {
|
||||
t.Helper()
|
||||
require.Equalf(t, wantStatus, response.StatusCode, "unexpected status, body=%s", response.Body)
|
||||
if target != nil {
|
||||
require.NoError(t, decodeStrictJSON([]byte(response.Body), target))
|
||||
}
|
||||
}
|
||||
|
||||
func decodeStrictJSON(payload []byte, target any) error {
|
||||
decoder := json.NewDecoder(bytes.NewReader(payload))
|
||||
decoder.DisallowUnknownFields()
|
||||
if err := decoder.Decode(target); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := decoder.Decode(&struct{}{}); err != io.EOF {
|
||||
if err == nil {
|
||||
return errors.New("unexpected trailing JSON input")
|
||||
}
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// silenceUnused keeps fmt referenced by future debug formatting needs.
|
||||
var _ = fmt.Sprintf
|
||||
Reference in New Issue
Block a user