Replace initializer with IIFE in the vendor tree

README.md

@@ -37,16 +37,6 @@ Installation
 ember install ember-test-selectors
 ```
 
-When you install the addon, it should automatically generate a helper located at
-`tests/helpers/test-selectors.js` as well as add an import to load that helper
-in `tests/test-helper.js`. This sets up the component `data-test-*` auto-binding
-for integration tests.
-
-You can do this manually as well:
-
-```
-$ ember generate ember-test-selectors
-```
 
 Usage
 ------------------------------------------------------------------------------

index.js

@@ -53,19 +53,16 @@ module.exports = {
 
       this._registeredWithBabel = true;
     }
-  },
 
-  treeForAddon: function() {
-    // remove our "addon" folder from the build if we're stripping test selectors
     if (!this._stripTestSelectors) {
-      return this._super.treeForAddon.apply(this, arguments);
+      app.import('vendor/ember-test-selectors/patch-component.js');
     }
   },
 
-  treeForApp: function() {
-    // remove our "app" folder from the build if we're stripping test selectors
+  treeForAddon: function() {
+    // remove our "addon" folder from the build if we're stripping test selectors
     if (!this._stripTestSelectors) {
-      return this._super.treeForApp.apply(this, arguments);
+      return this._super.treeForAddon.apply(this, arguments);
     }
   },
 

vendor/ember-test-selectors/patch-component.js

@@ -0,0 +1,15 @@
+/* global Ember */
+
+(function() {
+  var bindDataTestAttributes;
+
+  Ember.Component.reopen({
+    init: function() {
+      this._super.apply(this, arguments);
+      if (!bindDataTestAttributes) {
+        bindDataTestAttributes = require('ember-test-selectors/utils/bind-data-test-attributes')['default'];
+      }
+      bindDataTestAttributes(this);
+    }
+  });
+})();