mirror of
https://github.com/luckyyzh/pi-agent-integrated.git
synced 2026-10-03 11:09:34 +00:00
feat: integrate Pi backend and Pi Web
This commit is contained in:
@@ -0,0 +1,89 @@
|
||||
import { describe, expect, test, vi } from "vitest";
|
||||
import { InteractiveMode } from "../src/modes/interactive/interactive-mode.ts";
|
||||
|
||||
function createSettingsManager(warnings: { anthropicExtraUsage?: boolean } = {}) {
|
||||
return {
|
||||
getWarnings: vi.fn().mockReturnValue(warnings),
|
||||
};
|
||||
}
|
||||
|
||||
function createModelRuntime(credential: { type: "oauth" } | undefined, apiKey?: string) {
|
||||
return {
|
||||
checkAuth: vi.fn().mockResolvedValue(credential),
|
||||
getAuth: vi.fn().mockResolvedValue(apiKey ? { auth: { apiKey } } : undefined),
|
||||
};
|
||||
}
|
||||
|
||||
describe("InteractiveMode.maybeWarnAboutAnthropicSubscriptionAuth", () => {
|
||||
test("warns once when Anthropic subscription auth is detected", async () => {
|
||||
const modelRuntime = createModelRuntime(undefined, "sk-ant-oat01-test");
|
||||
const fakeThis: any = {
|
||||
anthropicSubscriptionWarningShown: false,
|
||||
settingsManager: createSettingsManager(),
|
||||
session: { modelRuntime },
|
||||
showWarning: vi.fn(),
|
||||
};
|
||||
|
||||
await (InteractiveMode as any).prototype.maybeWarnAboutAnthropicSubscriptionAuth.call(fakeThis, {
|
||||
provider: "anthropic",
|
||||
});
|
||||
await (InteractiveMode as any).prototype.maybeWarnAboutAnthropicSubscriptionAuth.call(fakeThis, {
|
||||
provider: "anthropic",
|
||||
});
|
||||
|
||||
expect(fakeThis.showWarning).toHaveBeenCalledTimes(1);
|
||||
expect(modelRuntime.getAuth).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
test("warns when Anthropic OAuth is stored even if token refresh lookup would fail", async () => {
|
||||
const modelRuntime = createModelRuntime({ type: "oauth" });
|
||||
const fakeThis: any = {
|
||||
anthropicSubscriptionWarningShown: false,
|
||||
settingsManager: createSettingsManager(),
|
||||
session: { modelRuntime },
|
||||
showWarning: vi.fn(),
|
||||
};
|
||||
|
||||
await (InteractiveMode as any).prototype.maybeWarnAboutAnthropicSubscriptionAuth.call(fakeThis, {
|
||||
provider: "anthropic",
|
||||
});
|
||||
|
||||
expect(fakeThis.showWarning).toHaveBeenCalledTimes(1);
|
||||
expect(modelRuntime.getAuth).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
test("does not warn for non-Anthropic models", async () => {
|
||||
const modelRuntime = createModelRuntime(undefined);
|
||||
const fakeThis: any = {
|
||||
anthropicSubscriptionWarningShown: false,
|
||||
settingsManager: createSettingsManager(),
|
||||
session: { modelRuntime },
|
||||
showWarning: vi.fn(),
|
||||
};
|
||||
|
||||
await (InteractiveMode as any).prototype.maybeWarnAboutAnthropicSubscriptionAuth.call(fakeThis, {
|
||||
provider: "openai",
|
||||
});
|
||||
|
||||
expect(fakeThis.showWarning).not.toHaveBeenCalled();
|
||||
expect(modelRuntime.getAuth).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
test("does not warn when Anthropic extra usage warning is disabled", async () => {
|
||||
const modelRuntime = createModelRuntime(undefined);
|
||||
const fakeThis: any = {
|
||||
anthropicSubscriptionWarningShown: false,
|
||||
settingsManager: createSettingsManager({ anthropicExtraUsage: false }),
|
||||
session: { modelRuntime },
|
||||
showWarning: vi.fn(),
|
||||
};
|
||||
|
||||
await (InteractiveMode as any).prototype.maybeWarnAboutAnthropicSubscriptionAuth.call(fakeThis, {
|
||||
provider: "anthropic",
|
||||
});
|
||||
|
||||
expect(fakeThis.showWarning).not.toHaveBeenCalled();
|
||||
expect(modelRuntime.checkAuth).not.toHaveBeenCalled();
|
||||
expect(modelRuntime.getAuth).not.toHaveBeenCalled();
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user