wehub-resource-sync
137 lines
3.8 KiB
TypeScript
137 lines
3.8 KiB
TypeScript
import {
|
|
afterEach,
|
|
beforeEach,
|
|
describe,
|
|
expect,
|
|
test,
|
|
rs,
|
|
} from "@rstest/core";
|
|
|
|
const ENV_KEYS = [
|
|
"NEXT_PUBLIC_BACKEND_BASE_URL",
|
|
"NEXT_PUBLIC_STATIC_WEBSITE_ONLY",
|
|
] as const;
|
|
|
|
type EnvSnapshot = Partial<
|
|
Record<(typeof ENV_KEYS)[number], string | undefined>
|
|
>;
|
|
|
|
function snapshotEnv(): EnvSnapshot {
|
|
const snapshot: EnvSnapshot = {};
|
|
for (const key of ENV_KEYS) {
|
|
snapshot[key] = process.env[key];
|
|
}
|
|
return snapshot;
|
|
}
|
|
|
|
function setEnv(key: (typeof ENV_KEYS)[number], value: string | undefined) {
|
|
const env = process.env as Record<string, string | undefined>;
|
|
if (value === undefined) {
|
|
delete env[key];
|
|
} else {
|
|
env[key] = value;
|
|
}
|
|
}
|
|
|
|
function restoreEnv(snapshot: EnvSnapshot) {
|
|
for (const key of ENV_KEYS) {
|
|
setEnv(key, snapshot[key]);
|
|
}
|
|
}
|
|
|
|
async function loadFreshArtifactUtils() {
|
|
rs.resetModules();
|
|
return await import("@/core/artifacts/utils");
|
|
}
|
|
|
|
describe("artifact URL helpers", () => {
|
|
let saved: EnvSnapshot;
|
|
|
|
beforeEach(() => {
|
|
saved = snapshotEnv();
|
|
setEnv("NEXT_PUBLIC_BACKEND_BASE_URL", undefined);
|
|
setEnv("NEXT_PUBLIC_STATIC_WEBSITE_ONLY", undefined);
|
|
});
|
|
|
|
afterEach(() => {
|
|
restoreEnv(saved);
|
|
});
|
|
|
|
test("maps static demo artifact paths to bundled public files", async () => {
|
|
setEnv("NEXT_PUBLIC_STATIC_WEBSITE_ONLY", "true");
|
|
|
|
const { resolveArtifactURL, urlOfArtifact } =
|
|
await loadFreshArtifactUtils();
|
|
|
|
expect(
|
|
urlOfArtifact({
|
|
filepath: "/mnt/user-data/outputs/index.html",
|
|
threadId: "thread-1",
|
|
}),
|
|
).toBe("/demo/threads/thread-1/user-data/outputs/index.html");
|
|
expect(
|
|
resolveArtifactURL("/mnt/user-data/outputs/style.css", "thread-1"),
|
|
).toBe("/demo/threads/thread-1/user-data/outputs/style.css");
|
|
});
|
|
|
|
test("returns stable artifact path references", async () => {
|
|
const { extractArtifactsFromThread } = await loadFreshArtifactUtils();
|
|
const threadWithoutArtifacts = { values: {} };
|
|
const artifacts = ["/mnt/user-data/outputs/chart.png"];
|
|
|
|
expect(extractArtifactsFromThread(threadWithoutArtifacts)).toBe(
|
|
extractArtifactsFromThread(threadWithoutArtifacts),
|
|
);
|
|
expect(extractArtifactsFromThread({ values: { artifacts } })).toBe(
|
|
artifacts,
|
|
);
|
|
});
|
|
|
|
test("resolves a relative message image when it uniquely matches an artifact", async () => {
|
|
const { resolveMessageImageURL } = await loadFreshArtifactUtils();
|
|
const artifacts = [
|
|
"/mnt/user-data/outputs/aws-agent-overview.png",
|
|
"/mnt/user-data/outputs/aws-agent-console-config.png",
|
|
];
|
|
|
|
expect(
|
|
resolveMessageImageURL("aws-agent-overview.png", "thread-1", artifacts),
|
|
).toBe(
|
|
"/api/threads/thread-1/artifacts/mnt/user-data/outputs/aws-agent-overview.png",
|
|
);
|
|
expect(
|
|
resolveMessageImageURL(
|
|
"./aws-agent-overview.png#detail",
|
|
"thread-1",
|
|
artifacts,
|
|
),
|
|
).toBe(
|
|
"/api/threads/thread-1/artifacts/mnt/user-data/outputs/aws-agent-overview.png#detail",
|
|
);
|
|
});
|
|
|
|
test("does not rewrite unregistered, ambiguous, or external message images", async () => {
|
|
const { resolveMessageImageURL } = await loadFreshArtifactUtils();
|
|
|
|
expect(resolveMessageImageURL("missing.png", "thread-1", [])).toBe(
|
|
"missing.png",
|
|
);
|
|
expect(
|
|
resolveMessageImageURL("shared.png", "thread-1", [
|
|
"/mnt/user-data/outputs/first/shared.png",
|
|
"/mnt/user-data/outputs/second/shared.png",
|
|
]),
|
|
).toBe("shared.png");
|
|
expect(
|
|
resolveMessageImageURL("../etc/secret.png", "thread-1", [
|
|
"/mnt/user-data/outputs/secret.png",
|
|
]),
|
|
).toBe("../etc/secret.png");
|
|
expect(
|
|
resolveMessageImageURL("https://example.com/image.png", "thread-1", [
|
|
"/mnt/user-data/outputs/image.png",
|
|
]),
|
|
).toBe("https://example.com/image.png");
|
|
});
|
|
});
|