DB Helper for testing

node-db-helper.js

'use strict';

const SqlBuilder = require('json-sql-builder2');
const fs = require('fs');
const path = require('path');

module.exports = class DbHelper {
    static async init(connection, name) {
        await this.execute(connection, name, 'truncate');
        await this.execute(connection, name, 'seed');
    }

    static async execute(connection, name, command) {
        const dir = path.resolve(__dirname, `../db/${name}/${command}`);
        if (fs.existsSync(dir)) {
            const files = fs.readdirSync(dir).map(_ => path.resolve(dir, _)).filter(_ => fs.statSync(_).isFile() && _.match(/\.sql$/i));
            const sqls = files.map(_ => fs.readFileSync(_, { encoding: 'utf8' }));
            const results = await Promise.all(sqls.map(async _ => await connection.query(_)));
            return results;
        }
    }

    static async insert(connection, $table, source, action) {
        const builder = new SqlBuilder('MySQL');
        const sql = source.map($documents => builder.$insert({ $table, $documents })).map(_ => `${connection.format(_.sql, _.values)};`).join('\n');
        const results = await connection.query(sql);
        if (action) {
            action(results);
        }
        return results;
    }

    static async select(connection, $from, options) {
        options = Object.assign({}, options);
        const $columns = options.columns ? options.columns : '*';
        const builder = new SqlBuilder('MySQL');
        const sql = [1].map(() => builder.$select({ $columns, $from })).map(_ => `${connection.format(_.sql, _.values)};`).join('\n');
        const results = await connection.query(sql);
        return results;
    }
};