17 lines
842 B
TypeScript
17 lines
842 B
TypeScript
import { afterEach, expect, it } from "vitest";
|
|
import { modelStorageKey, readSavedModel, saveModel } from "../src/model-storage.js";
|
|
import type { ModelConnectionInput } from "@agent-studio/shared";
|
|
afterEach(() => localStorage.clear());
|
|
const config = { providerId: "test", modelId: "test", apiKey: "test-secret" } as ModelConnectionInput;
|
|
it("never persists the API key", () => {
|
|
saveModel(config);
|
|
expect(localStorage.getItem(modelStorageKey)).not.toContain("test-secret");
|
|
expect(readSavedModel()?.apiKey).toBeUndefined();
|
|
expect(config.apiKey).toBe("test-secret");
|
|
});
|
|
it("removes keys from legacy saved configurations", () => {
|
|
localStorage.setItem(modelStorageKey, JSON.stringify(config));
|
|
expect(readSavedModel()?.apiKey).toBeUndefined();
|
|
expect(localStorage.getItem(modelStorageKey)).not.toContain("test-secret");
|
|
});
|