834 lines
26 KiB
TypeScript
834 lines
26 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from "vitest";
|
|
|
|
import {
|
|
MAX_VIEWERS_PER_ROOM_LIMIT,
|
|
createRoomResponseSchema,
|
|
roomAccessUpdateResponseSchema,
|
|
} from "../src/shared/protocol.ts";
|
|
import {
|
|
createScreenerServer,
|
|
type CreateServerOptions,
|
|
type ScreenerServer,
|
|
} from "../src/server/app.ts";
|
|
import type { ServerConfig } from "../src/server/config.ts";
|
|
import { RoomStore } from "../src/server/room-store.ts";
|
|
import { FakeSfuRoomControl } from "./fake-sfu-room-control.ts";
|
|
|
|
const allowedOrigin = "http://allowed.test";
|
|
const siteAccessPassword = "instance-access-password";
|
|
let runningServer: ScreenerServer | undefined;
|
|
afterEach(async () => {
|
|
await runningServer?.close();
|
|
runningServer = undefined;
|
|
});
|
|
|
|
function testConfig(overrides: Partial<ServerConfig> = {}): ServerConfig {
|
|
return {
|
|
nodeEnv: "test",
|
|
port: 0,
|
|
listenHost: "127.0.0.1",
|
|
publicBaseUrl: new URL("https://share.example.test"),
|
|
allowedOrigins: new Set([allowedOrigin]),
|
|
siteAccessPassword,
|
|
roomLeaseMs: 86_400_000,
|
|
maxViewersPerRoom: 8,
|
|
peerAssistedMedia: false,
|
|
endpointMediaCopyCapacity: 2,
|
|
stunUrls: [],
|
|
...overrides,
|
|
};
|
|
}
|
|
|
|
async function start(
|
|
config = testConfig(),
|
|
options: Pick<
|
|
CreateServerOptions,
|
|
| "now"
|
|
| "roomStore"
|
|
| "siteAccessTtlSeconds"
|
|
| "sfuTokenIssuer"
|
|
| "sfuRoomControl"
|
|
> = {},
|
|
): Promise<string> {
|
|
const sfuRoomControl =
|
|
options.sfuRoomControl ??
|
|
(config.livekitFallback ? new FakeSfuRoomControl() : undefined);
|
|
runningServer = await createScreenerServer({
|
|
...options,
|
|
...(sfuRoomControl ? { sfuRoomControl } : {}),
|
|
config,
|
|
serveFrontend: false,
|
|
});
|
|
const port = await runningServer.listen(0, "127.0.0.1");
|
|
return `http://127.0.0.1:${port}`;
|
|
}
|
|
|
|
function cookiePair(response: Response): string {
|
|
const setCookie = response.headers.get("set-cookie");
|
|
if (!setCookie) {
|
|
throw new Error("Expected a Set-Cookie header");
|
|
}
|
|
return setCookie.split(";", 1)[0]!;
|
|
}
|
|
|
|
async function login(baseUrl: string): Promise<Response> {
|
|
return fetch(`${baseUrl}/api/site-access`, {
|
|
method: "POST",
|
|
headers: {
|
|
Authorization: `Bearer ${siteAccessPassword}`,
|
|
Origin: allowedOrigin,
|
|
},
|
|
});
|
|
}
|
|
|
|
async function createRoom(
|
|
baseUrl: string,
|
|
cookie?: string,
|
|
codeEntryPolicy: "open" | "private" = "open",
|
|
roomPassword?: string,
|
|
preferredRoomId?: string,
|
|
): Promise<Response> {
|
|
return fetch(`${baseUrl}/api/rooms`, {
|
|
method: "POST",
|
|
headers: {
|
|
"Content-Type": "application/json",
|
|
Origin: allowedOrigin,
|
|
...(cookie ? { Cookie: cookie } : {}),
|
|
},
|
|
body: JSON.stringify({
|
|
codeEntryPolicy,
|
|
...(roomPassword === undefined ? {} : { roomPassword }),
|
|
...(preferredRoomId === undefined ? {} : { preferredRoomId }),
|
|
}),
|
|
});
|
|
}
|
|
|
|
async function updateRoomAccess(
|
|
baseUrl: string,
|
|
roomId: string,
|
|
hostToken: string | undefined,
|
|
body: unknown,
|
|
options: { cookie?: string; origin?: string; method?: string } = {},
|
|
): Promise<Response> {
|
|
return fetch(`${baseUrl}/api/rooms/${roomId}/access`, {
|
|
method: options.method ?? "POST",
|
|
headers: {
|
|
"Content-Type": "application/json",
|
|
Origin: options.origin ?? allowedOrigin,
|
|
...(options.cookie ? { Cookie: options.cookie } : {}),
|
|
...(hostToken ? { Authorization: `Bearer ${hostToken}` } : {}),
|
|
},
|
|
body:
|
|
(options.method ?? "POST") === "POST"
|
|
? JSON.stringify(body)
|
|
: undefined,
|
|
});
|
|
}
|
|
|
|
describe("site access", () => {
|
|
it("reports status and issues a stateless 12-hour cookie", async () => {
|
|
const baseUrl = await start();
|
|
const initial = await fetch(`${baseUrl}/api/site-access`);
|
|
|
|
expect(initial.status).toBe(200);
|
|
expect(initial.headers.get("cache-control")).toBe("no-store");
|
|
expect(await initial.json()).toEqual({
|
|
required: true,
|
|
authenticated: false,
|
|
});
|
|
|
|
const denied = await fetch(`${baseUrl}/api/site-access`, {
|
|
method: "POST",
|
|
headers: {
|
|
Authorization: "Bearer wrong-password",
|
|
Origin: allowedOrigin,
|
|
},
|
|
});
|
|
expect(denied.status).toBe(401);
|
|
expect(denied.headers.get("set-cookie")).toBeNull();
|
|
|
|
const authenticated = await login(baseUrl);
|
|
const setCookie = authenticated.headers.get("set-cookie") ?? "";
|
|
expect(authenticated.status).toBe(200);
|
|
expect(setCookie).toContain("screener-site-access=v1.");
|
|
expect(setCookie).toContain("Path=/");
|
|
expect(setCookie).toContain("Max-Age=43200");
|
|
expect(setCookie).toContain("HttpOnly");
|
|
expect(setCookie).toContain("SameSite=Strict");
|
|
expect(setCookie).not.toContain("Secure");
|
|
expect(setCookie).not.toContain(siteAccessPassword);
|
|
|
|
const status = await fetch(`${baseUrl}/api/site-access`, {
|
|
headers: { Cookie: cookiePair(authenticated) },
|
|
});
|
|
expect(await status.json()).toEqual({
|
|
required: true,
|
|
authenticated: true,
|
|
});
|
|
});
|
|
|
|
it("uses a secure __Host- cookie for production HTTPS", async () => {
|
|
const baseUrl = await start(testConfig({ nodeEnv: "production" }));
|
|
const response = await login(baseUrl);
|
|
const setCookie = response.headers.get("set-cookie") ?? "";
|
|
|
|
expect(setCookie).toContain("__Host-screener-site-access=");
|
|
expect(setCookie).toContain("Secure");
|
|
expect(setCookie).toContain("HttpOnly");
|
|
expect(setCookie).toContain("SameSite=Strict");
|
|
expect(setCookie).toContain("Path=/");
|
|
expect(setCookie).not.toContain("Domain=");
|
|
});
|
|
|
|
it("requires an allowed Origin and an empty login body", async () => {
|
|
const baseUrl = await start();
|
|
const authorization = {
|
|
Authorization: `Bearer ${siteAccessPassword}`,
|
|
};
|
|
|
|
expect(
|
|
(
|
|
await fetch(`${baseUrl}/api/site-access`, {
|
|
method: "POST",
|
|
headers: authorization,
|
|
})
|
|
).status,
|
|
).toBe(403);
|
|
expect(
|
|
(
|
|
await fetch(`${baseUrl}/api/site-access`, {
|
|
method: "POST",
|
|
headers: { ...authorization, Origin: "https://foreign.test" },
|
|
})
|
|
).status,
|
|
).toBe(403);
|
|
expect(
|
|
(
|
|
await fetch(`${baseUrl}/api/site-access`, {
|
|
method: "POST",
|
|
headers: { ...authorization, Origin: `${allowedOrigin}/path` },
|
|
})
|
|
).status,
|
|
).toBe(403);
|
|
expect(
|
|
(
|
|
await fetch(`${baseUrl}/api/site-access`, {
|
|
method: "POST",
|
|
headers: { ...authorization, Origin: allowedOrigin },
|
|
body: "{}",
|
|
})
|
|
).status,
|
|
).toBe(400);
|
|
});
|
|
|
|
it("rejects an expired or modified cookie", async () => {
|
|
let now = 1_000;
|
|
const baseUrl = await start(testConfig(), {
|
|
now: () => now,
|
|
siteAccessTtlSeconds: 1,
|
|
});
|
|
const authenticated = await login(baseUrl);
|
|
const cookie = cookiePair(authenticated);
|
|
|
|
const modified = await fetch(`${baseUrl}/api/site-access`, {
|
|
headers: { Cookie: `${cookie}x` },
|
|
});
|
|
expect((await modified.json()).authenticated).toBe(false);
|
|
|
|
now = 2_000;
|
|
const expired = await fetch(`${baseUrl}/api/site-access`, {
|
|
headers: { Cookie: cookie },
|
|
});
|
|
expect(await expired.json()).toEqual({
|
|
required: true,
|
|
authenticated: false,
|
|
});
|
|
});
|
|
|
|
it("is immediately authenticated when the access password is empty", async () => {
|
|
const baseUrl = await start(
|
|
testConfig({ siteAccessPassword: undefined }),
|
|
);
|
|
const status = await fetch(`${baseUrl}/api/site-access`);
|
|
|
|
expect(await status.json()).toEqual({
|
|
required: false,
|
|
authenticated: true,
|
|
});
|
|
const post = await fetch(`${baseUrl}/api/site-access`, {
|
|
method: "POST",
|
|
headers: { Origin: allowedOrigin },
|
|
});
|
|
expect(post.status).toBe(200);
|
|
expect(post.headers.get("set-cookie")).toBeNull();
|
|
});
|
|
});
|
|
|
|
describe("room HTTP API", () => {
|
|
it("requires a logged-in browser when access protection is enabled", async () => {
|
|
const baseUrl = await start();
|
|
const anonymous = await createRoom(baseUrl);
|
|
expect(anonymous.status).toBe(401);
|
|
expect(anonymous.headers.get("www-authenticate")).toBeNull();
|
|
|
|
const bearerBypass = await fetch(`${baseUrl}/api/rooms`, {
|
|
method: "POST",
|
|
headers: {
|
|
Authorization: `Bearer ${siteAccessPassword}`,
|
|
"Content-Type": "application/json",
|
|
Origin: allowedOrigin,
|
|
},
|
|
body: JSON.stringify({ codeEntryPolicy: "open" }),
|
|
});
|
|
expect(bearerBypass.status).toBe(401);
|
|
});
|
|
|
|
it("creates a room with an independent fragment-only Viewer grant", async () => {
|
|
const baseUrl = await start();
|
|
const authenticated = await login(baseUrl);
|
|
const response = await createRoom(baseUrl, cookiePair(authenticated));
|
|
|
|
expect(response.status).toBe(201);
|
|
expect(response.headers.get("cache-control")).toBe("no-store");
|
|
const body = createRoomResponseSchema.parse(await response.json());
|
|
const invite = new URL(body.inviteUrl);
|
|
expect(body.roomId).toMatch(/^[1-9]\d{3}$/);
|
|
expect(invite.origin).toBe("https://share.example.test");
|
|
expect(invite.pathname).toBe(`/r/${body.roomId}`);
|
|
expect(invite.hash).toMatch(/^#v=[A-Za-z0-9_-]{21}[AQgw]$/);
|
|
expect(invite.search).toBe("");
|
|
expect(body.codeEntryPolicy).toBe("open");
|
|
expect("viewerGrantExpiresAt" in body).toBe(false);
|
|
expect(body.expiresAt).toBeTruthy();
|
|
expect(body.roomLeaseSeconds).toBe(86_400);
|
|
expect("iceConfig" in body).toBe(false);
|
|
});
|
|
|
|
it("passes the shared Viewer ceiling to room admission", async () => {
|
|
const baseUrl = await start(
|
|
testConfig({
|
|
siteAccessPassword: undefined,
|
|
maxViewersPerRoom: MAX_VIEWERS_PER_ROOM_LIMIT,
|
|
}),
|
|
);
|
|
|
|
expect((await createRoom(baseUrl)).status).toBe(201);
|
|
expect(runningServer?.roomStore.maxViewersPerRoom).toBe(
|
|
MAX_VIEWERS_PER_ROOM_LIMIT,
|
|
);
|
|
});
|
|
|
|
it("creates private rooms with optional passwords atomically", async () => {
|
|
const baseUrl = await start();
|
|
const authenticated = await login(baseUrl);
|
|
const cookie = cookiePair(authenticated);
|
|
const response = await createRoom(
|
|
baseUrl,
|
|
cookie,
|
|
"private",
|
|
"room-password",
|
|
);
|
|
expect(response.status).toBe(201);
|
|
expect(createRoomResponseSchema.parse(await response.json())).toMatchObject({
|
|
codeEntryPolicy: "private",
|
|
});
|
|
|
|
const withoutPassword = await createRoom(baseUrl, cookie, "private");
|
|
expect(withoutPassword.status).toBe(201);
|
|
expect(
|
|
createRoomResponseSchema.parse(await withoutPassword.json()),
|
|
).toMatchObject({ codeEntryPolicy: "private" });
|
|
});
|
|
|
|
it("allows explicit open creation without site access in local mode", async () => {
|
|
const baseUrl = await start(
|
|
testConfig({ siteAccessPassword: undefined }),
|
|
);
|
|
const response = await createRoom(baseUrl, undefined, "open");
|
|
expect(response.status).toBe(201);
|
|
const body = createRoomResponseSchema.parse(await response.json());
|
|
expect(body).toMatchObject({
|
|
codeEntryPolicy: "open",
|
|
});
|
|
expect(new URL(body.inviteUrl).hash).toContain("#v=");
|
|
});
|
|
|
|
it("allocates unique four-digit codes concurrently", async () => {
|
|
const baseUrl = await start();
|
|
const authenticated = await login(baseUrl);
|
|
const cookie = cookiePair(authenticated);
|
|
const create = () => createRoom(baseUrl, cookie);
|
|
|
|
const responses = await Promise.all([create(), create()]);
|
|
const rooms = await Promise.all(
|
|
responses.map(async (response) =>
|
|
createRoomResponseSchema.parse(await response.json()),
|
|
),
|
|
);
|
|
expect(new Set(rooms.map((room) => room.roomId)).size).toBe(2);
|
|
expect(rooms.every((room) => room.expiresAt !== null)).toBe(true);
|
|
});
|
|
|
|
it("reuses a free preferred code and never replaces an occupied room", async () => {
|
|
const baseUrl = await start(testConfig({ roomLeaseMs: 90_000 }));
|
|
const authenticated = await login(baseUrl);
|
|
const cookie = cookiePair(authenticated);
|
|
|
|
const preferred = createRoomResponseSchema.parse(
|
|
await (await createRoom(baseUrl, cookie, "open", undefined, "4321")).json(),
|
|
);
|
|
const fallback = createRoomResponseSchema.parse(
|
|
await (await createRoom(baseUrl, cookie, "open", undefined, "4321")).json(),
|
|
);
|
|
|
|
expect(preferred.roomId).toBe("4321");
|
|
expect(preferred.roomLeaseSeconds).toBe(90);
|
|
expect(fallback.roomId).not.toBe("4321");
|
|
expect(fallback.hostToken).not.toBe(preferred.hostToken);
|
|
});
|
|
|
|
it("manages dormant room access without starting sharing or renewing", async () => {
|
|
let nowMs = 0;
|
|
const baseUrl = await start(
|
|
testConfig({ roomLeaseMs: 1_000 }),
|
|
{ now: () => nowMs },
|
|
);
|
|
const authenticated = await login(baseUrl);
|
|
const cookie = cookiePair(authenticated);
|
|
const createdResponse = await createRoom(baseUrl, cookie);
|
|
const room = createRoomResponseSchema.parse(await createdResponse.json());
|
|
nowMs = 900;
|
|
|
|
const policy = await updateRoomAccess(
|
|
baseUrl,
|
|
room.roomId,
|
|
room.hostToken,
|
|
{ action: "set-code-entry-policy", policy: "private" },
|
|
{ cookie },
|
|
);
|
|
expect(policy.status).toBe(200);
|
|
expect(policy.headers.get("cache-control")).toBe("no-store");
|
|
expect(roomAccessUpdateResponseSchema.parse(await policy.json())).toEqual({
|
|
type: "code-entry-policy-updated",
|
|
codeEntryPolicy: "private",
|
|
viewerPasswordEnabled: false,
|
|
});
|
|
|
|
const password = await updateRoomAccess(
|
|
baseUrl,
|
|
room.roomId,
|
|
room.hostToken,
|
|
{ action: "set-viewer-password", password: "room-password" },
|
|
{ cookie },
|
|
);
|
|
expect(roomAccessUpdateResponseSchema.parse(await password.json())).toEqual({
|
|
type: "viewer-password-updated",
|
|
enabled: true,
|
|
});
|
|
const removedPassword = await updateRoomAccess(
|
|
baseUrl,
|
|
room.roomId,
|
|
room.hostToken,
|
|
{ action: "set-viewer-password", password: null },
|
|
{ cookie },
|
|
);
|
|
expect(
|
|
roomAccessUpdateResponseSchema.parse(await removedPassword.json()),
|
|
).toEqual({
|
|
type: "viewer-password-updated",
|
|
enabled: false,
|
|
});
|
|
|
|
const rotated = await updateRoomAccess(
|
|
baseUrl,
|
|
room.roomId,
|
|
room.hostToken,
|
|
{ action: "rotate-viewer-grant" },
|
|
{ cookie },
|
|
);
|
|
const rotatedBody = roomAccessUpdateResponseSchema.parse(
|
|
await rotated.json(),
|
|
);
|
|
expect(rotatedBody.type).toBe("viewer-grant-updated");
|
|
expect(rotatedBody.type === "viewer-grant-updated" && rotatedBody.inviteUrl)
|
|
.toMatch(/#v=[A-Za-z0-9_-]{22}$/);
|
|
|
|
const revoked = await updateRoomAccess(
|
|
baseUrl,
|
|
room.roomId,
|
|
room.hostToken,
|
|
{ action: "revoke-viewer-grant" },
|
|
{ cookie },
|
|
);
|
|
expect(roomAccessUpdateResponseSchema.parse(await revoked.json())).toMatchObject({
|
|
type: "viewer-grant-updated",
|
|
inviteUrl: null,
|
|
});
|
|
expect(runningServer?.roomStore.getConnectedHost(room.roomId)).toBeUndefined();
|
|
|
|
nowMs = 1_001;
|
|
expect(
|
|
runningServer?.roomStore
|
|
.expireRooms()
|
|
.map((expired) => expired.roomId),
|
|
).toContain(room.roomId);
|
|
});
|
|
|
|
it("requires same-origin site access and the exact room Host token", async () => {
|
|
const baseUrl = await start();
|
|
const authenticated = await login(baseUrl);
|
|
const cookie = cookiePair(authenticated);
|
|
const first = createRoomResponseSchema.parse(
|
|
await (await createRoom(baseUrl, cookie)).json(),
|
|
);
|
|
const second = createRoomResponseSchema.parse(
|
|
await (await createRoom(baseUrl, cookie)).json(),
|
|
);
|
|
const action = { action: "rotate-viewer-grant" };
|
|
|
|
expect(
|
|
(await updateRoomAccess(baseUrl, first.roomId, first.hostToken, action))
|
|
.status,
|
|
).toBe(401);
|
|
expect(
|
|
(
|
|
await updateRoomAccess(baseUrl, first.roomId, first.hostToken, action, {
|
|
cookie,
|
|
origin: "https://foreign.test",
|
|
})
|
|
).status,
|
|
).toBe(403);
|
|
expect(
|
|
(await updateRoomAccess(baseUrl, first.roomId, undefined, action, { cookie }))
|
|
.status,
|
|
).toBe(404);
|
|
expect(
|
|
(
|
|
await updateRoomAccess(baseUrl, first.roomId, "wrong-token", action, {
|
|
cookie,
|
|
})
|
|
).status,
|
|
).toBe(404);
|
|
expect(
|
|
(
|
|
await updateRoomAccess(baseUrl, first.roomId, second.hostToken, action, {
|
|
cookie,
|
|
})
|
|
).status,
|
|
).toBe(404);
|
|
expect(
|
|
(
|
|
await updateRoomAccess(
|
|
baseUrl,
|
|
first.roomId,
|
|
first.hostToken,
|
|
{ action: "unknown" },
|
|
{ cookie },
|
|
)
|
|
).status,
|
|
).toBe(400);
|
|
const method = await updateRoomAccess(
|
|
baseUrl,
|
|
first.roomId,
|
|
first.hostToken,
|
|
action,
|
|
{ cookie, method: "GET" },
|
|
);
|
|
expect(method.status).toBe(405);
|
|
expect(method.headers.get("allow")).toBe("POST");
|
|
});
|
|
|
|
it("rejects malformed room requests and foreign browser origins", async () => {
|
|
const baseUrl = await start(
|
|
testConfig({ siteAccessPassword: undefined }),
|
|
);
|
|
const withBody = await fetch(`${baseUrl}/api/rooms`, {
|
|
method: "POST",
|
|
headers: {
|
|
"Content-Type": "application/json",
|
|
Origin: allowedOrigin,
|
|
},
|
|
body: "{}",
|
|
});
|
|
expect(withBody.status).toBe(400);
|
|
|
|
const foreign = await fetch(`${baseUrl}/api/rooms`, {
|
|
method: "POST",
|
|
headers: { "Content-Type": "application/json", Origin: "https://foreign.test" },
|
|
body: JSON.stringify({ codeEntryPolicy: "open" }),
|
|
});
|
|
expect(foreign.status).toBe(403);
|
|
});
|
|
|
|
it("returns service unavailable at the global room bound", async () => {
|
|
const baseUrl = await start(
|
|
testConfig({ siteAccessPassword: undefined }),
|
|
{
|
|
roomStore: new RoomStore({
|
|
leaseMs: 86_400_000,
|
|
maxRooms: 1,
|
|
maxViewersPerRoom: 8,
|
|
}),
|
|
},
|
|
);
|
|
const create = () => createRoom(baseUrl);
|
|
|
|
expect((await create()).status).toBe(201);
|
|
expect((await create()).status).toBe(503);
|
|
});
|
|
});
|
|
|
|
describe("server HTTP listener and health", () => {
|
|
it("starts with an injected optional SFU token issuer", async () => {
|
|
const baseUrl = await start(
|
|
testConfig({
|
|
peerAssistedMedia: true,
|
|
livekitFallback: {
|
|
url: "ws://livekit.test:7880",
|
|
apiUrl: "http://livekit.test:7880",
|
|
apiKey: "test-key",
|
|
apiSecret: "s".repeat(32),
|
|
},
|
|
}),
|
|
{
|
|
sfuTokenIssuer: {
|
|
issueToken: async () => "unused-test-token",
|
|
},
|
|
},
|
|
);
|
|
|
|
const response = await fetch(`${baseUrl}/healthz`);
|
|
expect(response.status).toBe(200);
|
|
});
|
|
|
|
it("clears a stale managed LiveKit room before a restarted server serves traffic", async () => {
|
|
const roomControl = new FakeSfuRoomControl();
|
|
const staleFence = {
|
|
roomId: "42",
|
|
shareGeneration: "stale_share_generation",
|
|
publicationGeneration: "stale_publication_generation",
|
|
};
|
|
roomControl.seedRoom(staleFence, ["host", "viewer:stale"]);
|
|
const baseUrl = await start(
|
|
testConfig({
|
|
peerAssistedMedia: true,
|
|
livekitFallback: {
|
|
url: "ws://livekit.test:7880",
|
|
apiUrl: "http://127.0.0.1:7880",
|
|
apiKey: "test-key",
|
|
apiSecret: "s".repeat(32),
|
|
},
|
|
}),
|
|
{
|
|
sfuTokenIssuer: {
|
|
issueToken: async () => "unused-test-token",
|
|
},
|
|
sfuRoomControl: roomControl,
|
|
},
|
|
);
|
|
|
|
expect(roomControl.initializeCalls).toBe(1);
|
|
expect(roomControl.startupDeletedRoomNames).toHaveLength(1);
|
|
expect(roomControl.rooms.size).toBe(0);
|
|
expect((await fetch(`${baseUrl}/healthz`)).status).toBe(200);
|
|
});
|
|
|
|
it("binds the application listener before LiveKit reconciliation", async () => {
|
|
let releaseInitialization!: () => void;
|
|
const roomControl = new FakeSfuRoomControl();
|
|
roomControl.initializeBarrier = new Promise<void>((resolve) => {
|
|
releaseInitialization = resolve;
|
|
});
|
|
runningServer = await createScreenerServer({
|
|
config: testConfig({
|
|
peerAssistedMedia: true,
|
|
livekitFallback: {
|
|
url: "ws://livekit.test:7880",
|
|
apiUrl: "http://127.0.0.1:7880",
|
|
apiKey: "test-key",
|
|
apiSecret: "s".repeat(32),
|
|
},
|
|
}),
|
|
serveFrontend: false,
|
|
sfuRoomControl: roomControl,
|
|
sfuTokenIssuer: {
|
|
issueToken: async () => "unused-test-token",
|
|
},
|
|
});
|
|
|
|
const listen = runningServer.listen(0, "127.0.0.1");
|
|
try {
|
|
await vi.waitFor(() => {
|
|
expect(roomControl.initializeCalls).toBe(1);
|
|
expect(runningServer?.httpServer.address()).not.toBeNull();
|
|
});
|
|
const address = runningServer.httpServer.address();
|
|
if (!address || typeof address === "string") {
|
|
throw new Error("Expected a TCP server address");
|
|
}
|
|
expect(runningServer.httpServer.listenerCount("upgrade")).toBe(0);
|
|
const response = await fetch(`http://127.0.0.1:${address.port}/healthz`);
|
|
expect(response.status).toBe(503);
|
|
expect(response.headers.get("retry-after")).toBe("1");
|
|
} finally {
|
|
releaseInitialization();
|
|
}
|
|
|
|
await expect(listen).resolves.toBeGreaterThan(0);
|
|
expect(runningServer.httpServer.listenerCount("upgrade")).toBe(1);
|
|
});
|
|
|
|
it("makes no LiveKit call when another process owns the listener", async () => {
|
|
const owner = await createScreenerServer({
|
|
config: testConfig(),
|
|
serveFrontend: false,
|
|
});
|
|
const ownerPort = await owner.listen(0, "127.0.0.1");
|
|
const roomControl = new FakeSfuRoomControl();
|
|
const contender = await createScreenerServer({
|
|
config: testConfig({
|
|
peerAssistedMedia: true,
|
|
livekitFallback: {
|
|
url: "ws://livekit.test:7880",
|
|
apiUrl: "http://127.0.0.1:7880",
|
|
apiKey: "test-key",
|
|
apiSecret: "s".repeat(32),
|
|
},
|
|
}),
|
|
serveFrontend: false,
|
|
sfuRoomControl: roomControl,
|
|
sfuTokenIssuer: {
|
|
issueToken: async () => "unused-test-token",
|
|
},
|
|
});
|
|
|
|
try {
|
|
await expect(contender.listen(ownerPort, "127.0.0.1")).rejects.toMatchObject({
|
|
code: "EADDRINUSE",
|
|
});
|
|
expect(roomControl.initializeCalls).toBe(0);
|
|
} finally {
|
|
await contender.close();
|
|
await owner.close();
|
|
}
|
|
});
|
|
|
|
it("holds listener ownership until an in-flight startup reconciliation settles", async () => {
|
|
let releaseInitialization!: () => void;
|
|
const roomControl = new FakeSfuRoomControl();
|
|
roomControl.initializeBarrier = new Promise<void>((resolve) => {
|
|
releaseInitialization = resolve;
|
|
});
|
|
runningServer = await createScreenerServer({
|
|
config: testConfig({
|
|
peerAssistedMedia: true,
|
|
livekitFallback: {
|
|
url: "ws://livekit.test:7880",
|
|
apiUrl: "http://127.0.0.1:7880",
|
|
apiKey: "test-key",
|
|
apiSecret: "s".repeat(32),
|
|
},
|
|
}),
|
|
serveFrontend: false,
|
|
sfuRoomControl: roomControl,
|
|
sfuTokenIssuer: {
|
|
issueToken: async () => "unused-test-token",
|
|
},
|
|
});
|
|
const listen = runningServer.listen(0, "127.0.0.1");
|
|
await vi.waitFor(() => {
|
|
expect(roomControl.initializeCalls).toBe(1);
|
|
expect(runningServer?.httpServer.address()).not.toBeNull();
|
|
});
|
|
const address = runningServer.httpServer.address();
|
|
if (!address || typeof address === "string") {
|
|
throw new Error("Expected a TCP server address");
|
|
}
|
|
|
|
let closed = false;
|
|
const closing = runningServer.close().then(() => {
|
|
closed = true;
|
|
});
|
|
await new Promise((resolve) => setTimeout(resolve, 20));
|
|
expect(closed).toBe(false);
|
|
expect(runningServer.httpServer.listening).toBe(true);
|
|
|
|
const contender = await createScreenerServer({
|
|
config: testConfig(),
|
|
serveFrontend: false,
|
|
});
|
|
try {
|
|
await expect(
|
|
contender.listen(address.port, "127.0.0.1"),
|
|
).rejects.toMatchObject({ code: "EADDRINUSE" });
|
|
} finally {
|
|
await contender.close();
|
|
}
|
|
|
|
releaseInitialization();
|
|
await expect(listen).resolves.toBe(address.port);
|
|
await closing;
|
|
runningServer = undefined;
|
|
expect(closed).toBe(true);
|
|
});
|
|
|
|
it("closes the listener when LiveKit startup reconciliation fails", async () => {
|
|
const roomControl = new FakeSfuRoomControl();
|
|
roomControl.initializeError = new Error("startup reconciliation failed");
|
|
runningServer = await createScreenerServer({
|
|
config: testConfig({
|
|
peerAssistedMedia: true,
|
|
livekitFallback: {
|
|
url: "ws://livekit.test:7880",
|
|
apiUrl: "http://127.0.0.1:7880",
|
|
apiKey: "test-key",
|
|
apiSecret: "s".repeat(32),
|
|
},
|
|
}),
|
|
serveFrontend: false,
|
|
sfuRoomControl: roomControl,
|
|
sfuTokenIssuer: {
|
|
issueToken: async () => "unused-test-token",
|
|
},
|
|
});
|
|
|
|
await expect(runningServer.listen(0, "127.0.0.1")).rejects.toThrow(
|
|
"startup reconciliation failed",
|
|
);
|
|
expect(roomControl.initializeCalls).toBe(1);
|
|
expect(runningServer.httpServer.listening).toBe(false);
|
|
expect(runningServer.httpServer.listenerCount("upgrade")).toBe(0);
|
|
});
|
|
|
|
it("uses the configured listen host by default", async () => {
|
|
runningServer = await createScreenerServer({
|
|
config: testConfig(),
|
|
serveFrontend: false,
|
|
});
|
|
await runningServer.listen(0);
|
|
const address = runningServer.httpServer.address();
|
|
if (!address || typeof address === "string") {
|
|
throw new Error("Expected a TCP server address");
|
|
}
|
|
expect(address.address).toBe("127.0.0.1");
|
|
});
|
|
|
|
it("reports process liveness without access checks", async () => {
|
|
const baseUrl = await start();
|
|
const response = await fetch(`${baseUrl}/healthz`, {
|
|
headers: { Origin: "https://foreign.test" },
|
|
});
|
|
|
|
expect(response.status).toBe(200);
|
|
expect(response.headers.get("cache-control")).toBe("no-store");
|
|
expect(await response.json()).toEqual({ status: "ok" });
|
|
});
|
|
|
|
it("only accepts GET health checks", async () => {
|
|
const baseUrl = await start();
|
|
const response = await fetch(`${baseUrl}/healthz`, { method: "POST" });
|
|
expect(response.status).toBe(405);
|
|
expect(response.headers.get("allow")).toBe("GET");
|
|
});
|
|
});
|