109 lines
3.9 KiB
TypeScript
109 lines
3.9 KiB
TypeScript
import { describe, it, expect, beforeEach } from "vitest";
|
|
import { useSettingsStore } from "@/stores/settingsStore";
|
|
import type { ProviderConfig } from "@/lib/tauriCommands";
|
|
|
|
const mockProvider: ProviderConfig = {
|
|
name: "openai",
|
|
api_url: "https://api.openai.com/v1",
|
|
api_key: "sk-test-key",
|
|
model: "gpt-4o",
|
|
};
|
|
|
|
const DEFAULT_PII_PATTERNS = ["email", "ip_address", "phone", "ssn", "credit_card", "hostname", "password", "api_key"];
|
|
|
|
describe("Settings Store", () => {
|
|
beforeEach(() => {
|
|
localStorage.clear();
|
|
useSettingsStore.setState({
|
|
theme: "dark",
|
|
ai_providers: [],
|
|
active_provider: undefined,
|
|
default_provider: "ollama",
|
|
default_model: "llama3.2:3b",
|
|
ollama_url: "http://localhost:11434",
|
|
pii_enabled_patterns: Object.fromEntries(DEFAULT_PII_PATTERNS.map((id) => [id, true])),
|
|
});
|
|
});
|
|
|
|
it("adds a provider", () => {
|
|
useSettingsStore.getState().addProvider(mockProvider);
|
|
expect(useSettingsStore.getState().ai_providers).toHaveLength(1);
|
|
expect(useSettingsStore.getState().ai_providers[0].name).toBe("openai");
|
|
});
|
|
|
|
it("removes a provider", () => {
|
|
useSettingsStore.getState().addProvider(mockProvider);
|
|
useSettingsStore.getState().removeProvider(0);
|
|
expect(useSettingsStore.getState().ai_providers).toHaveLength(0);
|
|
});
|
|
|
|
it("updates a provider", () => {
|
|
useSettingsStore.getState().addProvider(mockProvider);
|
|
useSettingsStore.getState().updateProvider(0, { ...mockProvider, model: "gpt-4o-mini" });
|
|
expect(useSettingsStore.getState().ai_providers[0].model).toBe("gpt-4o-mini");
|
|
});
|
|
|
|
it("toggles theme", () => {
|
|
useSettingsStore.getState().setTheme("light");
|
|
expect(useSettingsStore.getState().theme).toBe("light");
|
|
});
|
|
|
|
it("does not persist API keys to localStorage", () => {
|
|
useSettingsStore.getState().addProvider(mockProvider);
|
|
const raw = localStorage.getItem("tftsr-settings");
|
|
expect(raw).toBeTruthy();
|
|
expect(raw).not.toContain("sk-test-key");
|
|
});
|
|
});
|
|
|
|
describe("Settings Store — PII patterns", () => {
|
|
beforeEach(() => {
|
|
localStorage.clear();
|
|
useSettingsStore.setState({
|
|
theme: "dark",
|
|
ai_providers: [],
|
|
active_provider: undefined,
|
|
default_provider: "ollama",
|
|
default_model: "llama3.2:3b",
|
|
ollama_url: "http://localhost:11434",
|
|
pii_enabled_patterns: Object.fromEntries(DEFAULT_PII_PATTERNS.map((id) => [id, true])),
|
|
});
|
|
});
|
|
|
|
it("initializes all 8 PII patterns as enabled by default", () => {
|
|
const patterns = useSettingsStore.getState().pii_enabled_patterns;
|
|
for (const id of DEFAULT_PII_PATTERNS) {
|
|
expect(patterns[id]).toBe(true);
|
|
}
|
|
});
|
|
|
|
it("setPiiPattern disables a single pattern", () => {
|
|
useSettingsStore.getState().setPiiPattern("email", false);
|
|
expect(useSettingsStore.getState().pii_enabled_patterns["email"]).toBe(false);
|
|
});
|
|
|
|
it("setPiiPattern does not affect other patterns", () => {
|
|
useSettingsStore.getState().setPiiPattern("email", false);
|
|
for (const id of DEFAULT_PII_PATTERNS.filter((id) => id !== "email")) {
|
|
expect(useSettingsStore.getState().pii_enabled_patterns[id]).toBe(true);
|
|
}
|
|
});
|
|
|
|
it("setPiiPattern re-enables a disabled pattern", () => {
|
|
useSettingsStore.getState().setPiiPattern("ssn", false);
|
|
useSettingsStore.getState().setPiiPattern("ssn", true);
|
|
expect(useSettingsStore.getState().pii_enabled_patterns["ssn"]).toBe(true);
|
|
});
|
|
|
|
it("pii_enabled_patterns is persisted to localStorage", () => {
|
|
useSettingsStore.getState().setPiiPattern("api_key", false);
|
|
const raw = localStorage.getItem("tftsr-settings");
|
|
expect(raw).toBeTruthy();
|
|
// Zustand persist wraps state in { state: {...}, version: ... }
|
|
const parsed = JSON.parse(raw!);
|
|
const stored = parsed.state ?? parsed;
|
|
expect(stored.pii_enabled_patterns.api_key).toBe(false);
|
|
expect(stored.pii_enabled_patterns.email).toBe(true);
|
|
});
|
|
});
|