added: db:create / db:drop tasks (sequelize#540)

sushantdhiman · web-flow · commit c035f54dc59c · 2017-09-16T17:19:53.000+05:30
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -0,0 +1,22 @@
+version: '2'
+
+services:
+  # PostgreSQL
+  postgres-95:
+    image: camptocamp/postgis:9.5
+    environment:
+      POSTGRES_USER: postgres
+      POSTGRES_PASSWORD: postgres
+      POSTGRES_DB: sequelize_test
+    ports:
+      - "5432:5432"
+
+  # MySQL
+  mysql-57:
+    image: mysql:5.7
+    environment:
+      MYSQL_ALLOW_EMPTY_PASSWORD: "yes"
+      MYSQL_DATABASE: sequelize_test
+      MYSQL_USER: root
+    ports:
+      - "3306:3306"
diff --git a/src/commands/database.js b/src/commands/database.js
@@ -0,0 +1,94 @@
+import { _baseOptions } from '../core/yargs';
+import { logMigrator } from '../core/migrator';
+
+import helpers from '../helpers';
+import { cloneDeep, defaults } from 'lodash';
+import clc from 'cli-color';
+
+const Sequelize = helpers.generic.getSequelize();
+
+exports.builder = yargs => _baseOptions(yargs).help().argv;
+exports.handler = async function (args) {
+  const command = args._[0];
+
+  // legacy, gulp used to do this
+  await helpers.config.init();
+
+  const sequelize = getDatabaseLessSequelize();
+  const config = helpers.config.readConfig();
+
+  switch (command) {
+    case 'db:create':
+      await sequelize.query(`CREATE DATABASE ${config.database}`, {
+        type: sequelize.QueryTypes.RAW
+      }).catch(e => {
+        helpers.view.error(`Error: ${e.message}`);
+        process.exit(1);
+      });
+
+      helpers.view.log(
+        'Database',
+        clc.blueBright(config.database),
+        'created.'
+      );
+
+      break;
+    case 'db:drop':
+      await sequelize.query(`DROP DATABASE ${config.database}`, {
+        type: sequelize.QueryTypes.RAW
+      }).catch(e => {
+        helpers.view.error(`Error: ${e.message}`);
+        process.exit(1);
+      });
+
+      helpers.view.log(
+        'Database',
+        clc.blueBright(config.database),
+        'dropped.'
+      );
+
+      break;
+  }
+
+  process.exit(0);
+};
+
+function getDatabaseLessSequelize () {
+  let config = null;
+
+  try {
+    config = helpers.config.readConfig();
+  } catch (e) {
+    helpers.view.error(`Error: ${e.message}`);
+    process.exit(1);
+  }
+
+  config = cloneDeep(config);
+  config = defaults(config, { logging: logMigrator });
+
+  switch (config.dialect) {
+    case 'postgres':
+    case 'postgres-native':
+      config.database = 'postgres';
+      break;
+
+    case 'mysql':
+      delete config.database;
+      break;
+
+    case 'mssql':
+      config.database = 'master';
+      break;
+
+    default:
+      helpers.view.error(`Dialect ${config.dialect} does not support db:create / db:drop commands`);
+      process.exit(1);
+  }
+
+  try {
+    return new Sequelize(config);
+  } catch (e) {
+    helpers.view.error(`Error: ${e.message}`);
+    process.exit(1);
+  }
+}
diff --git a/src/core/migrator.js b/src/core/migrator.js
@@ -5,7 +5,7 @@ import _ from 'lodash';
 
 const Sequelize = helpers.generic.getSequelize();
 
-function logMigrator (s) {
+export function logMigrator (s) {
   if (s.indexOf('Executing') !== 0) {
     helpers.view.log(s);
   }
diff --git a/src/sequelize.js b/src/sequelize.js
@@ -28,10 +28,10 @@ import seedOne from './commands/seed_one';
 import migrationGenerate from './commands/migration_generate';
 import modelGenerate from './commands/model_generate';
 import seedGenerate from './commands/seed_generate';
+import database from './commands/database';
 
 import helpers from './helpers/index';
 
-
 helpers.view.teaser();
 
 const cli = yargs
@@ -44,6 +44,8 @@ const cli = yargs
   .command('db:seed:undo', 'Deletes data from the database', seedOne)
   .command('db:seed:all', 'Run every seeder', seed)
   .command('db:seed:undo:all', 'Deletes data from the database', seed)
+  .command('db:create', 'Create database specified by configuration', database)
+  .command('db:drop', 'Drop database specified by configuration', database)
   .command('init', 'Initializes project', init)
   .command('init:config', 'Initializes configuration', init)
   .command('init:migrations', 'Initializes migrations', init)
diff --git a/test/db/db-create.test.js b/test/db/db-create.test.js
@@ -0,0 +1,64 @@
+const expect = require('expect.js');
+const Support = require(__dirname + '/../support');
+const helpers = require(__dirname + '/../support/helpers');
+const gulp = require('gulp');
+const _ = require('lodash');
+
+const prepare = function (flag, callback, options) {
+  options = _.assign({ config: {} }, options || {});
+
+  const configPath = 'config/config.json';
+  const config = _.assign({}, helpers.getTestConfig(), options.config);
+  const configContent = JSON.stringify(config);
+
+  gulp
+    .src(Support.resolveSupportPath('tmp'))
+    .pipe(helpers.clearDirectory())
+    .pipe(helpers.runCli('init'))
+    .pipe(helpers.removeFile(configPath))
+    .pipe(helpers.overwriteFile(configContent, configPath))
+    .pipe(helpers.runCli(flag, { pipeStdout: true }))
+    .pipe(helpers.teardown(callback));
+};
+
+describe(Support.getTestDialectTeaser('db:create'), () => {
+  if (Support.dialectIsPostgres()) {
+    it('correctly creates database', function (done) {
+      const databaseName = `my_test_db_${_.random(10000, 99999)}`;
+      prepare(
+        'db:create',
+        () => {
+          this.sequelize.query(`SELECT 1 as exists FROM pg_database WHERE datname = '${databaseName}';`, {
+            type: this.sequelize.QueryTypes.SELECT
+          }).then(result => {
+            expect(result[0].exists).to.eql(1);
+            done();
+          });
+        }, {
+          config: {
+            database: databaseName
+          }
+        });
+    });
+  }
+
+  if (Support.dialectIsMySQL()) {
+    it('correctly creates database', function (done) {
+      const databaseName = `my_test_db_${_.random(10000, 99999)}`;
+      prepare(
+        'db:create',
+        () => {
+          this.sequelize.query(`SELECT IF('${databaseName}' IN(SELECT SCHEMA_NAME FROM INFORMATION_SCHEMA.SCHEMATA), 1, 0) AS found;`, {
+            type: this.sequelize.QueryTypes.SELECT
+          }).then(result => {
+            expect(result[0].found).to.eql(1);
+            done();
+          });
+        }, {
+          config: {
+            database: databaseName
+          }
+        });
+    });
+  }
+});
diff --git a/test/db/db-drop.test.js b/test/db/db-drop.test.js
@@ -0,0 +1,65 @@
+const expect = require('expect.js');
+const Support = require(__dirname + '/../support');
+const helpers = require(__dirname + '/../support/helpers');
+const gulp = require('gulp');
+const _ = require('lodash');
+
+const prepare = function (flag, callback, options) {
+  options = _.assign({ config: {} }, options || {});
+
+  const configPath = 'config/config.json';
+  const config = _.assign({}, helpers.getTestConfig(), options.config);
+  const configContent = JSON.stringify(config);
+
+  gulp
+    .src(Support.resolveSupportPath('tmp'))
+    .pipe(helpers.clearDirectory())
+    .pipe(helpers.runCli('init'))
+    .pipe(helpers.removeFile(configPath))
+    .pipe(helpers.overwriteFile(configContent, configPath))
+    .pipe(helpers.runCli('db:create'))
+    .pipe(helpers.runCli(flag, { pipeStdout: true }))
+    .pipe(helpers.teardown(callback));
+};
+
+describe(Support.getTestDialectTeaser('db:drop'), () => {
+  if (Support.dialectIsPostgres()) {
+    it('correctly drops database', function (done) {
+      const databaseName = `my_test_db_${_.random(10000, 99999)}`;
+      prepare(
+        'db:drop',
+        () => {
+          this.sequelize.query(`SELECT 1 as exists FROM pg_database WHERE datname = '${databaseName}';`, {
+            type: this.sequelize.QueryTypes.SELECT
+          }).then(result => {
+            expect(result).to.be.empty;
+            done();
+          });
+        }, {
+          config: {
+            database: databaseName
+          }
+        });
+    });
+  }
+
+  if (Support.dialectIsMySQL()) {
+    it('correctly drops database', function (done) {
+      const databaseName = `my_test_db_${_.random(10000, 99999)}`;
+      prepare(
+        'db:drop',
+        () => {
+          this.sequelize.query(`SELECT IF('${databaseName}' IN(SELECT SCHEMA_NAME FROM INFORMATION_SCHEMA.SCHEMATA), 1, 0) AS found;`, {
+            type: this.sequelize.QueryTypes.SELECT
+          }).then(result => {
+            expect(result[0].found).to.eql(0);
+            done();
+          });
+        }, {
+          config: {
+            database: databaseName
+          }
+        });
+    });
+  }
+});
diff --git a/test/support/index.js b/test/support/index.js
@@ -144,7 +144,7 @@ var Support = {
   },
 
   dialectIsMySQL: function (strict) {
-    var envDialect = process.env.DIALECT || 'mysql';
+    var envDialect = this.getTestDialect();
 
     if (strict === undefined) {
       strict = false;
@@ -157,6 +157,11 @@ var Support = {
     }
   },
 
+  dialectIsPostgres: function (strict) {
+    var envDialect = this.getTestDialect();
+    return ['postgres', 'postgres-native'].indexOf(envDialect) !== -1;
+  },
+
   getTestDialectTeaser: function (moduleName) {
     var dialect = this.getTestDialect();
 

Original file line number	Diff line number	Diff line change
`@@ -5,7 +5,7 @@ import _ from 'lodash';`
`5`	`5`
`6`	`6`	`const Sequelize = helpers.generic.getSequelize();`
`7`	`7`
`8`		`-function logMigrator (s) {`
	`8`	`+export function logMigrator (s) {`
`9`	`9`	`if (s.indexOf('Executing') !== 0) {`
`10`	`10`	`helpers.view.log(s);`
`11`	`11`	`}`