Function startRpcServer

packages/plugin/src/shared/rpc-client.test.ts:59–100

(handler: (method: string) => Response | object)

}

async function startRpcServer(handler: (method: string) => Response | object): Promise<TestServer> {
    const server = createServer(async (req, res) => {
        if (req.method === "GET" && req.url === "/health") {
            res.writeHead(200, { "Content-Type": "application/json" });
            res.end(JSON.stringify({ ok: true }));
            return;
        }

        if (req.method === "POST" && req.url?.startsWith("/rpc/")) {
            const method = req.url.slice("/rpc/".length);
            const result = handler(method);
            if (result instanceof Response) {
                res.writeHead(result.status, { "Content-Type": "application/json" });
                res.end(await result.text());
                return;
            }
            res.writeHead(200, { "Content-Type": "application/json" });
            res.end(JSON.stringify(result));
            return;
        }

        res.writeHead(404);
        res.end("Not Found");
    });

    await new Promise<void>((resolve, reject) => {
        server.once("error", reject);
        server.listen(0, "127.0.0.1", () => resolve());
    });
    const addr = server.address();
    if (!addr || typeof addr === "string") throw new Error("failed to bind test server");

    const testServer = {
        port: addr.port,
        close: () =>
            new Promise<void>((resolve, reject) => {
                server.close((err) => (err ? reject(err) : resolve()));
            }),
    };
    servers.push(testServer);
    return testServer;
}

async function closeServer(server: TestServer): Promise<void> {
    servers = servers.filter((s) => s !== server);