mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-24 14:42:11 -03:00
test(frontend): add storage and core initialization specs
This commit is contained in:
142
static/js/utils/storageHelpers.test.js
Normal file
142
static/js/utils/storageHelpers.test.js
Normal file
@@ -0,0 +1,142 @@
|
||||
import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
|
||||
import * as storageHelpers from './storageHelpers.js';
|
||||
|
||||
const {
|
||||
getStorageItem,
|
||||
setStorageItem,
|
||||
removeStorageItem,
|
||||
getSessionItem,
|
||||
setSessionItem,
|
||||
removeSessionItem,
|
||||
migrateStorageItems
|
||||
} = storageHelpers;
|
||||
|
||||
const createFakeStorage = () => {
|
||||
const store = new Map();
|
||||
return {
|
||||
getItem: vi.fn((key) => (store.has(key) ? store.get(key) : null)),
|
||||
setItem: vi.fn((key, value) => {
|
||||
store.set(key, value);
|
||||
}),
|
||||
removeItem: vi.fn((key) => {
|
||||
store.delete(key);
|
||||
}),
|
||||
clear: vi.fn(() => {
|
||||
store.clear();
|
||||
}),
|
||||
key: vi.fn((index) => Array.from(store.keys())[index] ?? null),
|
||||
get length() {
|
||||
return store.size;
|
||||
},
|
||||
_store: store
|
||||
};
|
||||
};
|
||||
|
||||
let localStorageMock;
|
||||
let sessionStorageMock;
|
||||
let consoleLogMock;
|
||||
|
||||
beforeEach(() => {
|
||||
localStorageMock = createFakeStorage();
|
||||
sessionStorageMock = createFakeStorage();
|
||||
vi.stubGlobal('localStorage', localStorageMock);
|
||||
vi.stubGlobal('sessionStorage', sessionStorageMock);
|
||||
consoleLogMock = vi.spyOn(console, 'log').mockImplementation(() => {});
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
vi.unstubAllGlobals();
|
||||
vi.restoreAllMocks();
|
||||
});
|
||||
|
||||
describe('storageHelpers namespace utilities', () => {
|
||||
it('returns parsed JSON for prefixed localStorage items', () => {
|
||||
localStorage.setItem('lora_manager_preferences', JSON.stringify({ theme: 'dark' }));
|
||||
|
||||
const result = getStorageItem('preferences');
|
||||
|
||||
expect(result).toEqual({ theme: 'dark' });
|
||||
expect(localStorage.getItem).toHaveBeenCalledWith('lora_manager_preferences');
|
||||
});
|
||||
|
||||
it('falls back to legacy keys and migrates them to the namespace', () => {
|
||||
localStorage.setItem('legacy_key', 'value');
|
||||
|
||||
const value = getStorageItem('legacy_key');
|
||||
|
||||
expect(value).toBe('value');
|
||||
expect(localStorage.getItem('lora_manager_legacy_key')).toBe('value');
|
||||
});
|
||||
|
||||
it('serializes objects when setting prefixed localStorage values', () => {
|
||||
const data = { ids: [1, 2, 3] };
|
||||
|
||||
setStorageItem('data', data);
|
||||
|
||||
expect(localStorage.setItem).toHaveBeenCalledWith('lora_manager_data', JSON.stringify(data));
|
||||
expect(localStorage.getItem('lora_manager_data')).toEqual(JSON.stringify(data));
|
||||
});
|
||||
|
||||
it('removes both prefixed and legacy localStorage entries', () => {
|
||||
localStorage.setItem('lora_manager_temp', '123');
|
||||
localStorage.setItem('temp', '456');
|
||||
|
||||
removeStorageItem('temp');
|
||||
|
||||
expect(localStorage.getItem('lora_manager_temp')).toBeNull();
|
||||
expect(localStorage.getItem('temp')).toBeNull();
|
||||
});
|
||||
|
||||
it('returns parsed JSON for session storage items', () => {
|
||||
sessionStorage.setItem('lora_manager_session', JSON.stringify({ page: 'loras' }));
|
||||
|
||||
const session = getSessionItem('session');
|
||||
|
||||
expect(session).toEqual({ page: 'loras' });
|
||||
});
|
||||
|
||||
it('stores primitives in session storage directly', () => {
|
||||
setSessionItem('token', 'abc123');
|
||||
|
||||
expect(sessionStorage.setItem).toHaveBeenCalledWith('lora_manager_token', 'abc123');
|
||||
expect(sessionStorage.getItem('lora_manager_token')).toBe('abc123');
|
||||
});
|
||||
|
||||
it('removes session storage entries by namespace', () => {
|
||||
sessionStorage.setItem('lora_manager_flag', '1');
|
||||
|
||||
removeSessionItem('flag');
|
||||
|
||||
expect(sessionStorage.getItem('lora_manager_flag')).toBeNull();
|
||||
});
|
||||
});
|
||||
|
||||
describe('migrateStorageItems', () => {
|
||||
it('migrates known keys and logs completion', () => {
|
||||
const setStorageSpy = vi.spyOn(storageHelpers, 'setStorageItem');
|
||||
localStorage.setItem('theme', '"light"');
|
||||
localStorage.setItem('loras_filters', JSON.stringify({ sort: 'asc' }));
|
||||
localStorage.setItem('nsfwBlurLevel', '3');
|
||||
|
||||
migrateStorageItems();
|
||||
|
||||
expect(setStorageSpy).toHaveBeenCalledTimes(3);
|
||||
expect(localStorage.getItem('lora_manager_theme')).toBe('light');
|
||||
expect(localStorage.getItem('lora_manager_loras_filters')).toBe(JSON.stringify({ sort: 'asc' }));
|
||||
expect(localStorage.getItem('loras_filters')).toBeNull();
|
||||
expect(localStorage.getItem('lora_manager_nsfwBlurLevel')).toBe('3');
|
||||
expect(localStorage.getItem('nsfwBlurLevel')).toBeNull();
|
||||
expect(localStorage.getItem('lora_manager_migration_completed')).toBe('true');
|
||||
expect(consoleLogMock).toHaveBeenCalledWith('Lora Manager: Storage migration completed');
|
||||
});
|
||||
|
||||
it('skips migration when already completed and logs notice', () => {
|
||||
const setStorageSpy = vi.spyOn(storageHelpers, 'setStorageItem');
|
||||
localStorage.setItem('lora_manager_migration_completed', 'true');
|
||||
|
||||
migrateStorageItems();
|
||||
|
||||
expect(setStorageSpy).not.toHaveBeenCalled();
|
||||
expect(consoleLogMock).toHaveBeenCalledWith('Lora Manager: Storage migration already completed');
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user