mirror of
https://github.com/openclaw/openclaw.git
synced 2026-05-06 10:50:44 +00:00
test: avoid canvas host socket setup
This commit is contained in:
@@ -1,7 +1,6 @@
|
||||
import fs from "node:fs/promises";
|
||||
import { createServer, type IncomingMessage } from "node:http";
|
||||
import type { IncomingMessage } from "node:http";
|
||||
import { createRequire } from "node:module";
|
||||
import type { AddressInfo } from "node:net";
|
||||
import os from "node:os";
|
||||
import path from "node:path";
|
||||
import type { Duplex } from "node:stream";
|
||||
@@ -24,6 +23,13 @@ type TrackingWebSocket = {
|
||||
send: (message: string) => void;
|
||||
};
|
||||
|
||||
type CapturedResponse = {
|
||||
handled: boolean;
|
||||
status: number;
|
||||
headers: Record<string, number | string | string[]>;
|
||||
body: string;
|
||||
};
|
||||
|
||||
function isLoopbackBindDenied(error: unknown) {
|
||||
const code = (error as NodeJS.ErrnoException | undefined)?.code;
|
||||
return code === "EPERM" || code === "EACCES";
|
||||
@@ -56,6 +62,37 @@ function createMockWatcherState() {
|
||||
};
|
||||
}
|
||||
|
||||
async function captureHandlerResponse(
|
||||
handler: Pick<import("./server.js").CanvasHostHandler, "handleHttpRequest">,
|
||||
url: string,
|
||||
method = "GET",
|
||||
): Promise<CapturedResponse> {
|
||||
const response: CapturedResponse = {
|
||||
handled: false,
|
||||
status: 200,
|
||||
headers: {},
|
||||
body: "",
|
||||
};
|
||||
const res = {
|
||||
statusCode: 200,
|
||||
setHeader(name: string, value: number | string | readonly string[]) {
|
||||
response.headers[name.toLowerCase()] = Array.isArray(value) ? [...value] : value;
|
||||
return this;
|
||||
},
|
||||
end(chunk?: string | Buffer) {
|
||||
response.status = this.statusCode;
|
||||
response.body = Buffer.isBuffer(chunk) ? chunk.toString("utf8") : (chunk ?? "");
|
||||
return this;
|
||||
},
|
||||
};
|
||||
response.handled = await handler.handleHttpRequest(
|
||||
{ method, url } as IncomingMessage,
|
||||
res as import("node:http").ServerResponse,
|
||||
);
|
||||
response.status = res.statusCode;
|
||||
return response;
|
||||
}
|
||||
|
||||
describe("canvas host", () => {
|
||||
const quietRuntime = {
|
||||
...defaultRuntime,
|
||||
@@ -92,12 +129,6 @@ describe("canvas host", () => {
|
||||
...overrides,
|
||||
});
|
||||
|
||||
const fetchCanvasHtml = async (port: number) => {
|
||||
const res = await realFetch(`http://127.0.0.1:${port}${CANVAS_HOST_PATH}/`);
|
||||
const html = await res.text();
|
||||
return { res, html };
|
||||
};
|
||||
|
||||
beforeAll(async () => {
|
||||
vi.doUnmock("undici");
|
||||
vi.resetModules();
|
||||
@@ -128,52 +159,55 @@ describe("canvas host", () => {
|
||||
|
||||
it("creates a default index.html when missing", async () => {
|
||||
const dir = await createCaseDir();
|
||||
let server: Awaited<ReturnType<typeof startFixtureCanvasHost>>;
|
||||
try {
|
||||
server = await startFixtureCanvasHost(dir);
|
||||
} catch (error) {
|
||||
if (isLoopbackBindDenied(error)) {
|
||||
return;
|
||||
}
|
||||
throw error;
|
||||
}
|
||||
const handler = await createCanvasHostHandler({
|
||||
runtime: quietRuntime,
|
||||
rootDir: dir,
|
||||
basePath: CANVAS_HOST_PATH,
|
||||
allowInTests: true,
|
||||
watchFactory: watcherState.watchFactory as unknown as Parameters<
|
||||
typeof createCanvasHostHandler
|
||||
>[0]["watchFactory"],
|
||||
webSocketServerClass: WebSocketServerClass,
|
||||
});
|
||||
|
||||
try {
|
||||
const { res, html } = await fetchCanvasHtml(server.port);
|
||||
expect(res.status).toBe(200);
|
||||
expect(html).toContain("Interactive test page");
|
||||
expect(html).toContain("openclawSendUserAction");
|
||||
expect(html).toContain(CANVAS_WS_PATH);
|
||||
expect(html).toContain('document.createElement("span")');
|
||||
expect(html).not.toContain("statusEl.innerHTML");
|
||||
const response = await captureHandlerResponse(handler, `${CANVAS_HOST_PATH}/`);
|
||||
expect(response.status).toBe(200);
|
||||
expect(response.body).toContain("Interactive test page");
|
||||
expect(response.body).toContain("openclawSendUserAction");
|
||||
expect(response.body).toContain(CANVAS_WS_PATH);
|
||||
expect(response.body).toContain('document.createElement("span")');
|
||||
expect(response.body).not.toContain("statusEl.innerHTML");
|
||||
} finally {
|
||||
await server.close();
|
||||
await handler.close();
|
||||
}
|
||||
});
|
||||
|
||||
it("skips live reload injection when disabled", async () => {
|
||||
const dir = await createCaseDir();
|
||||
await fs.writeFile(path.join(dir, "index.html"), "<html><body>no-reload</body></html>", "utf8");
|
||||
let server: Awaited<ReturnType<typeof startFixtureCanvasHost>>;
|
||||
try {
|
||||
server = await startFixtureCanvasHost(dir, { liveReload: false });
|
||||
} catch (error) {
|
||||
if (isLoopbackBindDenied(error)) {
|
||||
return;
|
||||
}
|
||||
throw error;
|
||||
}
|
||||
const handler = await createCanvasHostHandler({
|
||||
runtime: quietRuntime,
|
||||
rootDir: dir,
|
||||
basePath: CANVAS_HOST_PATH,
|
||||
allowInTests: true,
|
||||
liveReload: false,
|
||||
watchFactory: watcherState.watchFactory as unknown as Parameters<
|
||||
typeof createCanvasHostHandler
|
||||
>[0]["watchFactory"],
|
||||
webSocketServerClass: WebSocketServerClass,
|
||||
});
|
||||
|
||||
try {
|
||||
const { res, html } = await fetchCanvasHtml(server.port);
|
||||
expect(res.status).toBe(200);
|
||||
expect(html).toContain("no-reload");
|
||||
expect(html).not.toContain(CANVAS_WS_PATH);
|
||||
const response = await captureHandlerResponse(handler, `${CANVAS_HOST_PATH}/`);
|
||||
expect(response.status).toBe(200);
|
||||
expect(response.body).toContain("no-reload");
|
||||
expect(response.body).not.toContain(CANVAS_WS_PATH);
|
||||
|
||||
const wsRes = await realFetch(`http://127.0.0.1:${server.port}${CANVAS_WS_PATH}`);
|
||||
expect(wsRes.status).toBe(404);
|
||||
const wsResponse = await captureHandlerResponse(handler, CANVAS_WS_PATH);
|
||||
expect(wsResponse.status).toBe(404);
|
||||
} finally {
|
||||
await server.close();
|
||||
await handler.close();
|
||||
}
|
||||
});
|
||||
|
||||
@@ -192,77 +226,35 @@ describe("canvas host", () => {
|
||||
webSocketServerClass: WebSocketServerClass,
|
||||
});
|
||||
|
||||
const server = createServer((req, res) => {
|
||||
void (async () => {
|
||||
if (await handler.handleHttpRequest(req, res)) {
|
||||
return;
|
||||
}
|
||||
res.statusCode = 404;
|
||||
res.setHeader("Content-Type", "text/plain; charset=utf-8");
|
||||
res.end("Not Found");
|
||||
})();
|
||||
});
|
||||
server.on("upgrade", (req, socket, head) => {
|
||||
if (handler.handleUpgrade(req, socket, head)) {
|
||||
return;
|
||||
}
|
||||
socket.destroy();
|
||||
});
|
||||
try {
|
||||
await new Promise<void>((resolve, reject) => {
|
||||
const onError = (error: Error) => {
|
||||
server.off("listening", onListening);
|
||||
reject(error);
|
||||
};
|
||||
const onListening = () => {
|
||||
server.off("error", onError);
|
||||
resolve();
|
||||
};
|
||||
server.once("error", onError);
|
||||
server.once("listening", onListening);
|
||||
server.listen(0, "127.0.0.1");
|
||||
});
|
||||
} catch (error) {
|
||||
await handler.close();
|
||||
if (isLoopbackBindDenied(error)) {
|
||||
return;
|
||||
}
|
||||
throw error;
|
||||
}
|
||||
const port = (server.address() as AddressInfo).port;
|
||||
|
||||
try {
|
||||
const res = await realFetch(`http://127.0.0.1:${port}${CANVAS_HOST_PATH}/`);
|
||||
const html = await res.text();
|
||||
expect(res.status).toBe(200);
|
||||
expect(html).toContain("v1");
|
||||
expect(html).toContain(CANVAS_WS_PATH);
|
||||
|
||||
const miss = await realFetch(`http://127.0.0.1:${port}/`);
|
||||
expect(miss.status).toBe(404);
|
||||
} finally {
|
||||
await new Promise<void>((resolve, reject) =>
|
||||
server.close((err) => (err ? reject(err) : resolve())),
|
||||
);
|
||||
}
|
||||
const originalClose = handler.close;
|
||||
const closeSpy = vi.fn(async () => originalClose());
|
||||
handler.close = closeSpy;
|
||||
|
||||
const hosted = await startCanvasHost({
|
||||
runtime: quietRuntime,
|
||||
handler,
|
||||
ownsHandler: false,
|
||||
port: 0,
|
||||
listenHost: "127.0.0.1",
|
||||
allowInTests: true,
|
||||
});
|
||||
|
||||
try {
|
||||
expect(hosted.port).toBeGreaterThan(0);
|
||||
const response = await captureHandlerResponse(handler, `${CANVAS_HOST_PATH}/`);
|
||||
expect(response.status).toBe(200);
|
||||
expect(response.body).toContain("v1");
|
||||
expect(response.body).toContain(CANVAS_WS_PATH);
|
||||
|
||||
const miss = await captureHandlerResponse(handler, "/");
|
||||
expect(miss.handled).toBe(false);
|
||||
|
||||
handler.close = closeSpy;
|
||||
const hosted = await startCanvasHost({
|
||||
runtime: quietRuntime,
|
||||
handler,
|
||||
ownsHandler: false,
|
||||
port: 0,
|
||||
listenHost: "127.0.0.1",
|
||||
allowInTests: true,
|
||||
});
|
||||
|
||||
try {
|
||||
expect(hosted.port).toBeGreaterThan(0);
|
||||
} finally {
|
||||
await hosted.close();
|
||||
expect(closeSpy).not.toHaveBeenCalled();
|
||||
}
|
||||
} finally {
|
||||
await hosted.close();
|
||||
expect(closeSpy).not.toHaveBeenCalled();
|
||||
await originalClose();
|
||||
}
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user