puter/paseo
1
0
Fork 0
mirror of https://github.com/getpaseo/paseo.git synced 2026-11-11 02:46:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Keep usage logins visible and explain unavailable quota (#5876)

Browse Source 
* fix(usage): keep existing logins visible and explain unavailable quota

Credential discovery must preserve accounts whose login exists even when it has expired. Keep vendor failures on the discovered account and isolate discovery failures without inventing provider cards.

* test(usage): align credential route tests with account discovery
This commit is contained in:
Mohamed Boudra
2026-10-02 00:30:11 +01:00
committed by GitHub
parent b5b43edd65
commit 05e94ab52e
70 changed files with 1919 additions and 1110 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plugins/copilot-usage-source/server/usage.test.ts
+50 -7
View File
@@ -2,7 +2,7 @@ import { mkdtempSync, rmSync } from "node:fs";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import { identify, fetchUsage } from "./usage.js";
import { discover, fetchUsage } from "./usage.js";
import type { UsageReport } from "@getpaseo/plugin/server/usage";
function mockFetch(handlers: Map<string, () => Response>): typeof fetch {
@@ -65,14 +65,14 @@ describe("copilot usage source", () => {
) {
return {
listUsage: async () => {
const report = await fetchUsage({}, (url, init) => fetchApi(url, init));
const report = await fetchFirst((url, init) => fetchApi(url, init));
return {
providers: [
{
providerId: "copilot",
...report,
error: report.error ?? null,
planLabel: report.planLabel ?? null,
error: report.status === "error" ? report.error : null,
planLabel: report.status === "available" ? (report.planLabel ?? null) : null,
},
],
};
@@ -112,17 +112,60 @@ describe("copilot usage source", () => {
});
});
it("identify returns a key when fetch finds copilot credentials", async () => {
it("discovery returns a locator when fetch finds copilot credentials", async () => {
const previous = process.env["COPILOT_TOKEN"];
try {
process.env["COPILOT_TOKEN"] = "fixture-token";
let requested = false;
await fetchUsage({}, async () => {
await fetchFirst(async () => {
requested = true;
return new Response(null, { status: 401 });
});
expect(requested).toBe(true);
expect(await identify()).toEqual({ key: "default" });
expect(await discover()).toEqual([
{ key: "default", input: { store: "env", locator: "COPILOT_TOKEN" } },
]);
} finally {
if (previous === undefined) delete process.env["COPILOT_TOKEN"];
else process.env["COPILOT_TOKEN"] = previous;
}
});
describe("account discovery", () => {
it.each(["empty home", "unrelated files"])("returns no accounts for %s", async (scenario) => {
const { mkdtemp, writeFile, rm } = await import("node:fs/promises");
const directory = await mkdtemp(join(tmpdir(), "usage-empty-"));
const original = { ...process.env };
try {
for (const key of Object.keys(process.env)) delete process.env[key];
process.env.HOME = directory;
process.env.USERPROFILE = directory;
if (scenario === "unrelated files") await writeFile(join(directory, "unrelated.json"), "{}");
expect(await discover()).toEqual([]);
} finally {
for (const key of Object.keys(process.env)) delete process.env[key];
Object.assign(process.env, original);
await rm(directory, { recursive: true, force: true });
}
});
});
async function fetchFirst(fetchApi: typeof fetch) {
const accounts = await discover();
const account = accounts[0];
if (!account) throw new Error("No configured account");
return fetchUsage(account.input as Parameters<typeof fetchUsage>[0], fetchApi);
}
it.each([401, 403])("reports an existing login rejected with HTTP %i", async (status) => {
const previous = process.env["COPILOT_TOKEN"];
try {
process.env["COPILOT_TOKEN"] = "fixture-rejected-login";
const report = await fetchUsage(
{ store: "env", locator: "COPILOT_TOKEN" },
async () => new Response(null, { status }),
);
expect(report).toEqual({ status: "unavailable", problem: { kind: "rejected", status } });
} finally {
if (previous === undefined) delete process.env["COPILOT_TOKEN"];
else process.env["COPILOT_TOKEN"] = previous;