mpackdb

10.0 KB

1. import MPackDB, { PrimaryKeyType, IndexType } from '../src/MPackDB.js';
2. import { rm } from 'fs/promises';
4. const TEST_DB_PATH = 'db/cases';
6. console.log('='.repeat(60));
7. console.log('Comprehensive Index Test - All Cases');
8. console.log('='.repeat(60));
10. // Clean up
11. await rm(TEST_DB_PATH, { recursive: true, force: true });
13. // ============================================================================
14. // Case 1: No primary key, no indexes
15. // ============================================================================
16. console.log('\n📋 Case 1: No primary key, no indexes');
17. console.log('-'.repeat(60));
19. const db1 = new MPackDB(`${TEST_DB_PATH}/case1`);
20. await db1.insert({ name: 'Alice', age: 30 });
21. await db1.insert({ name: 'Bob', age: 25 });
22. await db1.insert({ name: 'Charlie', age: 35 });
24. const all1 = await db1.find();
25. console.log(`✓ Inserted 3 records, found ${all1.length}`);
27. // Find with query function (no index)
28. const filtered1 = await db1.find(r => r.age > 28);
29. console.log(`✓ Query function (age > 28): found ${filtered1.length} records`);
30. console.log(` Records: ${filtered1.map(r => r.name).join(', ')}`);
32. await db1.close();
34. // ============================================================================
35. // Case 2: Numeric primary key only (auto-indexed)
36. // ============================================================================
37. console.log('\n📋 Case 2: Numeric primary key only (auto-indexed)');
38. console.log('-'.repeat(60));
40. const db2 = new MPackDB(`${TEST_DB_PATH}/case2`, {
41. primaryKey: 'id',
42. primaryKeyType: PrimaryKeyType.NUMBER,
43. indexPersistThreshold: 2 // Low threshold for testing
44. });
46. await db2.insert({ name: 'Alice', age: 30 });
47. await db2.insert({ name: 'Bob', age: 25 });
48. await db2.insert({ name: 'Charlie', age: 35 });
50. // Wait for auto-persist
51. await new Promise(resolve => setTimeout(resolve, 100));
53. // Find by primary key (uses index)
54. const byId = await db2.find(1);
55. console.log(`✓ Find by ID (1): ${byId[0]?.name}`);
57. // Query function (no index for age)
58. const filtered2 = await db2.find(r => r.age > 28);
59. console.log(`✓ Query function (age > 28): found ${filtered2.length} records`);
61. await db2.close();
63. // ============================================================================
64. // Case 3: UUID primary key only (auto-indexed, lexical)
65. // ============================================================================
66. console.log('\n📋 Case 3: UUID primary key only (auto-indexed, lexical)');
67. console.log('-'.repeat(60));
69. const db3 = new MPackDB(`${TEST_DB_PATH}/case3`, {
70. primaryKey: 'uuid',
71. primaryKeyType: PrimaryKeyType.UUID,
72. indexPersistThreshold: 2
73. });
75. await db3.insert({ uuid: 'aaa-111', name: 'Alice', age: 30 });
76. await db3.insert({ uuid: 'bbb-222', name: 'Bob', age: 25 });
77. await db3.insert({ uuid: 'ccc-333', name: 'Charlie', age: 35 });
79. await new Promise(resolve => setTimeout(resolve, 100));
81. // Find by UUID
82. const byUuid = await db3.find('bbb-222');
83. console.log(`✓ Find by UUID (bbb-222): ${byUuid[0]?.name}`);
85. await db3.close();
87. // ============================================================================
88. // Case 4: String primary key only (auto-indexed, lexical)
89. // ============================================================================
90. console.log('\n📋 Case 4: String primary key only (auto-indexed, lexical)');
91. console.log('-'.repeat(60));
93. const db4 = new MPackDB(`${TEST_DB_PATH}/case4`, {
94. primaryKey: 'username',
95. primaryKeyType: PrimaryKeyType.STRING,
96. indexPersistThreshold: 2
97. });
99. await db4.insert({ username: 'alice123', name: 'Alice', age: 30 });
100. await db4.insert({ username: 'bob456', name: 'Bob', age: 25 });
101. await db4.insert({ username: 'charlie789', name: 'Charlie', age: 35 });
103. await new Promise(resolve => setTimeout(resolve, 100));
105. // Find by username
106. const byUsername = await db4.find('bob456');
107. console.log(`✓ Find by username (bob456): ${byUsername[0]?.name}`);
109. await db4.close();
111. // ============================================================================
112. // Case 5: One lexical index only (no primary key)
113. // ============================================================================
114. console.log('\n📋 Case 5: One lexical index only (no primary key)');
115. console.log('-'.repeat(60));
117. const db5 = new MPackDB(`${TEST_DB_PATH}/case5`, {
118. indexes: ['email'], // Lexical by default
119. indexPersistThreshold: 2
120. });
122. await db5.insert({ name: 'Alice', email: '[email protected]', age: 30 });
123. await db5.insert({ name: 'Bob', email: '[email protected]', age: 25 });
124. await db5.insert({ name: 'Charlie', email: '[email protected]', age: 35 });
126. await new Promise(resolve => setTimeout(resolve, 100));
128. // Find using indexed field
129. const byEmail = await db5.find(r => r.email === '[email protected]');
130. console.log(`✓ Find by email ([email protected]): ${byEmail[0]?.name}`);
132. // Query on non-indexed field
133. const byAge = await db5.find(r => r.age > 28);
134. console.log(`✓ Query on non-indexed field (age > 28): found ${byAge.length} records`);
136. await db5.close();
138. // ============================================================================
139. // Case 6: One numeric index only (no primary key)
140. // ============================================================================
141. console.log('\n📋 Case 6: One numeric index only (no primary key)');
142. console.log('-'.repeat(60));
144. const db6 = new MPackDB(`${TEST_DB_PATH}/case6`, {
145. indexes: ['*score'], // * prefix = numeric index
146. indexPersistThreshold: 2
147. });
149. await db6.insert({ name: 'Alice', score: 95 });
150. await db6.insert({ name: 'Bob', score: 87 });
151. await db6.insert({ name: 'Charlie', score: 92 });
153. await new Promise(resolve => setTimeout(resolve, 100));
155. // Find using numeric indexed field
156. const byScore = await db6.find(r => r.score >= 90);
157. console.log(`✓ Find by score (>= 90): found ${byScore.length} records`);
158. console.log(` Records: ${byScore.map(r => r.name).join(', ')}`);
160. await db6.close();
162. // ============================================================================
163. // Case 7: Multiple indexes (mixed types)
164. // ============================================================================
165. console.log('\n📋 Case 7: Multiple indexes (mixed types)');
166. console.log('-'.repeat(60));
168. const db7 = new MPackDB(`${TEST_DB_PATH}/case7`, {
169. primaryKey: 'id',
170. primaryKeyType: PrimaryKeyType.NUMBER,
171. indexes: ['email', '*age'], // email=lexical, age=numeric
172. indexPersistThreshold: 2
173. });
175. await db7.insert({ name: 'Alice', email: '[email protected]', age: 30 });
176. await db7.insert({ name: 'Bob', email: '[email protected]', age: 25 });
177. await db7.insert({ name: 'Charlie', email: '[email protected]', age: 35 });
179. await new Promise(resolve => setTimeout(resolve, 100));
181. // Find by primary key
182. const byId7 = await db7.find(0);
183. console.log(`✓ Find by ID (0): ${byId7[0]?.name}`);
185. // Find by lexical index
186. const byEmail7 = await db7.find(r => r.email === '[email protected]');
187. console.log(`✓ Find by email: ${byEmail7[0]?.name}`);
189. // Find by numeric index
190. const byAge7 = await db7.find(r => r.age >= 30);
191. console.log(`✓ Find by age (>= 30): found ${byAge7.length} records`);
193. await db7.close();
195. // ============================================================================
196. // Case 8: No primary key, multiple indexes
197. // ============================================================================
198. console.log('\n📋 Case 8: No primary key, multiple indexes');
199. console.log('-'.repeat(60));
201. const db8 = new MPackDB(`${TEST_DB_PATH}/case8`, {
202. indexes: ['category', '*price'], // category=lexical, price=numeric
203. indexPersistThreshold: 2
204. });
206. await db8.insert({ product: 'Laptop', category: 'Electronics', price: 999 });
207. await db8.insert({ product: 'Mouse', category: 'Electronics', price: 25 });
208. await db8.insert({ product: 'Desk', category: 'Furniture', price: 299 });
209. await db8.insert({ product: 'Chair', category: 'Furniture', price: 150 });
211. await new Promise(resolve => setTimeout(resolve, 100));
213. // Find by lexical index
214. const electronics = await db8.find(r => r.category === 'Electronics');
215. console.log(`✓ Find by category (Electronics): found ${electronics.length} records`);
217. // Find by numeric index
218. const expensive = await db8.find(r => r.price > 200);
219. console.log(`✓ Find by price (> 200): found ${expensive.length} records`);
220. console.log(` Products: ${expensive.map(r => r.product).join(', ')}`);
222. // Combined query (no single index can help)
223. const expensiveElectronics = await db8.find(r => r.category === 'Electronics' && r.price > 500);
224. console.log(`✓ Combined query: found ${expensiveElectronics.length} records`);
226. await db8.close();
228. // ============================================================================
229. // Verify all index files were created and persisted
230. // ============================================================================
231. console.log('\n📋 Verify Index Files');
232. console.log('-'.repeat(60));
234. const { readdirSync, statSync } = await import('fs');
236. const cases = [
237. { name: 'Case 1 (no indexes)', prefix: `${TEST_DB_PATH}/case1`, expected: 0 },
238. { name: 'Case 2 (id only)', prefix: `${TEST_DB_PATH}/case2`, expected: 1 },
239. { name: 'Case 3 (_id only)', prefix: `${TEST_DB_PATH}/case3`, expected: 1 },
240. { name: 'Case 4 (username only)', prefix: `${TEST_DB_PATH}/case4`, expected: 1 },
241. { name: 'Case 5 (email only)', prefix: `${TEST_DB_PATH}/case5`, expected: 1 },
242. { name: 'Case 6 (score only)', prefix: `${TEST_DB_PATH}/case6`, expected: 1 },
243. { name: 'Case 7 (id+email+age)', prefix: `${TEST_DB_PATH}/case7`, expected: 3 },
244. { name: 'Case 8 (category+price)', prefix: `${TEST_DB_PATH}/case8`, expected: 2 },
245. ];
247. for (const testCase of cases) {
248. // List all files in cases directory matching the prefix
249. const allFiles = readdirSync(TEST_DB_PATH);
250. const indexFiles = allFiles.filter(f =>
251. f.startsWith(testCase.prefix.replace(`${TEST_DB_PATH}/`, '')) &&
252. f.endsWith('.txt')
253. );
255. const nonEmpty = indexFiles.filter(f => {
256. const size = statSync(`${TEST_DB_PATH}/${f}`).size;
257. return size > 0;
258. });
260. const status = nonEmpty.length === testCase.expected ? '✓' : '✗';
261. console.log(`${status} ${testCase.name}: ${nonEmpty.length}/${testCase.expected} index files with data`);
262. if (nonEmpty.length > 0) {
263. console.log(` Files: ${nonEmpty.join(', ')}`);
264. }
265. }
267. console.log('\n' + '='.repeat(60));
268. console.log('✓ All index cases tested successfully!');
269. console.log('='.repeat(60));