Spaces:
Running
Running
File size: 2,443 Bytes
aa89cb5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | import { mkdtempSync, rmSync } from 'fs';
import { tmpdir } from 'os';
import { join } from 'path';
import { SQLiteDriver } from '../src/sql/sqlite';
describe('SQLite ID-space generator', () => {
test('uses portable optimistic allocation across independent drivers', async () => {
const directory = mkdtempSync(join(tmpdir(), 'teaql-id-space-'));
const database = join(directory, 'ids.db');
const first = new SQLiteDriver(database);
const second = new SQLiteDriver(database);
try {
await first.ensureSchema({});
expect(await first.transaction(session => first.nextId(session, 'Order'))).toBe('1');
expect(await second.transaction(session => second.nextId(session, 'Order'))).toBe('2');
expect(await first.transaction(session => first.nextId(session, 'Customer'))).toBe('1');
await first.transaction(session => first.ensureIdFloor(session, 'SeededType', '1001'));
expect(await second.transaction(session => second.nextId(session, 'SeededType'))).toBe('1002');
const values: string[] = [];
for (let index = 0; index < 20; index += 1) {
const driver = index % 2 === 0 ? first : second;
values.push(await driver.transaction(session => driver.nextId(session, 'Order')));
}
expect(new Set(values).size).toBe(20);
expect(values.map(Number).sort((a, b) => a - b)).toEqual(
Array.from({ length: 20 }, (_, index) => index + 3),
);
} finally {
await first.close();
await second.close();
rmSync(directory, { recursive: true, force: true });
}
});
test('re-reads and retries after losing an optimistic update', async () => {
const directory = mkdtempSync(join(tmpdir(), 'teaql-id-space-retry-'));
const driver = new SQLiteDriver(join(directory, 'ids.db'));
let selected = 0;
const session = {
async query(sql: string) {
if (sql.startsWith('SELECT')) {
selected += 1;
return { rows: [{ id: selected === 1 ? 7 : 8 }], rowCount: 1 };
}
if (sql.startsWith('UPDATE')) {
return { rows: [], rowCount: selected === 1 ? 0 : 1 };
}
throw new Error(`unexpected SQL: ${sql}`);
},
async *stream() { /* unused */ },
};
try {
expect(await driver.nextId(session, 'Order')).toBe('9');
} finally {
await driver.close();
rmSync(directory, { recursive: true, force: true });
}
});
});
|