acebase/spec/tempdb.js
2022-03-23 10:57:57 +01:00

38 lines
No EOL
1.4 KiB
JavaScript

const { AceBase, ID } = require("..");
const { pfs } = require('../src/promise-fs');
module.exports = {
async createTempDB(enable = { transactionLogging: false, logLevel: 'log', config: (options) => { } }) {
// Create temp db
const dbname = 'test-' + ID.generate();
const options = { storage: { path: __dirname }, logLevel: enable.logLevel || 'verbose' };
if (enable.transactionLogging === true) {
options.transactions = { log: true };
}
if (typeof enable.config === 'function') {
enable.config(options);
}
const db = new AceBase(dbname, options);
await db.ready();
const removeDB = async () => {
// Close database
await db.close();
// Remove database
const dbdir = `${__dirname}/${dbname}.acebase`;
if (process.versions.node.split('.')[0] < 12) {
// console.error(`Node ${process.version} cannot remove temp database directory ${dbdir}. Remove it manually!`);
const files = await pfs.readdir(dbdir);
await Promise.all(files.map(file => pfs.rm(dbdir + '/' + file)));
await pfs.rmdir(dbdir);
}
else {
await pfs.rmdir(dbdir, { recursive: true });
}
}
return { db, removeDB };
}
}