can1357
31b6f0bf31
- Derived descriptors, default-model map, env keys, login list, and refresh dispatch from one ProviderDefinition per provider. - Disabled OpenAI Codex stream obfuscation and interrupted whitespace-only tool-call argument deltas. - Derived auth-broker callback ports and paste-code login set from the registry.
106 lines
3.1 KiB
TypeScript
106 lines
3.1 KiB
TypeScript
import { beforeAll, describe, expect, it } from "bun:test";
|
|
import { getOAuthProviders } from "@oh-my-pi/pi-ai/oauth";
|
|
import { OAuthSelectorComponent } from "@oh-my-pi/pi-coding-agent/modes/components/oauth-selector";
|
|
import { initTheme } from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
|
|
import type { AuthStorage } from "@oh-my-pi/pi-coding-agent/session/auth-storage";
|
|
|
|
beforeAll(async () => {
|
|
await initTheme();
|
|
});
|
|
|
|
const authStorage = {
|
|
has: (_providerId: string) => false,
|
|
hasAuth: (_providerId: string) => false,
|
|
getCredentialOrigin: (_providerId: string) => undefined,
|
|
} as unknown as AuthStorage;
|
|
|
|
describe("OAuthSelectorComponent", () => {
|
|
it("fuzzy-filters overflowing provider lists from typed input", () => {
|
|
const providers = getOAuthProviders();
|
|
expect(providers.length).toBeGreaterThan(10);
|
|
const target =
|
|
providers.find(provider => provider.available && provider.id === "vllm") ??
|
|
providers.find(provider => provider.available) ??
|
|
providers[0];
|
|
expect(target).toBeDefined();
|
|
if (!target) return;
|
|
|
|
const selected: string[] = [];
|
|
const component = new OAuthSelectorComponent(
|
|
"login",
|
|
authStorage,
|
|
providerId => selected.push(providerId),
|
|
() => {},
|
|
);
|
|
|
|
for (const char of target.id) {
|
|
component.handleInput(char);
|
|
}
|
|
|
|
const rendered = component
|
|
.render(80)
|
|
.map(line => Bun.stripANSI(line))
|
|
.join("\n");
|
|
expect(rendered).toContain(target.name);
|
|
expect(rendered).toContain(`Search: ${target.id}`);
|
|
|
|
component.handleInput("\n");
|
|
expect(selected).toEqual([target.id]);
|
|
});
|
|
|
|
it("does not offer env-only providers as logout targets", () => {
|
|
const selected: string[] = [];
|
|
const component = new OAuthSelectorComponent(
|
|
"logout",
|
|
{
|
|
has: (_providerId: string) => false,
|
|
hasAuth: (providerId: string) => providerId === "opencode-go" || providerId === "opencode-zen",
|
|
getCredentialOrigin: (_providerId: string) => undefined,
|
|
} as unknown as AuthStorage,
|
|
providerId => selected.push(providerId),
|
|
() => {},
|
|
);
|
|
|
|
for (const char of "opencode-go") {
|
|
component.handleInput(char);
|
|
}
|
|
|
|
const rendered = component
|
|
.render(80)
|
|
.map(line => Bun.stripANSI(line))
|
|
.join("\n");
|
|
expect(rendered).toContain("No stored provider credentials to log out");
|
|
|
|
component.handleInput("\n");
|
|
expect(selected).toEqual([]);
|
|
});
|
|
|
|
it("offers stored providers as logout targets", () => {
|
|
const selected: string[] = [];
|
|
const component = new OAuthSelectorComponent(
|
|
"logout",
|
|
{
|
|
has: (providerId: string) => providerId === "opencode-go",
|
|
hasAuth: (providerId: string) => providerId === "opencode-go",
|
|
getCredentialOrigin: (_providerId: string) => undefined,
|
|
} as unknown as AuthStorage,
|
|
providerId => selected.push(providerId),
|
|
() => {},
|
|
);
|
|
|
|
for (const char of "opencode-go") {
|
|
component.handleInput(char);
|
|
}
|
|
|
|
const rendered = component
|
|
.render(80)
|
|
.map(line => Bun.stripANSI(line))
|
|
.join("\n");
|
|
expect(rendered).toContain("OpenCode Go");
|
|
expect(rendered).toContain("logged in");
|
|
|
|
component.handleInput("\n");
|
|
expect(selected).toEqual(["opencode-go"]);
|
|
});
|
|
});
|