339 lines
14 KiB
TypeScript
339 lines
14 KiB
TypeScript
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
|
|
|
function makeStorage() {
|
|
const store = new Map();
|
|
return {
|
|
getItem(key) {
|
|
return store.has(key) ? store.get(key) : null;
|
|
},
|
|
setItem(key, value) {
|
|
store.set(String(key), String(value));
|
|
},
|
|
removeItem(key) {
|
|
store.delete(String(key));
|
|
},
|
|
};
|
|
}
|
|
|
|
function readHeader(headers, name) {
|
|
if (!headers) return null;
|
|
if (typeof headers.get === "function") return headers.get(name);
|
|
return headers[name] ?? headers[String(name || "").toLowerCase()] ?? null;
|
|
}
|
|
|
|
describe("api client request deduplication", () => {
|
|
beforeEach(() => {
|
|
vi.resetModules();
|
|
globalThis.localStorage = makeStorage();
|
|
globalThis.sessionStorage = makeStorage();
|
|
const toastAdd = vi.fn();
|
|
globalThis.app = { ui: {}, extensionManager: { toast: { add: toastAdd } } };
|
|
globalThis.window = {
|
|
addEventListener: vi.fn(),
|
|
removeEventListener: vi.fn(),
|
|
dispatchEvent: vi.fn(),
|
|
location: { origin: "http://localhost" },
|
|
app: globalThis.app,
|
|
};
|
|
globalThis.CustomEvent = class {
|
|
constructor(type, init) {
|
|
this.type = type;
|
|
this.detail = init?.detail;
|
|
}
|
|
};
|
|
globalThis.fetch = vi.fn(async () => ({
|
|
status: 200,
|
|
headers: { get: (name) => (name === "content-type" ? "application/json" : null) },
|
|
json: async () => ({ ok: true, data: { id: 7 } }),
|
|
}));
|
|
});
|
|
|
|
afterEach(() => {
|
|
vi.restoreAllMocks();
|
|
});
|
|
|
|
it("shares the same in-flight promise for duplicate asset metadata requests", async () => {
|
|
let resolveFetch;
|
|
const fetchPromise = new Promise((resolve) => {
|
|
resolveFetch = resolve;
|
|
});
|
|
globalThis.fetch = vi.fn(() => fetchPromise);
|
|
|
|
const client = await import("../api/client.js");
|
|
const first = client.getAssetMetadata(7);
|
|
const _second = client.getAssetMetadata(7);
|
|
|
|
expect(globalThis.fetch).toHaveBeenCalledTimes(0);
|
|
await Promise.resolve();
|
|
expect(globalThis.fetch).toHaveBeenCalledTimes(1);
|
|
|
|
resolveFetch({
|
|
status: 200,
|
|
headers: { get: (name) => (name === "content-type" ? "application/json" : null) },
|
|
json: async () => ({ ok: true, data: { id: 7 } }),
|
|
});
|
|
|
|
const result = await first;
|
|
expect(result.ok).toBe(true);
|
|
|
|
await client.getAssetMetadata(7);
|
|
expect(globalThis.fetch).toHaveBeenCalledTimes(2);
|
|
});
|
|
|
|
it("clears the pending entry after a rejected request", async () => {
|
|
globalThis.fetch = vi.fn(async () => ({
|
|
status: 500,
|
|
headers: { get: (name) => (name === "content-type" ? "application/json" : null) },
|
|
json: async () => ({ ok: false, error: "server sad" }),
|
|
}));
|
|
|
|
const client = await import("../api/client.js");
|
|
const first = await client.vectorFindSimilar(5, 10);
|
|
const second = await client.vectorFindSimilar(5, 10);
|
|
|
|
expect(first.ok).toBe(false);
|
|
expect(second.ok).toBe(false);
|
|
expect(globalThis.fetch).toHaveBeenCalledTimes(2);
|
|
});
|
|
|
|
it("invalidates the available tags cache after a successful tag update", async () => {
|
|
globalThis.fetch = vi.fn(async (url) => {
|
|
if (String(url).includes("/mjr/am/tags")) {
|
|
return {
|
|
status: 200,
|
|
headers: {
|
|
get: (name) => (name === "content-type" ? "application/json" : null),
|
|
},
|
|
json: async () => ({ ok: true, data: ["old-tag"] }),
|
|
};
|
|
}
|
|
if (String(url).includes("/mjr/am/asset/tags")) {
|
|
return {
|
|
status: 200,
|
|
headers: {
|
|
get: (name) => (name === "content-type" ? "application/json" : null),
|
|
},
|
|
json: async () => ({ ok: true, data: { asset_id: 7, tags: [] } }),
|
|
};
|
|
}
|
|
return {
|
|
status: 200,
|
|
headers: { get: (name) => (name === "content-type" ? "application/json" : null) },
|
|
json: async () => ({ ok: true, data: null }),
|
|
};
|
|
});
|
|
|
|
const client = await import("../api/client.js");
|
|
|
|
const first = await client.getAvailableTags();
|
|
expect(first.ok).toBe(true);
|
|
expect(first.data).toEqual(["old-tag"]);
|
|
|
|
await client.updateAssetTags(7, []);
|
|
await client.getAvailableTags();
|
|
|
|
const tagFetches = globalThis.fetch.mock.calls.filter(([url]) =>
|
|
String(url).includes("/mjr/am/tags"),
|
|
);
|
|
expect(tagFetches).toHaveLength(2);
|
|
});
|
|
|
|
it("rewrites bootstrap auth failures into actionable write guidance", async () => {
|
|
globalThis.fetch = vi.fn(async (url) => {
|
|
if (String(url).includes("/mjr/am/settings/security/bootstrap-token")) {
|
|
return {
|
|
status: 403,
|
|
headers: {
|
|
get: (name) => (name === "content-type" ? "application/json" : null),
|
|
},
|
|
json: async () => ({
|
|
ok: false,
|
|
code: "BOOTSTRAP_DISABLED",
|
|
error: "Bootstrap token is disabled for remote clients unless an authenticated ComfyUI user is requesting initial provisioning.",
|
|
}),
|
|
};
|
|
}
|
|
if (String(url).includes("/mjr/am/asset/tags")) {
|
|
return {
|
|
status: 401,
|
|
headers: {
|
|
get: (name) => (name === "content-type" ? "application/json" : null),
|
|
},
|
|
json: async () => ({
|
|
ok: false,
|
|
code: "AUTH_REQUIRED",
|
|
error: "Write operation blocked: missing or invalid API token.",
|
|
}),
|
|
};
|
|
}
|
|
return {
|
|
status: 200,
|
|
headers: { get: (name) => (name === "content-type" ? "application/json" : null) },
|
|
json: async () => ({ ok: true, data: null }),
|
|
};
|
|
});
|
|
|
|
const client = await import("../api/client.js");
|
|
const result = await client.updateAssetTags(7, ["remote"]);
|
|
|
|
expect(result.ok).toBe(false);
|
|
expect(result.error).toContain("Sign in to ComfyUI");
|
|
expect(result.error).toContain("Settings -> Security");
|
|
expect(globalThis.app.extensionManager.toast.add).toHaveBeenCalledTimes(1);
|
|
const [payload] = globalThis.app.extensionManager.toast.add.mock.calls[0];
|
|
expect(String(payload?.summary || "")).toContain("Majoor remote write access");
|
|
expect(String(payload?.detail || "")).toContain("Sign in to ComfyUI");
|
|
});
|
|
|
|
it("explains the HTTP token transport toggle when remote LAN writes use plain HTTP", async () => {
|
|
globalThis.fetch = vi.fn(async (url) => {
|
|
if (String(url).includes("/mjr/am/settings/security/bootstrap-token")) {
|
|
return {
|
|
status: 200,
|
|
headers: {
|
|
get: (name) => (name === "content-type" ? "application/json" : null),
|
|
},
|
|
json: async () => ({ ok: true, data: { token: "mjr_remote_token" } }),
|
|
};
|
|
}
|
|
if (String(url).includes("/mjr/am/asset/tags")) {
|
|
return {
|
|
status: 403,
|
|
headers: {
|
|
get: (name) => (name === "content-type" ? "application/json" : null),
|
|
},
|
|
json: async () => ({
|
|
ok: false,
|
|
code: "FORBIDDEN",
|
|
error: "Write operation blocked: API token over insecure transport.",
|
|
}),
|
|
};
|
|
}
|
|
return {
|
|
status: 200,
|
|
headers: { get: (name) => (name === "content-type" ? "application/json" : null) },
|
|
json: async () => ({ ok: true, data: null }),
|
|
};
|
|
});
|
|
|
|
const client = await import("../api/client.js");
|
|
const result = await client.updateAssetTags(7, ["remote"]);
|
|
|
|
expect(result.ok).toBe(false);
|
|
expect(result.error).toContain("plain HTTP");
|
|
expect(result.error).toContain("Allow HTTP Token Transport");
|
|
expect(globalThis.app.extensionManager.toast.add).toHaveBeenCalledTimes(1);
|
|
const [payload] = globalThis.app.extensionManager.toast.add.mock.calls[0];
|
|
expect(String(payload?.detail || "")).toContain("Allow HTTP Token Transport");
|
|
});
|
|
|
|
it("keeps a bootstrapped token after settings-change cache invalidation", async () => {
|
|
const listeners = new Map();
|
|
globalThis.window.addEventListener = vi.fn((type, listener) => {
|
|
listeners.set(String(type), listener);
|
|
});
|
|
globalThis.window.dispatchEvent = vi.fn((event) => {
|
|
const listener = listeners.get(String(event?.type || ""));
|
|
if (typeof listener === "function") listener(event);
|
|
});
|
|
|
|
const token = "boot-token-1234567890";
|
|
globalThis.fetch = vi.fn(async (url, options = {}) => {
|
|
if (String(url).includes("/mjr/am/settings/security/bootstrap-token")) {
|
|
return {
|
|
status: 200,
|
|
headers: {
|
|
get: (name) => (name === "content-type" ? "application/json" : null),
|
|
},
|
|
json: async () => ({ ok: true, data: { token } }),
|
|
};
|
|
}
|
|
if (String(url).includes("/mjr/am/asset/tags")) {
|
|
const sentToken = readHeader(options.headers, "X-MJR-Token");
|
|
return {
|
|
status: sentToken === token ? 200 : 401,
|
|
headers: {
|
|
get: (name) => (name === "content-type" ? "application/json" : null),
|
|
},
|
|
json: async () =>
|
|
sentToken === token
|
|
? { ok: true, data: { asset_id: 7, tags: ["local"] } }
|
|
: {
|
|
ok: false,
|
|
code: "AUTH_REQUIRED",
|
|
error: "Write operation blocked: missing or invalid API token.",
|
|
},
|
|
};
|
|
}
|
|
return {
|
|
status: 200,
|
|
headers: { get: (name) => (name === "content-type" ? "application/json" : null) },
|
|
json: async () => ({ ok: true, data: null }),
|
|
};
|
|
});
|
|
|
|
const client = await import("../api/client.js");
|
|
const result = await client.updateAssetTags(7, ["local"]);
|
|
|
|
expect(result.ok).toBe(true);
|
|
const tagCall = globalThis.fetch.mock.calls.find(([url]) =>
|
|
String(url).includes("/mjr/am/asset/tags"),
|
|
);
|
|
expect(readHeader(tagCall?.[1]?.headers, "X-MJR-Token")).toBe(token);
|
|
});
|
|
|
|
it("drops a stale header token when bootstrap refreshes only the auth cookie", async () => {
|
|
const staleToken = "stale-token-1234567890";
|
|
let tagCalls = 0;
|
|
globalThis.fetch = vi.fn(async (url, options = {}) => {
|
|
if (String(url).includes("/mjr/am/settings/security/bootstrap-token")) {
|
|
return {
|
|
status: 200,
|
|
headers: {
|
|
get: (name) => (name === "content-type" ? "application/json" : null),
|
|
},
|
|
json: async () => ({ ok: true, data: { token_hint: "...fresh" } }),
|
|
};
|
|
}
|
|
if (String(url).includes("/mjr/am/asset/tags")) {
|
|
tagCalls += 1;
|
|
const sentToken = readHeader(options.headers, "X-MJR-Token");
|
|
if (tagCalls === 1) {
|
|
expect(sentToken).toBe(staleToken);
|
|
return {
|
|
status: 401,
|
|
headers: {
|
|
get: (name) => (name === "content-type" ? "application/json" : null),
|
|
},
|
|
json: async () => ({
|
|
ok: false,
|
|
code: "AUTH_REQUIRED",
|
|
error: "Write operation blocked: missing or invalid API token.",
|
|
}),
|
|
};
|
|
}
|
|
expect(sentToken).toBeNull();
|
|
return {
|
|
status: 200,
|
|
headers: {
|
|
get: (name) => (name === "content-type" ? "application/json" : null),
|
|
},
|
|
json: async () => ({ ok: true, data: { asset_id: 7, tags: ["remote"] } }),
|
|
};
|
|
}
|
|
return {
|
|
status: 200,
|
|
headers: { get: (name) => (name === "content-type" ? "application/json" : null) },
|
|
json: async () => ({ ok: true, data: null }),
|
|
};
|
|
});
|
|
|
|
const client = await import("../api/client.js");
|
|
client.setRuntimeSecurityToken(staleToken);
|
|
const result = await client.updateAssetTags(7, ["remote"]);
|
|
|
|
expect(result.ok).toBe(true);
|
|
expect(tagCalls).toBe(2);
|
|
});
|
|
});
|