Expose schemaTable option from postgrator

command-line-options.js

@@ -39,6 +39,10 @@ const optionDefinitions = [
         name: 'migration-directory', description: "A directory to run migration files from. Default: 'migrations''",
         alias: 'm', type: String, typeLabel: '{underline directory}', defaultValue: 'migrations',
     },
+    {
+        name: 'schema-table', description: "The name of the table where the version of the database will be stored. Default: 'schemaversion'",
+        alias: 'b', type: String, typeLabel: '{underline name}', defaultValue: 'schemaversion',
+    },
     {
         name: 'secure', description: 'Secure connection (Azure). Default: false',
         alias: 's', type: Boolean,

postgrator-cli.js

@@ -70,6 +70,7 @@ function getPostgratorConfigFromCommandLineArgs(commandLineArgs) {
         database: commandLineArgs.database,
         username: commandLineArgs.username,
         password: commandLineArgs.password,
+        schemaTable: commandLineArgs['schema-table'],
         options: { encrypt: commandLineArgs.secure || false },
     };
 }
@@ -137,7 +138,7 @@ function areConflictingMigrations(migrationA, migrationB) {
         && migrationA.filename !== migrationB.filename;
 }
 
-function migrate(postgrator, to, detectVersionConflicts, migrationDirectory) {
+function migrate(postgrator, to, detectVersionConflicts, migrationDirectory, schemaTable) {
     let toVersion;
 
     return postgrator.getMigrations()
@@ -150,7 +151,7 @@ function migrate(postgrator, to, detectVersionConflicts, migrationDirectory) {
         .then((migrateToVersion) => {
             toVersion = migrateToVersion;
             return postgrator.getDatabaseVersion().catch(() => {
-                logMessage('table schemaversion does not exist - creating it.');
+                logMessage(`table ${schemaTable} does not exist - creating it.`);
                 return 0;
             });
         })
@@ -282,7 +283,7 @@ function run(commandLineArgs, callback) {
             (migration) => logMessage(`running ${migration.filename}`)
         );
 
-        const migratePromise = migrate(postgrator, migrateTo, detectVersionConflicts, postgratorConfig.migrationDirectory);
+        const migratePromise = migrate(postgrator, migrateTo, detectVersionConflicts, postgratorConfig.migrationDirectory, postgratorConfig.schemaTable);
 
         promiseToCallback(migratePromise, (err, migrations) => {
             // connection is closed, or will close in the case of SQL Server

test/postgrator-cli-tests.js

@@ -41,8 +41,9 @@ function removeVersionTable(options, callback) {
     console.log(`\n----- ${config.driver} removing tables -----`);
     const Postgrator = require('../node_modules/postgrator/postgrator.js');
     const pg = new Postgrator(config);
+    const versionTable = options['schema-table'];
 
-    promiseToCallback(pg.runQuery('DROP TABLE IF EXISTS schemaversion, animal, person'), (err) => {
+    promiseToCallback(pg.runQuery(`DROP TABLE IF EXISTS ${versionTable}, animal, person`), (err) => {
         assert.ifError(err);
         callback(err);
     });
@@ -465,6 +466,36 @@ function buildTestsForOptions(options) {
             return callback();
         });
     });
+
+    tests.push((callback) => {
+        options['schema-table'] = 'myschemaversion';
+        removeVersionTable(options, (err) => {
+            assert.ifError(err);
+            return callback();
+        });
+    });
+
+    tests.push((callback) => {
+        console.log('\n----- testing migration to 001 with custom version table-----');
+        options['schema-table'] = 'myschemaversion';
+        options.to = 1;
+        postgratorCli.run(options, (err, migrations) => {
+            assert.ifError(err);
+            assert.equal(migrations.length, 1);
+            assert.equal(migrations[0].version, 1);
+            return callback();
+        });
+    });
+
+    tests.push(resetMigrations);
+
+    tests.push((callback) => {
+        options['schema-table'] = 'myschemaversion';
+        removeVersionTable(options, (err) => {
+            assert.ifError(err);
+            return callback();
+        });
+    });
 }
 
 const options = {
@@ -476,6 +507,7 @@ const options = {
     username: 'postgrator',
     password: 'postgrator',
     'migration-directory': 'test/migrations',
+    'schema-table': 'schemaversion',
 };
 
 // Command line parameters