stratos/utility/db/cleanDb.js

'use strict';

// Import required modules
const path = require('path');

// Import user defined modules
const DatabaseConnectionPool =
	require(path.join(__dirname, '../../lib/DatabaseConnectionPool'));

// Import test data object
const { data } = require('./testData');

/**
 * cleanDb() Removes all records from the tables in the database to be inserted
 * into
 *
 * @return {void}
 */
async function cleanDb() {
	const conn = await new DatabaseConnectionPool();

	/*
	 * Remove records from tables in reverse order to which they depend on
	 * each other
	 */
	const tables = Object.keys(data).reverse();
	tables.push('sessions');

	for (const table of tables) {
		try {
			/* eslint-disable-next-line no-await-in-loop */
			await conn.runQuery(`DELETE FROM ${table};`);
		} catch (e) {
			if (table !== 'sessions')
				throw e;
		}
	}

	conn.close();
}

module.exports = cleanDb;