Disable cache storage on live JSON routes

This commit is contained in:
kris
2026-04-07 13:42:46 +08:00
parent 6153e94000
commit cc1afe8173
8 changed files with 150 additions and 19 deletions

View File

@@ -0,0 +1,107 @@
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);
});