Extend test framework detection to ember-qunit and ember-mocha

ember-cli · Dec 6, 2017 · ae397a8 · ae397a8
1 parent 9258873
commit ae397a8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -134,9 +134,9 @@ let app = new EmberApp(defaults, {
 });
 ```
 
-- `testGenerator` is automatically detected if `ember-cli-qunit` or
-  `ember-cli-mocha` are used, but can also be set to `qunit` and
-  `mocha` manually.
+- `testGenerator` is automatically detected if `ember-qunit`/`ember-cli-qunit`
+  or `ember-mocha`/`ember-cli-mocha` are used, but can also be set to `qunit`
+  and `mocha` manually.
 
 - `group` can be set to `false` to go back to the previous behavior where
   every generated test was contained in its own separate module.

diff --git a/index.js b/index.js
@@ -20,9 +20,9 @@ module.exports = {
     var VersionChecker = require('ember-cli-version-checker');
     var checker = new VersionChecker(this);
 
-    if (checker.for('ember-cli-qunit', 'npm').exists()) {
+    if (checker.for('ember-qunit', 'npm').exists() || checker.for('ember-cli-qunit', 'npm').exists()) {
       this._testGenerator = 'qunit';
-    } else if (checker.for('ember-cli-mocha', 'npm').exists()) {
+    } else if (checker.for('ember-mocha', 'npm').exists() || checker.for('ember-cli-mocha', 'npm').exists()) {
       this._testGenerator = 'mocha';
     }
   },