Files
plainpages/src/kratos-public.test.ts

122 lines
7.1 KiB
TypeScript

// Kratos public-API client (§4): typed fetch wrappers over Ory Kratos' public endpoints.
// Guards the request contracts (URLs, JSON-accept, cookie relay) and the result mapping
// (200/401/4xx, validation-flow vs success, tokenized JWT). Live wiring is verified by the
// flow pages (§4); these catch contract drift with a mock fetch.
import { test } from "node:test";
import assert from "node:assert/strict";
import { createKratosPublic, KratosError } from "./kratos-public.ts";
const BASE = "http://kratos:4433";
function res(status: number, body?: unknown, setCookie: string[] = []): Response {
const headers = new Headers();
if (body !== undefined) headers.set("content-type", "application/json");
for (const c of setCookie) headers.append("set-cookie", c);
return new Response(body === undefined ? null : JSON.stringify(body), { status, headers });
}
// Records each call so a test can assert URL/method/headers/body.
function recorder(handler: (url: string, init: RequestInit | undefined) => Response) {
const calls: { body: string | undefined; headers: Headers; method: string; url: string }[] = [];
const fetchImpl = (async (input: unknown, init?: RequestInit) => {
calls.push({
body: init?.body as string | undefined,
headers: new Headers(init?.headers),
method: init?.method ?? "GET",
url: String(input),
});
return handler(String(input), init);
}) as typeof fetch;
return { calls, fetchImpl };
}
test("initBrowserFlow gets /self-service/<type>/browser as JSON, relays Set-Cookie, forwards return_to", async () => {
const flow = { id: "f1", ui: { action: `${BASE}/self-service/login?flow=f1`, method: "POST", nodes: [] } };
const { calls, fetchImpl } = recorder(() => res(200, flow, ["csrf_token=abc; Path=/; HttpOnly"]));
const out = await createKratosPublic({ baseUrl: BASE, fetchImpl }).initBrowserFlow("login", { returnTo: "http://app/after" });
assert.deepEqual(out.flow, flow);
assert.deepEqual(out.setCookie, ["csrf_token=abc; Path=/; HttpOnly"]);
assert.match(calls[0]!.url, /\/self-service\/login\/browser\?return_to=http%3A%2F%2Fapp%2Fafter$/);
assert.equal(calls[0]!.headers.get("accept"), "application/json");
});
test("getFlow fetches the flow by id forwarding the browser cookie", async () => {
const flow = { id: "f2", ui: { action: "x", method: "POST", nodes: [] } };
const { calls, fetchImpl } = recorder(() => res(200, flow));
const out = await createKratosPublic({ baseUrl: BASE, fetchImpl }).getFlow("registration", "f2", { cookie: "csrf_token=abc" });
assert.deepEqual(out, flow);
assert.match(calls[0]!.url, /\/self-service\/registration\/flows\?id=f2$/);
assert.equal(calls[0]!.headers.get("cookie"), "csrf_token=abc");
});
test("getFlow throws a KratosError carrying the status when the flow is gone (410)", async () => {
const { fetchImpl } = recorder(() => res(410, { error: { id: "self_service_flow_expired" } }));
await assert.rejects(
createKratosPublic({ baseUrl: BASE, fetchImpl }).getFlow("login", "old"),
(e: unknown) => e instanceof KratosError && e.status === 410,
);
});
test("submitFlow POSTs urlencoded to the action and reports success + relays Set-Cookie", async () => {
const { calls, fetchImpl } = recorder(() => res(200, { session: { active: true } }, ["plainpages_session=s; Path=/"]));
const out = await createKratosPublic({ baseUrl: BASE, fetchImpl })
.submitFlow(`${BASE}/self-service/login?flow=f`, { body: "identifier=a&password=b", cookie: "csrf_token=abc" });
assert.equal(out.ok, true);
assert.equal(out.status, 200);
assert.deepEqual(out.setCookie, ["plainpages_session=s; Path=/"]);
assert.equal(calls[0]!.method, "POST");
assert.equal(calls[0]!.headers.get("content-type"), "application/x-www-form-urlencoded");
assert.equal(calls[0]!.body, "identifier=a&password=b");
});
test("submitFlow returns the re-rendered flow (no throw) on a 400 validation error", async () => {
const flow = { id: "f", ui: { action: "x", messages: [{ id: 4000006, text: "invalid credentials", type: "error" }], method: "POST", nodes: [] } };
const { fetchImpl } = recorder(() => res(400, flow));
const out = await createKratosPublic({ baseUrl: BASE, fetchImpl }).submitFlow(`${BASE}/x`, { body: "x=1" });
assert.equal(out.ok, false);
assert.equal(out.status, 400);
assert.deepEqual(out.body, flow);
});
test("submitFlow surfaces the redirect target — Location header or a 422 redirect_browser_to body", async () => {
const k = (handler: () => Response) => createKratosPublic({ baseUrl: BASE, fetchImpl: recorder(handler).fetchImpl });
const viaHeader = await k(() => new Response(null, { headers: new Headers({ location: "http://app/" }), status: 303 }))
.submitFlow(`${BASE}/x`, { body: "x=1" });
assert.equal(viaHeader.location, "http://app/");
const viaBody = await k(() => res(422, { redirect_browser_to: "http://app/login?flow=next" }))
.submitFlow(`${BASE}/x`, { body: "x=1" });
assert.equal(viaBody.location, "http://app/login?flow=next");
});
test("whoami returns the session on 200 (cookie forwarded) and null on 401", async () => {
const session = { active: true, identity: { id: "u1", traits: { email: "a@b" } } };
const { calls, fetchImpl } = recorder((url) => (url.endsWith("/sessions/whoami") ? res(200, session) : res(401)));
const k = createKratosPublic({ baseUrl: BASE, fetchImpl });
assert.deepEqual(await k.whoami({ cookie: "plainpages_session=s" }), session);
assert.equal(calls[0]!.headers.get("cookie"), "plainpages_session=s");
assert.equal(await createKratosPublic({ baseUrl: BASE, fetchImpl: (async () => res(401)) as typeof fetch }).whoami(), null);
});
test("whoami?tokenize_as mints a session JWT via the tokenizer template", async () => {
const session = { active: true, identity: { id: "u1" }, tokenized: "header.payload.sig" };
const { calls, fetchImpl } = recorder(() => res(200, session));
const out = await createKratosPublic({ baseUrl: BASE, fetchImpl }).whoami({ cookie: "plainpages_session=s", tokenizeAs: "plainpages" });
assert.equal(out?.tokenized, "header.payload.sig");
assert.match(calls[0]!.url, /\/sessions\/whoami\?tokenize_as=plainpages$/);
});
test("whoami throws on an unexpected upstream error", async () => {
const { fetchImpl } = recorder(() => res(500, { error: "boom" }));
await assert.rejects(createKratosPublic({ baseUrl: BASE, fetchImpl }).whoami(), KratosError);
});
test("createLogoutFlow returns the logout URL/token on 200 (cookie forwarded) and null on 401 (no session)", async () => {
const flow = { logout_token: "lt", logout_url: `${BASE}/self-service/logout?token=lt` };
const { calls, fetchImpl } = recorder((url) => (url.endsWith("/self-service/logout/browser") ? res(200, flow) : res(401)));
const out = await createKratosPublic({ baseUrl: BASE, fetchImpl }).createLogoutFlow({ cookie: "plainpages_session=s" });
assert.deepEqual(out, { logoutToken: "lt", logoutUrl: flow.logout_url });
assert.match(calls[0]!.url, /\/self-service\/logout\/browser$/);
assert.equal(calls[0]!.headers.get("cookie"), "plainpages_session=s");
assert.equal(await createKratosPublic({ baseUrl: BASE, fetchImpl: (async () => res(401)) as typeof fetch }).createLogoutFlow(), null);
});