113 lines
3.4 KiB
TypeScript
113 lines
3.4 KiB
TypeScript
import net from "node:net";
|
|
|
|
import { describe, expect, it, vi } from "vitest";
|
|
|
|
import {
|
|
CHUTES_TOKEN_ENDPOINT,
|
|
CHUTES_USERINFO_ENDPOINT,
|
|
} from "../agents/chutes-oauth.js";
|
|
import { loginChutes } from "./chutes-oauth.js";
|
|
|
|
async function getFreePort(): Promise<number> {
|
|
return await new Promise((resolve, reject) => {
|
|
const server = net.createServer();
|
|
server.once("error", reject);
|
|
server.listen(0, "127.0.0.1", () => {
|
|
const address = server.address();
|
|
if (!address || typeof address === "string") {
|
|
server.close(() => reject(new Error("No TCP address")));
|
|
return;
|
|
}
|
|
const port = address.port;
|
|
server.close((err) => (err ? reject(err) : resolve(port)));
|
|
});
|
|
});
|
|
}
|
|
|
|
describe("loginChutes", () => {
|
|
it("captures local redirect and exchanges code for tokens", async () => {
|
|
const port = await getFreePort();
|
|
const redirectUri = `http://127.0.0.1:${port}/oauth-callback`;
|
|
|
|
const fetchFn: typeof fetch = async (input, init) => {
|
|
const url = String(input);
|
|
if (url === CHUTES_TOKEN_ENDPOINT) {
|
|
return new Response(
|
|
JSON.stringify({
|
|
access_token: "at_local",
|
|
refresh_token: "rt_local",
|
|
expires_in: 3600,
|
|
}),
|
|
{ status: 200, headers: { "Content-Type": "application/json" } },
|
|
);
|
|
}
|
|
if (url === CHUTES_USERINFO_ENDPOINT) {
|
|
return new Response(JSON.stringify({ username: "local-user" }), {
|
|
status: 200,
|
|
headers: { "Content-Type": "application/json" },
|
|
});
|
|
}
|
|
return fetch(input, init);
|
|
};
|
|
|
|
const onPrompt = vi.fn(async () => {
|
|
throw new Error("onPrompt should not be called for local callback");
|
|
});
|
|
|
|
const creds = await loginChutes({
|
|
app: { clientId: "cid_test", redirectUri, scopes: ["openid"] },
|
|
onAuth: async ({ url }) => {
|
|
const state = new URL(url).searchParams.get("state");
|
|
expect(state).toBeTruthy();
|
|
await fetch(`${redirectUri}?code=code_local&state=${state}`);
|
|
},
|
|
onPrompt,
|
|
fetchFn,
|
|
});
|
|
|
|
expect(onPrompt).not.toHaveBeenCalled();
|
|
expect(creds.access).toBe("at_local");
|
|
expect(creds.refresh).toBe("rt_local");
|
|
expect(creds.email).toBe("local-user");
|
|
});
|
|
|
|
it("supports manual flow with pasted code", async () => {
|
|
const fetchFn: typeof fetch = async (input) => {
|
|
const url = String(input);
|
|
if (url === CHUTES_TOKEN_ENDPOINT) {
|
|
return new Response(
|
|
JSON.stringify({
|
|
access_token: "at_manual",
|
|
refresh_token: "rt_manual",
|
|
expires_in: 3600,
|
|
}),
|
|
{ status: 200, headers: { "Content-Type": "application/json" } },
|
|
);
|
|
}
|
|
if (url === CHUTES_USERINFO_ENDPOINT) {
|
|
return new Response(JSON.stringify({ username: "manual-user" }), {
|
|
status: 200,
|
|
headers: { "Content-Type": "application/json" },
|
|
});
|
|
}
|
|
return new Response("not found", { status: 404 });
|
|
};
|
|
|
|
const creds = await loginChutes({
|
|
app: {
|
|
clientId: "cid_test",
|
|
redirectUri: "http://127.0.0.1:1456/oauth-callback",
|
|
scopes: ["openid"],
|
|
},
|
|
manual: true,
|
|
onAuth: async () => {},
|
|
onPrompt: async () => "code_manual",
|
|
fetchFn,
|
|
});
|
|
|
|
expect(creds.access).toBe("at_manual");
|
|
expect(creds.refresh).toBe("rt_manual");
|
|
expect(creds.email).toBe("manual-user");
|
|
});
|
|
});
|