chore(test): add test for template hook

Signed-off-by: Tobias Gurtzick <[email protected]>

Author: wzrdtales

lib/template.js

@@ -4,8 +4,11 @@ const File = require('./file.js');
 const { util: dbmUtil, log } = require('db-migrate-shared');
 
 const Template = function () {
-  this.plugins = arguments[4];
-  delete arguments[4];
+  if (arguments[4] && typeof arguments[4].overwrite === 'function') {
+    this.plugins = arguments[4];
+    arguments[4] = undefined;
+    arguments.length--;
+  }
   this.templateType = arguments[3];
   // this is a tweak for node 4 this can be translated
   // to ...arguments instead in upcoming node versions

test/migration_test.js

@@ -234,17 +234,32 @@ function getTemplate () {
   lab.experiment('when template plugin is set', function () {
     lab.experiment('as sql file loader', function () {
       const name = 'test';
-      const plugins = Plugins.createSinglePlugin(
-        `template:overwrite:provider:${name}`,
-        opts => {
-          return `test all variables`;
-        }
-      );
-      const migration = new Template(fileName, dirName, date, name, plugins);
 
       lab.test('should return sql file loader template', () => {
-        const actual = migration.getTemplate();
-        Code.expect(actual).to.equal(`test all variables`);
+        const plugins = Plugins.createSinglePlugin(
+          `template:overwrite:provider:${name}`,
+          opts => {
+            return `test all variables`;
+          }
+        );
+        const migration = new Template(fileName, dirName, date, name, plugins);
+
+        Code.expect(migration.getTemplate()).to.equal(`test all variables`);
+      });
+
+      lab.test('should throw if plugin fails', () => {
+        const plugins = Plugins.createSinglePlugin(
+          `template:overwrite:provider:${name}`,
+          opts => {
+            throw new Error('test');
+          }
+        );
+        const migration = new Template(fileName, dirName, date, name, plugins);
+
+        Code.expect(migration.getTemplate.bind(migration)).to.throw(
+          Error,
+          'test'
+        );
       });
     });
 
@@ -263,6 +278,23 @@ function getTemplate () {
       });
     });
 
+    lab.experiment('as sql default ignore on init template', function () {
+      const migration = new Template(
+        fileName,
+        dirName,
+        date,
+        Template.TemplateType.SQL_FILE_LOADER_IGNORE_ON_INIT,
+        internals
+      );
+
+      lab.test('should return sql ignore on init template', () => {
+        const actual = migration.getTemplate();
+        Code.expect(actual).to.equal(
+          migration.sqlFileLoaderIgnoreOnInitTemplate()
+        );
+      });
+    });
+
     lab.experiment('as default coffee', function () {
       const migration = new Template(
         fileName,