mpackdb

2.3 KB

1. import assert from 'node:assert/strict';
2. import { rm } from 'fs/promises';
3. import MPackDB from '../src/MPackDB.js';
5. const TEST_DB_PATH = 'db/utxo-sized-index-reopen';
7. function makeUtxo(id, amount = 1000) {
8. return {
9. id,
10. stealthAddress: id.repeat(130).slice(0, 8320),
11. viewTag: id.slice(0, 16),
12. amount,
13. symbol: 'ADR',
14. blockNumber: amount,
15. txHash: id.padEnd(64, '0').slice(0, 64),
16. outputIndex: 0,
17. };
18. }
20. async function collect(cursor) {
21. const rows = [];
22. for await (const row of cursor) rows.push(row);
23. return rows;
24. }
26. await rm(TEST_DB_PATH, { recursive: true, force: true });
28. const ids = Array.from({ length: 240 }, (_, i) => `utxo-${String(i).padStart(5, '0')}`);
29. const db = new MPackDB(`${TEST_DB_PATH}/utxos`, {
30. primaryKey: 'id',
31. indexes: ['symbol', '*blockNumber'],
32. indexPersistThreshold: 10000,
33. });
35. for (let i = 0; i < ids.length; i++) {
36. await db.insert(makeUtxo(ids[i], i + 1));
37. }
38. await db.close();
40. const reopened = new MPackDB(`${TEST_DB_PATH}/utxos`, {
41. primaryKey: 'id',
42. indexes: ['symbol', '*blockNumber'],
43. indexPersistThreshold: 10000,
44. });
46. for (const id of ids) {
47. const rows = await collect(reopened.find(id));
48. assert.equal(rows.length, 1, `expected one indexed result for ${id}`);
49. assert.equal(rows[0].id, id);
50. }
51. await reopened.close();
53. const tombstoneDb = new MPackDB(`${TEST_DB_PATH}/tombstones`, {
54. primaryKey: 'id',
55. indexes: ['symbol', '*blockNumber'],
56. indexPersistThreshold: 10000,
57. });
59. await tombstoneDb.insert(makeUtxo('spent-utxo-0001', 12));
60. await tombstoneDb.insert(makeUtxo('live-utxo-0002', 24));
61. await tombstoneDb.delete('spent-utxo-0001');
62. await tombstoneDb.close();
64. const tombstoneReopened = new MPackDB(`${TEST_DB_PATH}/tombstones`, {
65. primaryKey: 'id',
66. indexes: ['symbol', '*blockNumber'],
67. indexPersistThreshold: 10000,
68. });
70. assert.deepEqual(await collect(tombstoneReopened.find('spent-utxo-0001')), []);
71. const liveRows = await collect(tombstoneReopened.find('live-utxo-0002'));
72. assert.equal(liveRows.length, 1);
73. assert.equal(liveRows[0].amount, 24);
74. await tombstoneReopened.compact();
75. assert.deepEqual(await collect(tombstoneReopened.find('spent-utxo-0001')), []);
76. assert.equal((await collect(tombstoneReopened.find('live-utxo-0002'))).length, 1);
77. await tombstoneReopened.close();
78. await rm(TEST_DB_PATH, { recursive: true, force: true });
80. console.log('✓ UTXO-sized indexed reopen/tombstone tests passed');