|
| 1 | +import { describe, expect, it, beforeEach, afterEach } from 'vitest'; |
| 2 | + |
| 3 | +describe('SqlJsAdapter', () => { |
| 4 | + let SqlJsAdapter: typeof import('../src/adapters/sqlJsAdapter.js').SqlJsAdapter; |
| 5 | + let adapter: InstanceType<typeof SqlJsAdapter>; |
| 6 | + |
| 7 | + beforeEach(async () => { |
| 8 | + const mod = await import('../src/adapters/sqlJsAdapter.js'); |
| 9 | + SqlJsAdapter = mod.SqlJsAdapter; |
| 10 | + adapter = new SqlJsAdapter({ file: ':memory:' }); |
| 11 | + await adapter.open(); |
| 12 | + }); |
| 13 | + |
| 14 | + afterEach(async () => { |
| 15 | + try { |
| 16 | + await adapter.close(); |
| 17 | + } catch { |
| 18 | + /* already closed */ |
| 19 | + } |
| 20 | + }); |
| 21 | + |
| 22 | + it('creates tables and inserts rows', async () => { |
| 23 | + await adapter.exec('CREATE TABLE t (id INTEGER PRIMARY KEY, name TEXT)'); |
| 24 | + await adapter.run('INSERT INTO t (name) VALUES (?)', ['hello']); |
| 25 | + const rows = await adapter.all<{ id: number; name: string }>('SELECT * FROM t'); |
| 26 | + expect(rows).toHaveLength(1); |
| 27 | + expect(rows[0].name).toBe('hello'); |
| 28 | + }); |
| 29 | + |
| 30 | + it('exec() handles multi-statement DDL', async () => { |
| 31 | + // This is the critical fix — exec() must run ALL statements, not just the first |
| 32 | + await adapter.exec(` |
| 33 | + CREATE TABLE IF NOT EXISTS table_a (id TEXT PRIMARY KEY, value TEXT); |
| 34 | + CREATE TABLE IF NOT EXISTS table_b (id TEXT PRIMARY KEY, ref TEXT); |
| 35 | + CREATE TABLE IF NOT EXISTS table_c ( |
| 36 | + id TEXT PRIMARY KEY, |
| 37 | + special_col TEXT, |
| 38 | + created_at INTEGER NOT NULL DEFAULT 0 |
| 39 | + ); |
| 40 | + CREATE INDEX IF NOT EXISTS idx_c_special ON table_c(special_col); |
| 41 | + `); |
| 42 | + |
| 43 | + // Verify all three tables exist |
| 44 | + const tables = await adapter.all<{ name: string }>( |
| 45 | + "SELECT name FROM sqlite_master WHERE type='table' ORDER BY name" |
| 46 | + ); |
| 47 | + const tableNames = tables.map((t) => t.name); |
| 48 | + expect(tableNames).toContain('table_a'); |
| 49 | + expect(tableNames).toContain('table_b'); |
| 50 | + expect(tableNames).toContain('table_c'); |
| 51 | + |
| 52 | + // Verify the special_col column exists on table_c |
| 53 | + const cols = await adapter.all<{ name: string }>('PRAGMA table_info(table_c)'); |
| 54 | + const colNames = cols.map((c) => c.name); |
| 55 | + expect(colNames).toContain('id'); |
| 56 | + expect(colNames).toContain('special_col'); |
| 57 | + expect(colNames).toContain('created_at'); |
| 58 | + |
| 59 | + // Verify the index was created |
| 60 | + const indexes = await adapter.all<{ name: string }>( |
| 61 | + "SELECT name FROM sqlite_master WHERE type='index' AND tbl_name='table_c'" |
| 62 | + ); |
| 63 | + expect(indexes.some((idx) => idx.name === 'idx_c_special')).toBe(true); |
| 64 | + }); |
| 65 | + |
| 66 | + it('exec() with many tables matches foundation DB pattern', async () => { |
| 67 | + // Simulates the wilds.ai foundation DB DDL pattern: 10+ tables in one exec() |
| 68 | + await adapter.exec(` |
| 69 | + CREATE TABLE IF NOT EXISTS companions ( |
| 70 | + companion_id TEXT PRIMARY KEY, |
| 71 | + account_id TEXT NOT NULL, |
| 72 | + slug TEXT NOT NULL |
| 73 | + ); |
| 74 | + CREATE UNIQUE INDEX IF NOT EXISTS idx_companions_slug |
| 75 | + ON companions(account_id, slug); |
| 76 | +
|
| 77 | + CREATE TABLE IF NOT EXISTS sessions ( |
| 78 | + session_id TEXT PRIMARY KEY, |
| 79 | + account_id TEXT NOT NULL, |
| 80 | + blueprint_id TEXT |
| 81 | + ); |
| 82 | +
|
| 83 | + CREATE TABLE IF NOT EXISTS messages ( |
| 84 | + id TEXT PRIMARY KEY, |
| 85 | + session_id TEXT NOT NULL, |
| 86 | + role TEXT NOT NULL, |
| 87 | + content TEXT NOT NULL |
| 88 | + ); |
| 89 | +
|
| 90 | + CREATE TABLE IF NOT EXISTS blueprints ( |
| 91 | + blueprint_id TEXT PRIMARY KEY, |
| 92 | + world_name TEXT, |
| 93 | + genres TEXT, |
| 94 | + visibility TEXT NOT NULL DEFAULT 'public' |
| 95 | + ); |
| 96 | +
|
| 97 | + CREATE TABLE IF NOT EXISTS guest_identities ( |
| 98 | + guest_id TEXT PRIMARY KEY, |
| 99 | + ip_hash TEXT, |
| 100 | + created_at_ms INTEGER NOT NULL, |
| 101 | + last_seen_at_ms INTEGER NOT NULL |
| 102 | + ); |
| 103 | + CREATE INDEX IF NOT EXISTS idx_guest_ip ON guest_identities(ip_hash); |
| 104 | +
|
| 105 | + CREATE TABLE IF NOT EXISTS usage_events ( |
| 106 | + id TEXT PRIMARY KEY, |
| 107 | + actor_id TEXT NOT NULL, |
| 108 | + action TEXT NOT NULL |
| 109 | + ); |
| 110 | + `); |
| 111 | + |
| 112 | + const tables = await adapter.all<{ name: string }>( |
| 113 | + "SELECT name FROM sqlite_master WHERE type='table' ORDER BY name" |
| 114 | + ); |
| 115 | + const names = tables.map((t) => t.name); |
| 116 | + expect(names).toContain('companions'); |
| 117 | + expect(names).toContain('sessions'); |
| 118 | + expect(names).toContain('messages'); |
| 119 | + expect(names).toContain('blueprints'); |
| 120 | + expect(names).toContain('guest_identities'); |
| 121 | + expect(names).toContain('usage_events'); |
| 122 | + |
| 123 | + // Verify ip_hash column exists (this was the original bug) |
| 124 | + const cols = await adapter.all<{ name: string }>('PRAGMA table_info(guest_identities)'); |
| 125 | + expect(cols.map((c) => c.name)).toContain('ip_hash'); |
| 126 | + |
| 127 | + // Verify we can query ip_hash without error |
| 128 | + const result = await adapter.all( |
| 129 | + 'SELECT guest_id FROM guest_identities WHERE ip_hash = ?', |
| 130 | + ['abc'] |
| 131 | + ); |
| 132 | + expect(result).toHaveLength(0); |
| 133 | + }); |
| 134 | + |
| 135 | + it('run() executes single statements', async () => { |
| 136 | + await adapter.exec('CREATE TABLE t (id INTEGER PRIMARY KEY, v TEXT)'); |
| 137 | + const result = await adapter.run('INSERT INTO t (v) VALUES (?)', ['test']); |
| 138 | + expect(result.changes).toBe(1); |
| 139 | + }); |
| 140 | + |
| 141 | + it('get() returns single row or undefined', async () => { |
| 142 | + await adapter.exec('CREATE TABLE t (id INTEGER PRIMARY KEY, v TEXT)'); |
| 143 | + await adapter.run('INSERT INTO t (v) VALUES (?)', ['found']); |
| 144 | + |
| 145 | + const row = await adapter.get<{ v: string }>('SELECT v FROM t WHERE id = 1'); |
| 146 | + expect(row?.v).toBe('found'); |
| 147 | + |
| 148 | + const missing = await adapter.get('SELECT v FROM t WHERE id = 999'); |
| 149 | + expect(missing).toBeFalsy(); |
| 150 | + }); |
| 151 | + |
| 152 | + it('transaction commits on success', async () => { |
| 153 | + await adapter.exec('CREATE TABLE t (id INTEGER PRIMARY KEY, v TEXT)'); |
| 154 | + |
| 155 | + await adapter.transaction(async (trx) => { |
| 156 | + await trx.run('INSERT INTO t (v) VALUES (?)', ['a']); |
| 157 | + await trx.run('INSERT INTO t (v) VALUES (?)', ['b']); |
| 158 | + }); |
| 159 | + |
| 160 | + const rows = await adapter.all('SELECT * FROM t'); |
| 161 | + expect(rows).toHaveLength(2); |
| 162 | + }); |
| 163 | + |
| 164 | + it('transaction rolls back on error', async () => { |
| 165 | + await adapter.exec('CREATE TABLE t (id INTEGER PRIMARY KEY, v TEXT NOT NULL)'); |
| 166 | + |
| 167 | + try { |
| 168 | + await adapter.transaction(async (trx) => { |
| 169 | + await trx.run('INSERT INTO t (v) VALUES (?)', ['ok']); |
| 170 | + throw new Error('deliberate failure'); |
| 171 | + }); |
| 172 | + } catch { |
| 173 | + /* expected */ |
| 174 | + } |
| 175 | + |
| 176 | + const rows = await adapter.all('SELECT * FROM t'); |
| 177 | + expect(rows).toHaveLength(0); |
| 178 | + }); |
| 179 | + |
| 180 | + it('close() prevents further operations', async () => { |
| 181 | + await adapter.close(); |
| 182 | + await expect(adapter.all('SELECT 1')).rejects.toThrow(); |
| 183 | + }); |
| 184 | +}); |
0 commit comments