|
1 | 1 | import { test, expect } from '@playwright/test'; |
2 | | -import * as fs from 'fs'; |
3 | | -import * as path from 'path'; |
4 | | -import { fileURLToPath } from 'url'; |
5 | | - |
6 | | -const __dirname = path.dirname(fileURLToPath(import.meta.url)); |
7 | | - |
8 | | -function getWorkerCode(): string { |
9 | | - const generated = fs.readFileSync(path.join(__dirname, '../src/persistenceWorker.generated.ts'), 'utf8'); |
10 | | - const match = generated.match(/export const WORKER_CODE = (.*);$/ms); |
11 | | - if (!match) throw new Error('Could not extract WORKER_CODE'); |
12 | | - return JSON.parse(match[1]); |
13 | | -} |
14 | | - |
15 | | -function createWorkerHelper(workerCode: string) { |
16 | | - return ` |
17 | | - window.__worker = (() => { |
18 | | - const blob = new Blob([${JSON.stringify(workerCode)}], { type: 'application/javascript' }); |
19 | | - const worker = new Worker(URL.createObjectURL(blob)); |
20 | | - let messageId = 0; |
21 | | - const pending = new Map(); |
22 | | -
|
23 | | - worker.onmessage = (e) => { |
24 | | - const { id, success, error, ...rest } = e.data; |
25 | | - const p = pending.get(id); |
26 | | - if (!p) return; |
27 | | - pending.delete(id); |
28 | | - if (success) p.resolve(rest); |
29 | | - else p.reject(new Error(error || 'Unknown worker error')); |
30 | | - }; |
31 | | -
|
32 | | - return { |
33 | | - post(msg, transfer = []) { |
34 | | - return new Promise((resolve, reject) => { |
35 | | - const id = messageId++; |
36 | | - pending.set(id, { resolve, reject }); |
37 | | - worker.postMessage({ ...msg, id }, transfer); |
38 | | - }); |
39 | | - } |
40 | | - }; |
41 | | - })(); |
42 | | - `; |
43 | | -} |
44 | 2 |
|
45 | 3 | test.describe('sqljsPersistence', () => { |
46 | | - let workerCode: string; |
47 | | - |
48 | | - test.beforeAll(() => { |
49 | | - workerCode = getWorkerCode(); |
50 | | - }); |
51 | | - |
52 | 4 | test.beforeEach(async ({ page }) => { |
53 | 5 | await page.goto('/'); |
54 | | - await page.evaluate(createWorkerHelper(workerCode)); |
55 | 6 | }); |
56 | 7 |
|
57 | 8 | test('save and load round-trips data correctly', async ({ page }) => { |
58 | 9 | const result = await page.evaluate(async () => { |
59 | | - const w = (window as any).__worker; |
| 10 | + const { sqljsPersistence } = (window as any).sqljsLib; |
60 | 11 | const dbName = 'test-roundtrip-' + Date.now(); |
61 | 12 | const passPhrase = 'test-passphrase'; |
62 | 13 |
|
63 | | - // Load creates a new db and returns a key |
64 | | - const loadResult = await w.post({ type: 'load', dbName, passPhrase }); |
65 | | - if (!loadResult.isNew) throw new Error('Expected isNew to be true'); |
| 14 | + const mockSqlJs = { |
| 15 | + Database: class { |
| 16 | + data: Uint8Array; |
| 17 | + constructor(data?: Uint8Array) { this.data = data ?? new Uint8Array(0); } |
| 18 | + export() { return this.data; } |
| 19 | + } |
| 20 | + }; |
| 21 | + |
| 22 | + const { key, salt } = await sqljsPersistence.load(dbName, passPhrase, mockSqlJs); |
66 | 23 |
|
67 | | - // Save some data |
68 | 24 | const testData = new Uint8Array([1, 2, 3, 4, 5, 42, 99, 200]); |
69 | | - await w.post( |
70 | | - { type: 'save', dbName, key: loadResult.key, salt: loadResult.salt, rawData: testData }, |
71 | | - [testData.buffer] |
72 | | - ); |
73 | | - |
74 | | - // Load it back |
75 | | - const loaded = await w.post({ type: 'load', dbName, passPhrase }); |
76 | | - return { |
77 | | - isNew: loaded.isNew, |
78 | | - data: Array.from(new Uint8Array(loaded.rawData)), |
79 | | - }; |
| 25 | + await sqljsPersistence.save(dbName, key, salt, { export: () => testData }); |
| 26 | + |
| 27 | + const { database: loaded } = await sqljsPersistence.load(dbName, passPhrase, mockSqlJs); |
| 28 | + return Array.from(loaded.data); |
80 | 29 | }); |
81 | 30 |
|
82 | | - expect(result.isNew).toBe(false); |
83 | | - expect(result.data).toEqual([1, 2, 3, 4, 5, 42, 99, 200]); |
| 31 | + expect(result).toEqual([1, 2, 3, 4, 5, 42, 99, 200]); |
84 | 32 | }); |
85 | 33 |
|
86 | | - test('load returns isNew for non-existent database', async ({ page }) => { |
| 34 | + test('load returns a new empty database for non-existent db', async ({ page }) => { |
87 | 35 | const result = await page.evaluate(async () => { |
88 | | - const w = (window as any).__worker; |
89 | | - const dbName = 'test-new-' + Date.now(); |
90 | | - const loadResult = await w.post({ type: 'load', dbName, passPhrase: 'pw' }); |
91 | | - return { isNew: loadResult.isNew }; |
| 36 | + const { sqljsPersistence } = (window as any).sqljsLib; |
| 37 | + const mockSqlJs = { |
| 38 | + Database: class { |
| 39 | + data: Uint8Array; |
| 40 | + constructor(data?: Uint8Array) { this.data = data ?? new Uint8Array(0); } |
| 41 | + export() { return this.data; } |
| 42 | + } |
| 43 | + }; |
| 44 | + |
| 45 | + const { database } = await sqljsPersistence.load('test-new-' + Date.now(), 'pw', mockSqlJs); |
| 46 | + return database.data.length; |
92 | 47 | }); |
93 | 48 |
|
94 | | - expect(result.isNew).toBe(true); |
| 49 | + expect(result).toBe(0); |
95 | 50 | }); |
96 | 51 |
|
97 | 52 | test('save overwrites previous data', async ({ page }) => { |
98 | 53 | const result = await page.evaluate(async () => { |
99 | | - const w = (window as any).__worker; |
| 54 | + const { sqljsPersistence } = (window as any).sqljsLib; |
100 | 55 | const dbName = 'test-overwrite-' + Date.now(); |
101 | 56 | const passPhrase = 'pw'; |
| 57 | + const mockSqlJs = { |
| 58 | + Database: class { |
| 59 | + data: Uint8Array; |
| 60 | + constructor(data?: Uint8Array) { this.data = data ?? new Uint8Array(0); } |
| 61 | + export() { return this.data; } |
| 62 | + } |
| 63 | + }; |
102 | 64 |
|
103 | | - const { key, salt } = await w.post({ type: 'load', dbName, passPhrase }); |
104 | | - |
105 | | - // Save first version |
106 | | - const data1 = new Uint8Array([10, 20, 30]); |
107 | | - await w.post({ type: 'save', dbName, key, salt, rawData: data1 }); |
| 65 | + const { key, salt } = await sqljsPersistence.load(dbName, passPhrase, mockSqlJs); |
108 | 66 |
|
109 | | - // Save second version |
110 | | - const data2 = new Uint8Array([77, 88]); |
111 | | - await w.post({ type: 'save', dbName, key, salt, rawData: data2 }); |
| 67 | + await sqljsPersistence.save(dbName, key, salt, { export: () => new Uint8Array([10, 20, 30]) }); |
| 68 | + await sqljsPersistence.save(dbName, key, salt, { export: () => new Uint8Array([77, 88]) }); |
112 | 69 |
|
113 | | - // Load should return second version |
114 | | - const loaded = await w.post({ type: 'load', dbName, passPhrase }); |
115 | | - return Array.from(new Uint8Array(loaded.rawData)); |
| 70 | + const { database: loaded } = await sqljsPersistence.load(dbName, passPhrase, mockSqlJs); |
| 71 | + return Array.from(loaded.data); |
116 | 72 | }); |
117 | 73 |
|
118 | 74 | expect(result).toEqual([77, 88]); |
119 | 75 | }); |
120 | 76 |
|
121 | 77 | test('dbExists returns false for non-existent database', async ({ page }) => { |
122 | 78 | const result = await page.evaluate(async () => { |
123 | | - const w = (window as any).__worker; |
124 | | - const dbName = 'test-noexist-' + Date.now(); |
125 | | - const res = await w.post({ type: 'exists', dbName }); |
126 | | - return res.exists; |
| 79 | + const { sqljsPersistence } = (window as any).sqljsLib; |
| 80 | + return sqljsPersistence.dbExists('test-noexist-' + Date.now()); |
127 | 81 | }); |
128 | 82 |
|
129 | 83 | expect(result).toBe(false); |
130 | 84 | }); |
131 | 85 |
|
132 | 86 | test('dbExists returns true after save', async ({ page }) => { |
133 | 87 | const result = await page.evaluate(async () => { |
134 | | - const w = (window as any).__worker; |
| 88 | + const { sqljsPersistence } = (window as any).sqljsLib; |
135 | 89 | const dbName = 'test-exists-' + Date.now(); |
136 | | - const passPhrase = 'pw'; |
| 90 | + const mockSqlJs = { |
| 91 | + Database: class { |
| 92 | + data: Uint8Array; |
| 93 | + constructor(data?: Uint8Array) { this.data = data ?? new Uint8Array(0); } |
| 94 | + export() { return this.data; } |
| 95 | + } |
| 96 | + }; |
137 | 97 |
|
138 | | - const { key, salt } = await w.post({ type: 'load', dbName, passPhrase }); |
139 | | - const data = new Uint8Array([1, 2, 3]); |
140 | | - await w.post({ type: 'save', dbName, key, salt, rawData: data }, [data.buffer]); |
| 98 | + const { key, salt } = await sqljsPersistence.load(dbName, 'pw', mockSqlJs); |
| 99 | + await sqljsPersistence.save(dbName, key, salt, { export: () => new Uint8Array([1, 2, 3]) }); |
141 | 100 |
|
142 | | - const res = await w.post({ type: 'exists', dbName }); |
143 | | - return res.exists; |
| 101 | + return sqljsPersistence.dbExists(dbName); |
144 | 102 | }); |
145 | 103 |
|
146 | 104 | expect(result).toBe(true); |
|
0 commit comments