Switch to treeForAddonTestSupport

mainmatter · Mar 31, 2017 · 700fc4d · 700fc4d
1 parent 43c72a6
commit 700fc4d
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 5 deletions.
diff --git a/addon/.eslintrc.js → addon-test-support/.eslintrc.js b/addon/.eslintrc.js → addon-test-support/.eslintrc.js
diff --git a/addon/index.js → addon-test-support/index.js b/addon/index.js → addon-test-support/index.js
diff --git a/addon/utils/bind-data-test-attributes.js → ...upport/utils/bind-data-test-attributes.js b/addon/utils/bind-data-test-attributes.js → ...upport/utils/bind-data-test-attributes.js
diff --git a/addon/.gitkeep b/addon/.gitkeep
diff --git a/index.js b/index.js
@@ -3,6 +3,7 @@
 /* eslint-env node */
 
 const VersionChecker = require('ember-cli-version-checker');
+const Funnel = require('broccoli-funnel');
 
 module.exports = {
   name: 'ember-test-selectors',
@@ -79,11 +80,20 @@ module.exports = {
     }
   },
 
-  treeForAddon() {
-    // remove our "addon" folder from the build if we're stripping test selectors
-    if (!this._stripTestSelectors) {
-      return this._super.treeForAddon.apply(this, arguments);
-    }
+  treeForAddonTestSupport(tree) {
+    // intentionally not calling _super here
+    // so that can have our `import`'s be
+    // import testSelector from 'ember-test-selectors';
+
+    let namespacedTree = new Funnel(tree, {
+      srcDir: '/',
+      destDir: `/${this.moduleName()}`,
+      annotation: `Addon#treeForTestSupport (${this.name})`,
+    });
+
+    return this.preprocessJs(namespacedTree, '/', this.name, {
+      registry: this.registry,
+    });
   },
 
   preprocessTree(type, tree) {