import test from "node:test"; import assert from "node:assert/strict"; import { AI_PROVIDER_PRESETS, DEFAULT_EDITOR_SETTINGS, normalizeAiConfig, normalizeEditorSettings, } from "../../apps/desktop/src/stores/settingsStore.ts"; test("defaults Redis scan page size to 1000 keys", () => { assert.equal(DEFAULT_EDITOR_SETTINGS.redisScanPageSize, 1000); assert.equal(normalizeEditorSettings({}).redisScanPageSize, 1000); }); test("keeps a saved Redis scan page size", () => { assert.equal(normalizeEditorSettings({ redisScanPageSize: 5000 }).redisScanPageSize, 5000); }); test("normalizes saved query result page size", () => { assert.equal(DEFAULT_EDITOR_SETTINGS.pageSize, 100); assert.equal(normalizeEditorSettings({ pageSize: 5000 }).pageSize, 5000); assert.equal(normalizeEditorSettings({ pageSize: 200000 }).pageSize, 100000); assert.equal(normalizeEditorSettings({ pageSize: 0 }).pageSize, 100); }); test("normalizes editor theme settings", () => { assert.equal(DEFAULT_EDITOR_SETTINGS.theme, "app"); assert.equal(normalizeEditorSettings({}).theme, "app"); assert.equal(normalizeEditorSettings({ theme: "app" }).theme, "app"); assert.equal(normalizeEditorSettings({ theme: "vscode-light" }).theme, "vscode-light"); assert.equal(normalizeEditorSettings({ theme: "invalid" as any }).theme, DEFAULT_EDITOR_SETTINGS.theme); }); test("defaults shortcut settings", () => { const settings = normalizeEditorSettings({}); assert.equal(settings.shortcuts.executeSql, "Mod+Enter"); assert.equal(settings.shortcuts.saveSql, "Mod+S"); assert.equal(settings.shortcuts.copyCurrentRow, "Mod+D"); assert.equal(settings.shortcuts.deleteCurrentRow, "Delete"); assert.equal(settings.shortcuts.newQuery, "Mod+T"); assert.equal(settings.shortcuts.focusSearch, "Mod+F"); assert.equal(settings.shortcuts.zoomInUi, "Mod+="); assert.equal(settings.shortcuts.zoomOutUi, "Mod+-"); assert.equal(settings.shortcuts.resetUiZoom, "Mod+0"); assert.equal(settings.shortcuts.refreshData, "F5"); assert.equal(settings.shortcuts.toggleTranspose, "Tab"); }); test("keeps saved shortcut overrides", () => { const settings = normalizeEditorSettings({ shortcuts: { executeSql: "Shift+Mod+Enter", copyCurrentRow: "Alt+Shift+D", deleteCurrentRow: "Backspace", newQuery: "Shift+Mod+N", zoomInUi: "Alt+Mod+=", } as any, }); assert.equal(settings.shortcuts.executeSql, "Shift+Mod+Enter"); assert.equal(settings.shortcuts.copyCurrentRow, "Alt+Shift+D"); assert.equal(settings.shortcuts.deleteCurrentRow, "Backspace"); assert.equal(settings.shortcuts.newQuery, "Shift+Mod+N"); assert.equal(settings.shortcuts.zoomInUi, "Alt+Mod+="); assert.equal(settings.shortcuts.saveSql, "Mod+S"); }); test("defaults sidebar activation to single click", () => { assert.equal(DEFAULT_EDITOR_SETTINGS.sidebarActivation, "single"); assert.equal(normalizeEditorSettings({}).sidebarActivation, "single"); }); test("defaults active tab sidebar selection to off", () => { assert.equal(DEFAULT_EDITOR_SETTINGS.autoSelectActiveSidebarNode, false); assert.equal(normalizeEditorSettings({}).autoSelectActiveSidebarNode, false); }); test("defaults data grid header display settings", () => { assert.equal(DEFAULT_EDITOR_SETTINGS.showColumnCommentsInHeader, false); assert.equal(DEFAULT_EDITOR_SETTINGS.compactColumnHeaderActions, true); assert.equal(normalizeEditorSettings({}).showColumnCommentsInHeader, false); assert.equal(normalizeEditorSettings({}).compactColumnHeaderActions, true); }); test("keeps saved data grid header display settings", () => { const settings = normalizeEditorSettings({ showColumnCommentsInHeader: true, compactColumnHeaderActions: false, } as any); assert.equal(settings.showColumnCommentsInHeader, true); assert.equal(settings.compactColumnHeaderActions, false); }); test("normalizes table structure editor density", () => { assert.equal(DEFAULT_EDITOR_SETTINGS.structureEditorDensity, "compact"); assert.equal(normalizeEditorSettings({}).structureEditorDensity, "compact"); assert.equal(normalizeEditorSettings({ structureEditorDensity: "standard" }).structureEditorDensity, "standard"); assert.equal( normalizeEditorSettings({ structureEditorDensity: "comfortable" }).structureEditorDensity, "comfortable", ); assert.equal(normalizeEditorSettings({ structureEditorDensity: "invalid" as any }).structureEditorDensity, "compact"); }); test("normalizes grid drawer widths", () => { assert.equal(DEFAULT_EDITOR_SETTINGS.tableInfoDrawerWidth, 320); assert.equal(DEFAULT_EDITOR_SETTINGS.cellDetailDrawerWidth, 320); assert.equal(normalizeEditorSettings({}).tableInfoDrawerWidth, 320); assert.equal(normalizeEditorSettings({}).cellDetailDrawerWidth, 320); assert.equal(normalizeEditorSettings({ tableInfoDrawerWidth: 200 } as any).tableInfoDrawerWidth, 240); assert.equal(normalizeEditorSettings({ cellDetailDrawerWidth: 200 } as any).cellDetailDrawerWidth, 260); assert.equal(normalizeEditorSettings({ tableInfoDrawerWidth: 1000 } as any).tableInfoDrawerWidth, 900); assert.equal(normalizeEditorSettings({ cellDetailDrawerWidth: 456.7 } as any).cellDetailDrawerWidth, 457); }); test("keeps saved active tab sidebar selection", () => { assert.equal(normalizeEditorSettings({ autoSelectActiveSidebarNode: true } as any).autoSelectActiveSidebarNode, true); }); test("keeps saved sidebar activation", () => { assert.equal(normalizeEditorSettings({ sidebarActivation: "double" } as any).sidebarActivation, "double"); assert.equal(normalizeEditorSettings({ sidebarActivation: "invalid" } as any).sidebarActivation, "single"); }); test("normalizes saved sidebar hidden table prefixes", () => { assert.deepEqual(DEFAULT_EDITOR_SETTINGS.sidebarHiddenTablePrefixes, []); assert.deepEqual( normalizeEditorSettings({ sidebarHiddenTablePrefixes: [" app_", "app_", "", "ods."] } as any) .sidebarHiddenTablePrefixes, ["app_", "ods."], ); }); test("defaults column formatters to an empty record", () => { assert.deepEqual(DEFAULT_EDITOR_SETTINGS.columnFormatters, {}); assert.deepEqual(normalizeEditorSettings({}).columnFormatters, {}); }); test("keeps only valid saved column formatter configs", () => { const settings = normalizeEditorSettings({ columnFormatters: { "conn::db::public::users::created_at": { kind: "datetime", unit: "auto" }, "conn::db::public::users::bad_date": { kind: "datetime", unit: "bogus" }, "conn::db::public::users::name": { kind: "mask", prefix: 2, suffix: 2 }, "conn::db::public::users::payload": { kind: "json-path", path: "$.user.name" }, "conn::db::public::users::invalid_json": { kind: "json-path", path: "user.name" }, "conn::db::public::users::status": { kind: "custom-ref", formatterId: "fmt_1" }, }, customColumnFormatters: { fmt_1: { id: "fmt_1", name: "Status label", template: "status:${value}" }, fmt_empty_name: { id: "fmt_empty_name", name: "", template: "x:${value}" }, fmt_empty_template: { id: "fmt_empty_template", name: "Broken", template: "" }, }, } as any); assert.deepEqual(settings.columnFormatters, { "conn::db::public::users::created_at": { kind: "datetime", unit: "auto" }, "conn::db::public::users::name": { kind: "mask", prefix: 2, suffix: 2 }, "conn::db::public::users::payload": { kind: "json-path", path: "$.user.name" }, "conn::db::public::users::status": { kind: "custom-ref", formatterId: "fmt_1" }, }); assert.deepEqual(settings.customColumnFormatters, { fmt_1: { id: "fmt_1", name: "Status label", template: "status:${value}" }, }); }); test("AI provider presets include common hosted and local providers", () => { assert.equal(AI_PROVIDER_PRESETS.gemini.endpoint, "https://generativelanguage.googleapis.com"); assert.equal(AI_PROVIDER_PRESETS.gemini.model, "gemini-1.5-pro"); assert.equal(AI_PROVIDER_PRESETS.deepseek.endpoint, "https://api.deepseek.com/v1"); assert.equal(AI_PROVIDER_PRESETS.deepseek.model, "deepseek-v4-flash"); assert.equal(AI_PROVIDER_PRESETS.qwen.endpoint, "https://dashscope.aliyuncs.com/compatible-mode/v1"); assert.equal(AI_PROVIDER_PRESETS.ollama.endpoint, "http://localhost:11434/v1"); assert.equal(AI_PROVIDER_PRESETS.ollama.requiresApiKey, false); assert.equal(AI_PROVIDER_PRESETS.openai.iconSlug, "openai"); assert.equal(AI_PROVIDER_PRESETS.deepseek.iconSlug, "deepseek"); }); test("normalizes legacy AI config and fills provider defaults", () => { const legacy = normalizeAiConfig({ provider: "openai", apiKey: "key", endpoint: "https://api.openai.com/v1/chat/completions", model: "gpt-4o", } as any); assert.equal(legacy.apiStyle, "completions"); assert.equal(legacy.provider, "openai"); assert.equal(legacy.apiKey, "key"); const ollama = normalizeAiConfig({ provider: "ollama" } as any); assert.equal(ollama.endpoint, "http://localhost:11434/v1"); assert.equal(ollama.model, "llama3.1"); assert.equal(ollama.apiKey, ""); }); test("infers legacy AI provider from saved endpoint and model", () => { const deepseek = normalizeAiConfig({ apiKey: "key", endpoint: "https://api.deepseek.com/anthropic/v1/messages", model: "deepseek-v4-pro", } as any); assert.equal(deepseek.provider, "deepseek"); assert.equal(deepseek.endpoint, "https://api.deepseek.com/anthropic/v1/messages"); assert.equal(deepseek.model, "deepseek-v4-pro"); }); test("normalizeEditorSettings falls back to the default UI scale", () => { const settings = normalizeEditorSettings({}); assert.equal(settings.uiScale, DEFAULT_EDITOR_SETTINGS.uiScale); }); test("normalizeEditorSettings clamps UI scale into the supported range", () => { assert.equal(normalizeEditorSettings({ uiScale: 0.2 }).uiScale, 0.75); assert.equal(normalizeEditorSettings({ uiScale: 2.8 }).uiScale, 2); }); test("normalizeEditorSettings keeps valid UI scales with two-decimal precision", () => { assert.equal(normalizeEditorSettings({ uiScale: 1.125 }).uiScale, 1.13); });