hacker-news-cli/db/db.js
2019-02-10 16:07:09 -08:00

27 lines
634 B
JavaScript
Executable File

const Sequelize = require('sequelize');
const pkg = require('../package.json');
const databaseName =
pkg.name + (process.env.NODE_ENV === 'test' ? '-test' : '');
const createDB = () => {
const db = new Sequelize(
process.env.DATABASE_URL || `postgres://localhost:5432/${databaseName}`,
{
logging: false,
operatorsAliases: false
}
);
return db;
};
const db = createDB();
module.exports = db;
// This is a global Mocha hook used for resource cleanup.
// Otherwise, Mocha v4+ does not exit after tests.
if (process.env.NODE_ENV === 'test') {
after('close database connection', () => db.close());
}