mpackdb

4.0 KB

1. import MPackDB from '../src/MPackDB.js';
2. import { rm } from 'fs/promises';
4. const TEST_DB_PATH = 'db/causal-lock';
6. console.log('--- Causal Lock Test ---');
7. console.log('An operation STARTED while another operation holds the lock must wait.');
8. console.log('(The old depth-counter lock let any op that started during a critical');
9. console.log('section walk right in — no in-process serialization between lanes.)\n');
11. const sleep = ms => new Promise(r => setTimeout(r, ms));
13. // TEST 1: a withLock section STARTED while another withLock section is in
14. // flight must wait for it. The old depth-counter lock let it re-enter and
15. // interleave (find-then-insert produced duplicates).
16. await rm(TEST_DB_PATH, { recursive: true, force: true });
17. const db = new MPackDB(`${TEST_DB_PATH}/users`, { primaryKey: '*id', indexes: ['email'] });
18. await db.init();
20. const events = [];
22. const findIfAbsentInsert = origin => db.withLock(async () => {
23. events.push(`${origin}:start`);
24. const existing = await db.find(u => u.email === '[email protected]');
25. await sleep(100); // hold the lock with the check result in hand
26. if (existing.length === 0) {
27. await db.insert({ email: '[email protected]', origin });
28. }
29. events.push(`${origin}:end`);
30. });
32. const first = findIfAbsentInsert('first');
33. // Give the first section time to actually acquire the lock, then intrude
34. await sleep(30);
35. const second = findIfAbsentInsert('second');
36. await Promise.all([first, second]);
38. const records = await db.find(u => u.email === '[email protected]');
39. console.log(`TEST 1: records for [email protected]: ${records.length} (expected: 1)`);
40. console.log(` event order: ${events.join(' → ')}`);
41. const test1Pass = records.length === 1
42. && events.indexOf('first:end') < events.indexOf('second:start');
43. console.log(` ${test1Pass ? '✓ PASS' : '✗ FAIL — second withLock entered the critical section mid-flight'}\n`);
45. // TEST 2: nested operations inside withLock still re-enter (no deadlock)
46. console.log('TEST 2: nested find/insert/delete inside withLock (re-entrancy)');
47. const result = await Promise.race([
48. db.withLock(async () => {
49. const inserted = await db.insert({ email: '[email protected]' });
50. const found = await db.find(u => u.email === '[email protected]');
51. await db.delete(u => u.email === '[email protected]');
52. const gone = await db.find(u => u.email === '[email protected]');
53. return { inserted, found: found.length, gone: gone.length };
54. }),
55. sleep(5000).then(() => 'TIMEOUT'),
56. ]);
57. const test2Pass = result !== 'TIMEOUT' && result.found === 1 && result.gone === 0;
58. console.log(` result: ${JSON.stringify(result)}`);
59. console.log(` ${test2Pass ? '✓ PASS' : '✗ FAIL — nested ops deadlocked or misbehaved'}\n`);
61. // TEST 3: update (delete → insert via callback) works under the causal lock
62. console.log('TEST 3: update() nests delete+insert under one lock');
63. await db.insert({ email: '[email protected]', v: 1 });
64. const updated = await Promise.race([
65. db.update(u => u.email === '[email protected]', r => { r.v = 2; return r; }),
66. sleep(5000).then(() => 'TIMEOUT'),
67. ]);
68. const after = await db.find(u => u.email === '[email protected]');
69. const test3Pass = updated !== 'TIMEOUT' && after.length === 1 && after[0].v === 2;
70. console.log(` after update: ${JSON.stringify(after)}`);
71. console.log(` ${test3Pass ? '✓ PASS' : '✗ FAIL'}\n`);
73. // TEST 4: 20 concurrent inserts started in staggered waves stay serialized & unique
74. console.log('TEST 4: staggered concurrent inserts (unique numeric PKs)');
75. const waves = [];
76. for (let i = 0; i < 20; i++) {
77. waves.push((async () => {
78. await sleep(i * 7);
79. return db.insert({ email: `wave-${i}@example.com` });
80. })());
81. }
82. const ids = await Promise.all(waves);
83. const unique = new Set(ids);
84. const test4Pass = unique.size === 20;
85. console.log(` ${ids.length} inserts, ${unique.size} unique ids (expected: 20)`);
86. console.log(` ${test4Pass ? '✓ PASS' : '✗ FAIL — primary key collision'}\n`);
88. await db.close();
90. const allPass = test1Pass && test2Pass && test3Pass && test4Pass;
91. console.log(allPass ? '✓ All causal lock tests passed!' : '✗ Some causal lock tests FAILED');
92. if (!allPass) throw new Error('causal-lock tests FAILED');