mpackdb

6.1 KB

1. import MPackDB, { PrimaryKeyType } from '../src/MPackDB.js';
2. import { rm } from 'fs/promises';
4. const TEST_DB_PATH = 'db/unique-index';
6. console.log('--- Unique Index Test ---\n');
8. // Clean up
9. await rm(TEST_DB_PATH, { recursive: true, force: true });
11. // ============================================================
12. // TEST 1: Primary key is unique by default
13. // ============================================================
14. console.log('TEST 1: Primary key uniqueness (auto-enforced)');
15. const db1 = new MPackDB(`${TEST_DB_PATH}/test1`, {
16. primaryKey: '*id',
17. indexes: ['email']
18. });
20. await db1.insert({ email: '[email protected]' }); // id=0
21. await db1.insert({ email: '[email protected]' }); // id=1
23. // Try inserting with duplicate primary key
24. try {
25. await db1.insert({ id: 0, email: '[email protected]' }, { skipPrimaryKey: true });
26. console.log(' ✗ FAIL — duplicate primary key was accepted');
27. } catch (e) {
28. if (e.code === 'DUPLICATE_KEY') {
29. console.log(` ✓ PASS — rejected: ${e.message}`);
30. } else {
31. console.log(` ✗ FAIL — wrong error: ${e.message}`);
32. }
33. }
35. const all1 = await db1.find();
36. console.log(` Records: ${all1.length} (expected: 2)\n`);
37. await db1.close();
39. // ============================================================
40. // TEST 2: Unique secondary index (!email)
41. // ============================================================
42. console.log('TEST 2: Unique secondary index (!email)');
43. await rm(`${TEST_DB_PATH}/test2`, { recursive: true, force: true });
44. const db2 = new MPackDB(`${TEST_DB_PATH}/test2`, {
45. primaryKey: '*id',
46. indexes: ['!email', '*age']
47. });
49. await db2.insert({ email: '[email protected]', age: 30 });
50. await db2.insert({ email: '[email protected]', age: 25 });
52. // Try inserting duplicate email
53. try {
54. await db2.insert({ email: '[email protected]', age: 99 });
55. console.log(' ✗ FAIL — duplicate email was accepted');
56. } catch (e) {
57. if (e.code === 'DUPLICATE_KEY' && e.field === 'email') {
58. console.log(` ✓ PASS — rejected: ${e.message}`);
59. } else {
60. console.log(` ✗ FAIL — wrong error: ${e.message}`);
61. }
62. }
64. // Different email should work
65. await db2.insert({ email: '[email protected]', age: 35 });
66. const all2 = await db2.find();
67. console.log(` Records: ${all2.length} (expected: 3)`);
68. console.log(` ${all2.length === 3 ? '✓ PASS' : '✗ FAIL'}\n`);
69. await db2.close();
71. // ============================================================
72. // TEST 3: Unique index after delete (can reuse deleted key)
73. // ============================================================
74. console.log('TEST 3: Unique index after delete');
75. await rm(`${TEST_DB_PATH}/test3`, { recursive: true, force: true });
76. const db3 = new MPackDB(`${TEST_DB_PATH}/test3`, {
77. primaryKey: '*id',
78. indexes: ['!email']
79. });
81. await db3.insert({ email: '[email protected]' });
83. // Delete alice
84. await db3.delete(r => r.email === '[email protected]');
86. // Should now be able to insert same email
87. try {
88. await db3.insert({ email: '[email protected]' });
89. console.log(' ✓ PASS — reinserted after delete');
90. } catch (e) {
91. console.log(` ✗ FAIL — rejected after delete: ${e.message}`);
92. }
94. const all3 = await db3.find();
95. console.log(` Records: ${all3.length} (expected: 1)`);
96. console.log(` ${all3.length === 1 ? '✓ PASS' : '✗ FAIL'}\n`);
97. await db3.close();
99. // ============================================================
100. // TEST 4: Concurrent insert with unique index (the login pattern)
101. // ============================================================
102. console.log('TEST 4: Concurrent insert with unique email (race prevention)');
103. await rm(`${TEST_DB_PATH}/test4`, { recursive: true, force: true });
104. const db4 = new MPackDB(`${TEST_DB_PATH}/test4`, {
105. primaryKey: '@id',
106. indexes: ['!email']
107. });
109. // Simulate concurrent login: two inserts for same email at once
110. const results = await Promise.allSettled([
111. db4.insert({ email: '[email protected]' }),
112. db4.insert({ email: '[email protected]' }),
113. db4.insert({ email: '[email protected]' }),
114. ]);
116. const fulfilled = results.filter(r => r.status === 'fulfilled');
117. const rejected = results.filter(r => r.status === 'rejected' && r.reason.code === 'DUPLICATE_KEY');
119. console.log(` Succeeded: ${fulfilled.length} (expected: 1)`);
120. console.log(` Rejected (DUPLICATE_KEY): ${rejected.length} (expected: 2)`);
122. const all4 = await db4.find();
123. console.log(` Records in DB: ${all4.length} (expected: 1)`);
124. const pass4 = fulfilled.length === 1 && rejected.length === 2 && all4.length === 1;
125. console.log(` ${pass4 ? '✓ PASS' : '✗ FAIL'}\n`);
126. await db4.close();
128. // ============================================================
129. // TEST 5: Non-unique index allows duplicates
130. // ============================================================
131. console.log('TEST 5: Non-unique index allows duplicates');
132. await rm(`${TEST_DB_PATH}/test5`, { recursive: true, force: true });
133. const db5 = new MPackDB(`${TEST_DB_PATH}/test5`, {
134. primaryKey: '*id',
135. indexes: ['category'] // no ! prefix = not unique
136. });
138. await db5.insert({ category: 'Electronics' });
139. await db5.insert({ category: 'Electronics' });
140. await db5.insert({ category: 'Electronics' });
142. const all5 = await db5.find();
143. console.log(` Records: ${all5.length} (expected: 3)`);
144. console.log(` ${all5.length === 3 ? '✓ PASS — duplicates allowed' : '✗ FAIL'}\n`);
145. await db5.close();
147. // ============================================================
148. // TEST 6: Unique numeric index (!*score)
149. // ============================================================
150. console.log('TEST 6: Unique numeric index (!*score)');
151. await rm(`${TEST_DB_PATH}/test6`, { recursive: true, force: true });
152. const db6 = new MPackDB(`${TEST_DB_PATH}/test6`, {
153. primaryKey: '*id',
154. indexes: ['!*score']
155. });
157. await db6.insert({ score: 100 });
158. await db6.insert({ score: 200 });
160. try {
161. await db6.insert({ score: 100 });
162. console.log(' ✗ FAIL — duplicate score was accepted');
163. } catch (e) {
164. if (e.code === 'DUPLICATE_KEY' && e.field === 'score') {
165. console.log(` ✓ PASS — rejected: ${e.message}`);
166. } else {
167. console.log(` ✗ FAIL — wrong error: ${e.message}`);
168. }
169. }
170. await db6.close();
172. // ============================================================
173. // SUMMARY
174. // ============================================================
175. console.log('\n--- Done ---');