Sunil Srivatsa
35 lines
1.5 KiB
TypeScript
35 lines
1.5 KiB
TypeScript
import { afterEach, beforeEach, expect, test } from "bun:test";
|
|
import { AuthStorage } from "@oh-my-pi/pi-ai";
|
|
import { parseArgs } from "@oh-my-pi/pi-coding-agent/cli/args";
|
|
import { ModelRegistry } from "@oh-my-pi/pi-coding-agent/config/model-registry";
|
|
import { Settings } from "@oh-my-pi/pi-coding-agent/config/settings";
|
|
import { buildSessionOptions } from "@oh-my-pi/pi-coding-agent/main";
|
|
import { SessionManager } from "@oh-my-pi/pi-coding-agent/session/session-manager";
|
|
import { TempDir } from "@oh-my-pi/pi-utils";
|
|
|
|
let tempDir: TempDir;
|
|
let authStorage: AuthStorage;
|
|
|
|
beforeEach(async () => {
|
|
tempDir = await TempDir.create("@cli-explicit-extension-isolation-");
|
|
authStorage = await AuthStorage.create(tempDir.join("auth.db"));
|
|
});
|
|
|
|
afterEach(async () => {
|
|
authStorage.close();
|
|
await tempDir.remove();
|
|
});
|
|
|
|
test("buildSessionOptions retains explicit extensions and hooks under --no-extensions", async () => {
|
|
const extensionPath = tempDir.join("extension-package");
|
|
const hookPath = tempDir.join("hook.ts");
|
|
const parsed = parseArgs(["--no-extensions", "--extension", extensionPath, "--hook", hookPath]);
|
|
const settings = Settings.isolated();
|
|
const modelRegistry = new ModelRegistry(authStorage, tempDir.join("models.yml"));
|
|
|
|
const options = await buildSessionOptions(parsed, [], SessionManager.inMemory(), modelRegistry, settings);
|
|
|
|
expect(options.disableExtensionDiscovery).toBe(true);
|
|
expect(options.additionalExtensionPaths).toEqual([extensionPath, hookPath]);
|
|
});
|