108 lines
4.0 KiB
TypeScript
108 lines
4.0 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
import { mkdtemp, rm } from "node:fs/promises";
|
|
import { NextRequest } from "next/server";
|
|
|
|
let runtimeRoot = "";
|
|
let createAuthSession: (typeof import("../src/lib/boss-data"))["createAuthSession"];
|
|
let AUTH_SESSION_COOKIE = "";
|
|
let getConversationHomeRoute: (typeof import("../src/app/api/v1/conversations/home/route"))["GET"];
|
|
let getConversationsRoute: (typeof import("../src/app/api/v1/conversations/route"))["GET"];
|
|
let getFolderRoute: (typeof import("../src/app/api/v1/conversation-folders/[folderKey]/route"))["GET"];
|
|
let getProjectDetailRoute: (typeof import("../src/app/api/v1/projects/[projectId]/route"))["GET"];
|
|
let getDevicesRoute: (typeof import("../src/app/api/v1/devices/route"))["GET"];
|
|
let getSettingsRoute: (typeof import("../src/app/api/v1/settings/route"))["GET"];
|
|
|
|
async function setup() {
|
|
if (runtimeRoot) return;
|
|
|
|
runtimeRoot = await mkdtemp(path.join(os.tmpdir(), "boss-live-data-cache-headers-"));
|
|
process.env.BOSS_RUNTIME_ROOT = runtimeRoot;
|
|
process.env.BOSS_STATE_FILE = path.join(runtimeRoot, "boss-state.json");
|
|
|
|
const [homeRoute, conversationsRoute, folderRoute, projectRoute, devicesRoute, settingsRoute, dataModule, authModule] =
|
|
await Promise.all([
|
|
import("../src/app/api/v1/conversations/home/route.ts"),
|
|
import("../src/app/api/v1/conversations/route.ts"),
|
|
import("../src/app/api/v1/conversation-folders/[folderKey]/route.ts"),
|
|
import("../src/app/api/v1/projects/[projectId]/route.ts"),
|
|
import("../src/app/api/v1/devices/route.ts"),
|
|
import("../src/app/api/v1/settings/route.ts"),
|
|
import("../src/lib/boss-data.ts"),
|
|
import("../src/lib/boss-auth.ts"),
|
|
]);
|
|
|
|
getConversationHomeRoute = homeRoute.GET;
|
|
getConversationsRoute = conversationsRoute.GET;
|
|
getFolderRoute = folderRoute.GET;
|
|
getProjectDetailRoute = projectRoute.GET;
|
|
getDevicesRoute = devicesRoute.GET;
|
|
getSettingsRoute = settingsRoute.GET;
|
|
createAuthSession = dataModule.createAuthSession;
|
|
AUTH_SESSION_COOKIE = authModule.AUTH_SESSION_COOKIE;
|
|
}
|
|
|
|
test.after(async () => {
|
|
if (runtimeRoot) {
|
|
await rm(runtimeRoot, { recursive: true, force: true });
|
|
}
|
|
});
|
|
|
|
async function createAuthedRequest(url: string) {
|
|
const session = await createAuthSession({
|
|
account: "17600003315",
|
|
role: "highest_admin",
|
|
displayName: "Boss 超级管理员",
|
|
loginMethod: "password",
|
|
});
|
|
|
|
return new NextRequest(url, {
|
|
method: "GET",
|
|
headers: {
|
|
cookie: `${AUTH_SESSION_COOKIE}=${session.sessionToken}`,
|
|
},
|
|
});
|
|
}
|
|
|
|
function assertNoStoreHeader(response: Response) {
|
|
assert.equal(response.headers.get("Cache-Control"), "private, no-store, max-age=0");
|
|
}
|
|
|
|
test("live conversation and device routes disable cache storage", async () => {
|
|
await setup();
|
|
|
|
const homeResponse = await getConversationHomeRoute(
|
|
await createAuthedRequest("http://127.0.0.1:3000/api/v1/conversations/home"),
|
|
);
|
|
assertNoStoreHeader(homeResponse);
|
|
|
|
const conversationsResponse = await getConversationsRoute(
|
|
await createAuthedRequest("http://127.0.0.1:3000/api/v1/conversations"),
|
|
);
|
|
assertNoStoreHeader(conversationsResponse);
|
|
|
|
const folderResponse = await getFolderRoute(
|
|
await createAuthedRequest("http://127.0.0.1:3000/api/v1/conversation-folders/mac-studio%3Aboss"),
|
|
{ params: Promise.resolve({ folderKey: "mac-studio%3Aboss" }) },
|
|
);
|
|
assertNoStoreHeader(folderResponse);
|
|
|
|
const projectResponse = await getProjectDetailRoute(
|
|
await createAuthedRequest("http://127.0.0.1:3000/api/v1/projects/master-agent"),
|
|
{ params: Promise.resolve({ projectId: "master-agent" }) },
|
|
);
|
|
assertNoStoreHeader(projectResponse);
|
|
|
|
const devicesResponse = await getDevicesRoute(
|
|
await createAuthedRequest("http://127.0.0.1:3000/api/v1/devices"),
|
|
);
|
|
assertNoStoreHeader(devicesResponse);
|
|
|
|
const settingsResponse = await getSettingsRoute(
|
|
await createAuthedRequest("http://127.0.0.1:3000/api/v1/settings"),
|
|
);
|
|
assertNoStoreHeader(settingsResponse);
|
|
});
|