diff --git a/src/Settings.test.tsx b/src/Settings.test.tsx index f9aad15..448da99 100644 --- a/src/Settings.test.tsx +++ b/src/Settings.test.tsx @@ -1,5 +1,5 @@ import { describe, it, expect, vi, beforeEach } from 'vitest' -import { render, screen, fireEvent } from '@testing-library/react' +import { render, screen, fireEvent, waitFor } from '@testing-library/react' import Settings from './Settings' describe('Settings Component', () => { @@ -24,12 +24,12 @@ describe('Settings Component', () => { expect(screen.getByText('Appearance')).toBeInTheDocument() }) - it('loads initial state from localStorage', () => { - localStorage.setItem('papercache-apikey', 'sk-test-key') + it('loads initial state from localStorage', async () => { + localStorage.setItem('papercache-apikey-secure', 'sk-test-key') render() const apiKeyInput = screen.getByPlaceholderText('sk-...') as HTMLInputElement - expect(apiKeyInput.value).toBe('sk-test-key') + await waitFor(() => expect(apiKeyInput.value).toBe('sk-test-key')) }) it('updates state when inputs change', () => { @@ -41,7 +41,7 @@ describe('Settings Component', () => { expect((apiKeyInput as HTMLInputElement).value).toBe('sk-new-key') }) - it('saves settings to localStorage on Save Settings button click', () => { + it('saves settings to localStorage on Save Settings button click', async () => { render() const apiKeyInput = screen.getByPlaceholderText('sk-...') @@ -50,14 +50,16 @@ describe('Settings Component', () => { const saveButton = screen.getByText('Save Settings') fireEvent.click(saveButton) - expect(localStorage.getItem('papercache-apikey')).toBe('sk-new-key') - expect(window.electronAPI.closeWindow).toHaveBeenCalled() + await waitFor(() => { + expect(localStorage.getItem('papercache-apikey-secure')).toBe('sk-new-key') + expect(window.electronAPI.closeWindow).toHaveBeenCalled() + }) }) it('calls quitApp when Quit PaperCache is clicked', () => { render() - const quitButton = screen.getByText('Quit PaperCache') + const quitButton = screen.getByText('Quit') fireEvent.click(quitButton) expect(window.electronAPI.quitApp).toHaveBeenCalled() diff --git a/src/setupTests.ts b/src/setupTests.ts index 1351ebb..db112e3 100644 --- a/src/setupTests.ts +++ b/src/setupTests.ts @@ -31,5 +31,7 @@ if (!window.electronAPI) { exportNote: async () => true, openExternal: () => {}, openFile: () => {}, + safeStorageEncrypt: async (val: string) => val, + safeStorageDecrypt: async (val: string) => val, } as any }