Add "flat" module helpers.

lib/ember-qunit.js

@@ -3,6 +3,7 @@ import moduleForComponent   from 'ember-qunit/module-for-component';
 import moduleForModel       from 'ember-qunit/module-for-model';
 import QUnitAdapter         from 'ember-qunit/adapter';
 import setupTestFactory     from 'ember-qunit/setup-test-factory';
+import { buildOptionsWrapperFactory } from 'ember-qunit/qunit-module';
 import {
   setResolver,
   TestModule,
@@ -16,13 +17,21 @@ const setupTest = setupTestFactory(TestModule);
 const setupAcceptanceTest = setupTestFactory(TestModuleForAcceptance);
 const setupIntegrationTest = setupTestFactory(TestModuleForIntegration);
 
+
+const buildUnitOptions = buildOptionsWrapperFactory(TestModule);
+const buildIntegrationOptions = buildOptionsWrapperFactory(TestModuleForIntegration);
+const buildAcceptanceOptions = buildOptionsWrapperFactory(TestModuleForAcceptance);
+
 export {
   moduleFor,
   moduleForComponent,
   moduleForModel,
   setupTest,
   setupAcceptanceTest,
   setupIntegrationTest,
+  buildUnitOptions,
+  buildIntegrationOptions,
+  buildAcceptanceOptions,
   setResolver,
   QUnitAdapter
 };

lib/ember-qunit/qunit-module.js

@@ -17,10 +17,12 @@ function callbackFor(name, callbacks) {
   return callback;
 }
 
-export function createModule(Constructor, name, description, callbacks) {
-  if (!callbacks && typeof description === 'object') {
-    callbacks = description;
-    description = name;
+export function wrapCallbacks(Constructor, _name, _callbacks) {
+  let callbacks = _callbacks;
+  let name = _name;
+  if (!callbacks) {
+    callbacks = name;
+    name = '';
   }
 
   var before = callbackFor('before', callbacks);
@@ -29,34 +31,66 @@ export function createModule(Constructor, name, description, callbacks) {
   var after  = callbackFor('after', callbacks);
 
   var module;
-  var moduleName = typeof description === 'string' ? description : name;
+  var options = {};
+
+  if (callbacks) {
+    for (let key in callbacks) {
+      options[key] = callbacks[key];
+    }
+  }
+
+  options.before = function() {
+    module = new Constructor(name || '', callbacks);
 
-  qunitModule(moduleName, {
-    before() {
-      // storing this in closure scope to avoid exposing these
-      // private internals to the test context
-      module = new Constructor(name, description, callbacks);
+    if (before) {
       return before.apply(this, arguments);
-    },
+    }
+  };
 
-    beforeEach() {
-      // provide the test context to the underlying module
-      module.setContext(this);
+  options.beforeEach = function() {
+    // provide the test context to the underlying module
+    module.setContext(this);
 
-      return module.setup(...arguments).then(() => {
+    return module.setup(...arguments).then(() => {
+      if (beforeEach) {
         return beforeEach.apply(this, arguments);
-      });
-    },
-
-    afterEach() {
-      let result = afterEach.apply(this, arguments);
-      return Ember.RSVP.resolve(result).then(() => module.teardown(...arguments));
-    },
-
-    after() {
-      let result = after.apply(this, arguments);
-      module = null;
-      return result;
+      }
+    });
+  };
+
+  options.afterEach = function() {
+    let result;
+
+    if (afterEach) {
+      result = afterEach.apply(this, arguments);
     }
-  });
+
+    return Ember.RSVP.resolve(result).then(() => module.teardown(...arguments));
+  };
+
+  options.after = function() {
+    let result;
+
+    if (before) {
+      result = after.apply(this, arguments);
+    }
+
+    module = null;
+
+    return result;
+  };
+
+  return options;
+}
+
+export function buildOptionsWrapperFactory(Constructor) {
+  return function optionsWrapperFactory(name, callbacks) {
+    return wrapCallbacks(Constructor, name, callbacks);
+  };
+}
+
+export function createModule(Constructor, name, description, callbacks) {
+  let options = wrapCallbacks(Constructor, name, callbacks || description);
+
+  qunitModule(name, options);
 }

tests/build-integration-options-test.js

@@ -0,0 +1,17 @@
+/* global setTimeout */
+
+import Ember from 'ember';
+import { module, test, buildIntegrationOptions } from 'ember-qunit';
+import setupRegistry from './test-support/setup-registry';
+
+module('component:x-foo', buildIntegrationOptions({
+  before: setupRegistry
+}));
+
+test('renders', function(assert) {
+  assert.expect(1);
+
+  this.render(Ember.Handlebars.compile(`{{pretty-color name="red"}}`));
+
+  assert.equal(this.$('.color-name').text(), 'red');
+});

tests/setup-integration-test-test.js

@@ -2,30 +2,13 @@
 
 import Ember from 'ember';
 import { module, test, setupIntegrationTest } from 'ember-qunit';
-import { setResolverRegistry } from './test-support/resolver';
-
-var PrettyColor = Ember.Component.extend({
-  classNames: ['pretty-color'],
-  attributeBindings: ['style'],
-  style: function(){
-    return 'color: ' + this.get('name') + ';';
-  }.property('name')
-});
-
-function setupRegistry() {
-  setResolverRegistry({
-    'component:x-foo': Ember.Component.extend(),
-    'component:pretty-color': PrettyColor,
-    'template:components/pretty-color': Ember.Handlebars.compile('Pretty Color: <span class="color-name">{{name}}</span>')
-  });
-}
+import setupRegistry from './test-support/setup-registry';
 
 module('component:x-foo', function(hooks) {
-  hooks.before(setupRegistry);
-
   setupIntegrationTest(hooks);
 
   test('renders', function(assert) {
+    setupRegistry();
     assert.expect(1);
 
     this.render(Ember.Handlebars.compile(`{{pretty-color name="red"}}`));

tests/test-support/setup-registry.js

@@ -0,0 +1,18 @@
+import Ember from 'ember';
+import { setResolverRegistry } from './resolver';
+
+var PrettyColor = Ember.Component.extend({
+  classNames: ['pretty-color'],
+  attributeBindings: ['style'],
+  style: function(){
+    return 'color: ' + this.get('name') + ';';
+  }.property('name')
+});
+
+export default function setupRegistry() {
+  setResolverRegistry({
+    'component:x-foo': Ember.Component.extend(),
+    'component:pretty-color': PrettyColor,
+    'template:components/pretty-color': Ember.Handlebars.compile('Pretty Color: <span class="color-name">{{name}}</span>')
+  });
+}