windmill/frontend/src/lib/test-setup.ts

/**
 * Vitest setup file to mock browser globals for testing
 */

// Provides a real in-memory IndexedDB (indexedDB / IDBKeyRange / structuredClone)
// under the node test environment so the IndexedDB-backed session list and
// chat-history stores can be exercised in unit tests.
import 'fake-indexeddb/auto'
import { vi } from 'vitest'

// Mock localStorage
const localStorageMock = {
	store: {} as Record<string, string>,
	getItem(key: string) {
		return this.store[key] ?? null
	},
	setItem(key: string, value: string) {
		this.store[key] = value
	},
	removeItem(key: string) {
		delete this.store[key]
	},
	clear() {
		this.store = {}
	},
	get length() {
		return Object.keys(this.store).length
	},
	key(index: number) {
		return Object.keys(this.store)[index] ?? null
	}
}

// Mock sessionStorage
const sessionStorageMock = {
	store: {} as Record<string, string>,
	getItem(key: string) {
		return this.store[key] ?? null
	},
	setItem(key: string, value: string) {
		this.store[key] = value
	},
	removeItem(key: string) {
		delete this.store[key]
	},
	clear() {
		this.store = {}
	},
	get length() {
		return Object.keys(this.store).length
	},
	key(index: number) {
		return Object.keys(this.store)[index] ?? null
	}
}

// Assign to globalThis
Object.defineProperty(globalThis, 'localStorage', {
	value: localStorageMock,
	writable: true
})

Object.defineProperty(globalThis, 'sessionStorage', {
	value: sessionStorageMock,
	writable: true
})

// Some modules (e.g. svelte5Utils.useLocalStorageValue) gate browser-only
// behavior on `typeof window`. Provide a minimal window so they don't
// short-circuit during tests.
if (typeof (globalThis as any).window === 'undefined') {
	Object.defineProperty(globalThis, 'window', {
		value: globalThis,
		writable: true,
		configurable: true
	})
}

vi.mock('@codingame/monaco-vscode-standalone-typescript-language-features/worker', () => ({
	TypeScriptWorker: class TypeScriptWorker {
		private _mockScriptSnapshot?: {
			getText: (start: number, end: number) => string
			getLength: () => number
		}

		setMockCode(t: string) {
			this._mockScriptSnapshot = {
				getText: (start: number, end: number) => t.substring(start, end),
				getLength: () => t.length
			}
		}

		getScriptSnapshot(_fileName: string) {
			return this._mockScriptSnapshot
		}
		getScriptVersion(_fileName: string) {
			return '1'
		}
	},
	ts: undefined,
	initialize: undefined
}))