wehub-resource-sync
107 lines
3.2 KiB
TypeScript
107 lines
3.2 KiB
TypeScript
import { beforeAll, describe, expect, it, vi } from "bun:test";
|
|
import { LogoutAccountSelectorComponent } from "@oh-my-pi/pi-coding-agent/modes/components/logout-account-selector";
|
|
import { SelectorController } from "@oh-my-pi/pi-coding-agent/modes/controllers/selector-controller";
|
|
import { initTheme } from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
|
|
import type { InteractiveModeContext } from "@oh-my-pi/pi-coding-agent/modes/types";
|
|
import type { AuthStorage, StoredAuthCredential } from "@oh-my-pi/pi-coding-agent/session/auth-storage";
|
|
|
|
interface TestEditorContainer {
|
|
children: unknown[];
|
|
clear: () => void;
|
|
addChild: (child: unknown) => void;
|
|
}
|
|
|
|
function createEditorContainer(): TestEditorContainer {
|
|
return {
|
|
children: [],
|
|
clear() {
|
|
this.children = [];
|
|
},
|
|
addChild(child: unknown) {
|
|
this.children.push(child);
|
|
},
|
|
};
|
|
}
|
|
|
|
function createStoredCredential(id: number, email: string, accountId: string): StoredAuthCredential {
|
|
return {
|
|
id,
|
|
provider: "anthropic",
|
|
disabledCause: null,
|
|
credential: {
|
|
type: "oauth",
|
|
access: `access-${id}`,
|
|
refresh: `refresh-${id}`,
|
|
expires: Date.now() + 60_000,
|
|
email,
|
|
accountId,
|
|
},
|
|
};
|
|
}
|
|
|
|
beforeAll(async () => {
|
|
await initTheme();
|
|
});
|
|
|
|
describe("SelectorController logout", () => {
|
|
it("opens an account picker and removes only the selected credential", async () => {
|
|
const editorContainer = createEditorContainer();
|
|
const credentials = [
|
|
createStoredCredential(21, "a@example.com", "acct-a"),
|
|
createStoredCredential(22, "b@example.com", "acct-b"),
|
|
];
|
|
const removeCredential = vi.fn(async (_provider: string, credentialId: number) => {
|
|
const index = credentials.findIndex(row => row.id === credentialId);
|
|
if (index === -1) return false;
|
|
credentials.splice(index, 1);
|
|
return true;
|
|
});
|
|
const authStorage = {
|
|
reload: vi.fn(async () => undefined),
|
|
listStoredCredentials: (_provider?: string) => credentials,
|
|
getOAuthAccountIdentity: (_provider: string, _sessionId?: string) => ({ accountId: "acct-a" }),
|
|
getCredentialOrigin: (_provider: string) => ({ kind: "oauth" }),
|
|
describeCredentialSource: (_provider: string, _sessionId?: string) => undefined,
|
|
removeCredential,
|
|
} as unknown as AuthStorage;
|
|
const refresh = vi.fn(async () => undefined);
|
|
const presented = Promise.withResolvers<void>();
|
|
const ctx = {
|
|
editorContainer,
|
|
editor: {},
|
|
ui: {
|
|
setFocus: vi.fn(),
|
|
requestRender: vi.fn(),
|
|
},
|
|
session: {
|
|
sessionId: "session-logout-test",
|
|
modelRegistry: {
|
|
authStorage,
|
|
refresh,
|
|
},
|
|
},
|
|
showError: vi.fn(),
|
|
present: vi.fn(() => {
|
|
presented.resolve();
|
|
}),
|
|
} as unknown as InteractiveModeContext;
|
|
const controller = new SelectorController(ctx);
|
|
|
|
await controller.showOAuthSelector("logout", "anthropic");
|
|
|
|
const selector = editorContainer.children[0];
|
|
if (!(selector instanceof LogoutAccountSelectorComponent)) {
|
|
throw new Error("Expected logout account selector");
|
|
}
|
|
selector.handleInput("\x1b[B");
|
|
selector.handleInput("\n");
|
|
await presented.promise;
|
|
|
|
expect(removeCredential).toHaveBeenCalledWith("anthropic", 22);
|
|
expect(credentials.map(row => row.id)).toEqual([21]);
|
|
expect(refresh).toHaveBeenCalled();
|
|
expect(ctx.showError).not.toHaveBeenCalled();
|
|
expect(ctx.present).toHaveBeenCalled();
|
|
});
|
|
});
|