mpackdb

2.4 KB

1. import MPackDB from '../src/MPackDB.js';
2. import { rm, writeFile, utimes, stat } from 'fs/promises';
3. import { mkdir } from 'fs/promises';
5. const TEST_DB_PATH = 'db/stale-lock';
7. console.log('--- Stale Lock Recovery Test ---');
8. console.log('A crashed process leaves its lock file behind. Waiters must take it');
9. console.log('over after staleLockTimeout instead of hanging forever.\n');
11. await rm(TEST_DB_PATH, { recursive: true, force: true });
12. await mkdir(TEST_DB_PATH, { recursive: true });
14. // TEST 1: stale lock (old mtime) is removed and the operation proceeds
15. const lockPath = `${TEST_DB_PATH}/users.lock`;
16. await writeFile(lockPath, '99999'); // pretend PID of a dead process
17. const old = (Date.now() - 60000) / 1000; // 60s ago
18. await utimes(lockPath, old, old);
20. const db = new MPackDB(`${TEST_DB_PATH}/users`, { primaryKey: '*id', staleLockTimeout: 1000 });
21. const start = Date.now();
22. const id = await Promise.race([
23. db.insert({ name: 'survivor' }),
24. new Promise(r => setTimeout(() => r('TIMEOUT'), 10000)),
25. ]);
26. const took = Date.now() - start;
27. const test1Pass = id !== 'TIMEOUT';
28. console.log(`TEST 1: insert with stale lock present: ${test1Pass ? `succeeded in ${took}ms` : 'TIMED OUT'}`);
29. console.log(` ${test1Pass ? '✓ PASS' : '✗ FAIL — waiter hung on a dead process lock'}\n`);
31. // TEST 2: a FRESH lock is respected (no premature takeover)
32. const lockPath2 = `${TEST_DB_PATH}/users.lock`;
33. await writeFile(lockPath2, String(process.pid));
34. const start2 = Date.now();
35. const insert2 = db.insert({ name: 'patient' });
36. // While the fresh lock exists the insert must wait; release it after 400ms
37. let insert2Done = false;
38. insert2.then(() => insert2Done = true);
39. await new Promise(r => setTimeout(r, 400));
40. const stillWaiting = !insert2Done;
41. await rm(lockPath2, { force: true });
42. const id2 = await Promise.race([
43. insert2,
44. new Promise(r => setTimeout(() => r('TIMEOUT'), 10000)),
45. ]);
46. const test2Pass = stillWaiting && id2 !== 'TIMEOUT';
47. console.log(`TEST 2: fresh lock respected (waited: ${stillWaiting}), then acquired in ${Date.now() - start2}ms`);
48. console.log(` ${test2Pass ? '✓ PASS' : '✗ FAIL'}\n`);
50. await db.close();
52. // no lock file left behind
53. const lockLeft = await stat(lockPath).then(() => true).catch(() => false);
54. console.log(`Lock file cleaned up: ${!lockLeft ? '✓ PASS' : '✗ FAIL'}`);
56. const allPass = test1Pass && test2Pass && !lockLeft;
57. console.log(`\n${allPass ? '✓ All stale lock tests passed!' : '✗ Some stale lock tests FAILED'}`);
58. if (!allPass) throw new Error('stale-lock tests FAILED');