09427086d9
- Migrated settings and auth credential storage from JSON files (settings.json, auth.json) to SQLite database (agent.db). - Added AgentStorage class implementing unified SQLite storage with WAL mode, schema versioning, and prepared statement caching. - Added storage-migration module to handle one-time migration from legacy JSON files to SQLite with backup creation. - Updated credential migration message to reference agent.db instead of auth.json. - Removed file locking mechanism from AuthStorage since SQLite handles concurrency natively. - Added getAgentDbPath() config function returning path to SQLite database file.
105 lines
3.4 KiB
TypeScript
105 lines
3.4 KiB
TypeScript
import { afterEach, beforeEach, describe, expect, it } from "bun:test";
|
|
import { existsSync, mkdirSync, rmSync } from "node:fs";
|
|
import { join } from "node:path";
|
|
import { getAgentDbPath } from "../src/config";
|
|
import { AgentStorage } from "../src/core/agent-storage";
|
|
import { SettingsManager } from "../src/core/settings-manager";
|
|
|
|
describe("SettingsManager", () => {
|
|
let testDir: string;
|
|
let agentDir: string;
|
|
let projectDir: string;
|
|
|
|
beforeEach(() => {
|
|
// Use random UUID to isolate parallel test runs (SQLite files can't be shared)
|
|
testDir = join(process.cwd(), "test-settings-tmp", crypto.randomUUID());
|
|
agentDir = join(testDir, "agent");
|
|
projectDir = join(testDir, "project");
|
|
|
|
if (existsSync(testDir)) {
|
|
rmSync(testDir, { recursive: true });
|
|
}
|
|
mkdirSync(agentDir, { recursive: true });
|
|
mkdirSync(join(projectDir, ".pi"), { recursive: true });
|
|
});
|
|
|
|
afterEach(() => {
|
|
if (existsSync(testDir)) {
|
|
rmSync(testDir, { recursive: true });
|
|
}
|
|
});
|
|
|
|
// Tests that SettingsManager merges with DB state on save rather than blindly overwriting.
|
|
// This ensures external edits (via AgentStorage directly) aren't lost when the app saves.
|
|
describe("preserves externally added settings", () => {
|
|
it("should preserve enabledModels when changing thinking level", () => {
|
|
// Seed initial settings in DB
|
|
const storage = AgentStorage.open(getAgentDbPath(agentDir));
|
|
storage.saveSettings({
|
|
theme: "dark",
|
|
modelRoles: { default: "claude-sonnet" },
|
|
});
|
|
|
|
// Manager loads the initial state
|
|
const manager = SettingsManager.create(projectDir, agentDir);
|
|
|
|
// Simulate external edit (e.g., user modifying DB directly or another process)
|
|
storage.saveSettings({
|
|
theme: "dark",
|
|
modelRoles: { default: "claude-sonnet" },
|
|
enabledModels: ["claude-opus-4-5", "gpt-5.2-codex"],
|
|
});
|
|
|
|
// Manager saves a change - should merge, not overwrite
|
|
manager.setDefaultThinkingLevel("high");
|
|
|
|
const savedSettings = storage.getSettings() ?? {};
|
|
expect(savedSettings.enabledModels).toEqual(["claude-opus-4-5", "gpt-5.2-codex"]);
|
|
expect(savedSettings.defaultThinkingLevel).toBe("high");
|
|
expect(savedSettings.theme).toBe("dark");
|
|
expect(savedSettings.modelRoles?.default).toBe("claude-sonnet");
|
|
});
|
|
|
|
it("should preserve custom settings when changing theme", () => {
|
|
const storage = AgentStorage.open(getAgentDbPath(agentDir));
|
|
storage.saveSettings({
|
|
modelRoles: { default: "claude-sonnet" },
|
|
});
|
|
|
|
const manager = SettingsManager.create(projectDir, agentDir);
|
|
|
|
storage.saveSettings({
|
|
modelRoles: { default: "claude-sonnet" },
|
|
shellPath: "/bin/zsh",
|
|
extensions: ["/path/to/extension.ts"],
|
|
});
|
|
|
|
manager.setTheme("light");
|
|
|
|
const savedSettings = storage.getSettings() ?? {};
|
|
expect(savedSettings.shellPath).toBe("/bin/zsh");
|
|
expect(savedSettings.extensions).toEqual(["/path/to/extension.ts"]);
|
|
expect(savedSettings.theme).toBe("light");
|
|
});
|
|
|
|
it("should let in-memory changes override file changes for same key", () => {
|
|
const storage = AgentStorage.open(getAgentDbPath(agentDir));
|
|
storage.saveSettings({
|
|
theme: "dark",
|
|
});
|
|
|
|
const manager = SettingsManager.create(projectDir, agentDir);
|
|
|
|
storage.saveSettings({
|
|
theme: "dark",
|
|
defaultThinkingLevel: "low",
|
|
});
|
|
|
|
manager.setDefaultThinkingLevel("high");
|
|
|
|
const savedSettings = storage.getSettings() ?? {};
|
|
expect(savedSettings.defaultThinkingLevel).toBe("high");
|
|
});
|
|
});
|
|
});
|