Roo-Code/packages/vscode-shim/src/__tests__/storage.test.ts

import * as fs from "fs"
import * as path from "path"
import { tmpdir } from "os"

import { FileMemento } from "../storage/Memento.js"
import { FileSecretStorage } from "../storage/SecretStorage.js"

describe("FileMemento", () => {
	let tempDir: string
	let mementoPath: string

	beforeEach(() => {
		tempDir = fs.mkdtempSync(path.join(tmpdir(), "memento-test-"))
		mementoPath = path.join(tempDir, "state.json")
	})

	afterEach(() => {
		if (fs.existsSync(tempDir)) {
			fs.rmSync(tempDir, { recursive: true })
		}
	})

	it("should store and retrieve values", async () => {
		const memento = new FileMemento(mementoPath)

		await memento.update("key1", "value1")
		await memento.update("key2", 42)

		expect(memento.get("key1")).toBe("value1")
		expect(memento.get("key2")).toBe(42)
	})

	it("should return default value when key doesn't exist", () => {
		const memento = new FileMemento(mementoPath)

		expect(memento.get("nonexistent", "default")).toBe("default")
		expect(memento.get<number>("missing", 0)).toBe(0)
	})

	it("should persist data to file", async () => {
		const memento1 = new FileMemento(mementoPath)
		await memento1.update("persisted", "value")

		// Create new instance to verify persistence
		const memento2 = new FileMemento(mementoPath)
		expect(memento2.get("persisted")).toBe("value")
	})

	it("should delete values when updated with undefined", async () => {
		const memento = new FileMemento(mementoPath)

		await memento.update("key", "value")
		expect(memento.get("key")).toBe("value")

		await memento.update("key", undefined)
		expect(memento.get("key")).toBeUndefined()
	})

	it("should return all keys", async () => {
		const memento = new FileMemento(mementoPath)

		await memento.update("key1", "value1")
		await memento.update("key2", "value2")
		await memento.update("key3", "value3")

		const keys = memento.keys()
		expect(keys).toHaveLength(3)
		expect(keys).toContain("key1")
		expect(keys).toContain("key2")
		expect(keys).toContain("key3")
	})

	it("should clear all data", async () => {
		const memento = new FileMemento(mementoPath)

		await memento.update("key1", "value1")
		await memento.update("key2", "value2")

		memento.clear()

		expect(memento.keys()).toHaveLength(0)
		expect(memento.get("key1")).toBeUndefined()
	})
})

describe("FileSecretStorage", () => {
	let tempDir: string

	beforeEach(() => {
		tempDir = fs.mkdtempSync(path.join(tmpdir(), "secrets-test-"))
	})

	afterEach(() => {
		if (fs.existsSync(tempDir)) {
			fs.rmSync(tempDir, { recursive: true })
		}
	})

	it("should store and retrieve secrets", async () => {
		const storage = new FileSecretStorage(tempDir)

		await storage.store("apiKey", "sk-test-123")
		const retrieved = await storage.get("apiKey")

		expect(retrieved).toBe("sk-test-123")
	})

	it("should return undefined for non-existent secrets", async () => {
		const storage = new FileSecretStorage(tempDir)
		const result = await storage.get("nonexistent")

		expect(result).toBeUndefined()
	})

	it("should delete secrets", async () => {
		const storage = new FileSecretStorage(tempDir)

		await storage.store("apiKey", "sk-test-123")
		expect(await storage.get("apiKey")).toBe("sk-test-123")

		await storage.delete("apiKey")
		expect(await storage.get("apiKey")).toBeUndefined()
	})

	it("should persist secrets across instances", async () => {
		const storage1 = new FileSecretStorage(tempDir)
		await storage1.store("token", "persistent-value")

		const storage2 = new FileSecretStorage(tempDir)
		const value = await storage2.get("token")

		expect(value).toBe("persistent-value")
	})

	it("should fire onDidChange event when secret changes", async () => {
		const storage = new FileSecretStorage(tempDir)
		const events: string[] = []

		storage.onDidChange((e) => {
			events.push(e.key)
		})

		await storage.store("key1", "value1")
		await storage.store("key2", "value2")
		await storage.delete("key1")

		expect(events).toEqual(["key1", "key2", "key1"])
	})

	it("should clear all secrets", async () => {
		const storage = new FileSecretStorage(tempDir)

		await storage.store("key1", "value1")
		await storage.store("key2", "value2")

		storage.clearAll()

		expect(await storage.get("key1")).toBeUndefined()
		expect(await storage.get("key2")).toBeUndefined()
	})

	it("should create secrets.json file with restrictive permissions on Unix", async () => {
		if (process.platform === "win32") {
			// Skip on Windows
			return
		}

		const storage = new FileSecretStorage(tempDir)
		await storage.store("key", "value")

		const secretsPath = path.join(tempDir, "secrets.json")
		const stats = fs.statSync(secretsPath)
		const mode = stats.mode & 0o777

		// Should be 0600 (owner read/write only)
		expect(mode).toBe(0o600)
	})
})