Merge pull request #18 from ember-codemods/custom-console

Add console override for tests.

Author: Robert Jackson

test/helpers.js

@@ -0,0 +1,28 @@
+'use strict';
+
+const CONSOLE = Object.assign({}, console);
+
+let consoleOutput;
+
+function setupConsole() {
+  consoleOutput = [];
+
+  console.info = function () {
+    consoleOutput.push(...arguments);
+  };
+}
+
+function resetConsole() {
+  Object.assign(console, CONSOLE);
+  consoleOutput = undefined;
+}
+
+function getConsoleOutput() {
+  return consoleOutput;
+}
+
+module.exports = {
+  setupConsole,
+  getConsoleOutput,
+  resetConsole,
+};

test/lib/migrator/classic-to-flat-test.js

@@ -3,17 +3,20 @@ const fixturify = require('fixturify');
 const fse = require('fs-extra');
 const path = require('path');
 const Migrator = require('../../../lib/migrator');
+const { setupConsole, resetConsole } = require('../../helpers');
 
 assertDiff.options.strict = true;
 
 describe('structure = flat', function () {
   beforeEach(function () {
     this.tmpPath = 'tmp/process-files';
     fse.mkdirsSync(this.tmpPath);
+    setupConsole();
   });
 
   afterEach(function () {
     fse.removeSync(this.tmpPath);
+    resetConsole();
   });
 
   describe('For an app with component classes written in JavaScript', function () {