setup-test-factory.js

import Ember from 'ember';
import { before, beforeEach, afterEach, after } from 'mocha';
import { getContext } from 'ember-test-helpers';

export default function(Constructor) {
  return function setupTest(moduleName, options = {}) {
    var module;
    var contextBefore;

    if (Ember.typeOf(moduleName) === 'object') {
      options = moduleName;
      moduleName = '';
    }

    before(function() {
      module = new Constructor(moduleName, options);
    });

    beforeEach(function() {
      contextBefore = {};
      return module.setup().then(() => {
        var context = getContext();
        Object.keys(context).forEach(key => {
          contextBefore[key] = this[key];
          this[key] = context[key];
        });
      });
    });

    afterEach(function() {
      Object.keys(contextBefore).forEach(key => {
        this[key] = contextBefore[key];
      });
      return module.teardown();
    });

    after(function() {
      contextBefore = null;
      module = null;
    });
  };
}