diff --git a/index.js b/index.js index a406a97f..324f43da 100644 --- a/index.js +++ b/index.js @@ -68,12 +68,18 @@ module.exports = { appOrParent.options = appOrParent.options || {}; - if (checker.satisfies('^6.0.0-beta.1') || checker.satisfies('^7.0.0')) { + if (checker.satisfies('^6.0.0-beta.1')) { appOrParent.options.babel6 = appOrParent.options.babel6 || {}; appOrParent.options.babel6.plugins = appOrParent.options.babel6.plugins || []; appOrParent.options.babel6.plugins.push( require.resolve('./strip-data-test-properties-plugin6') ); + } else if (checker.satisfies('^7.0.0')) { + appOrParent.options.babel = appOrParent.options.babel || {}; + appOrParent.options.babel.plugins = appOrParent.options.babel.plugins || []; + appOrParent.options.babel.plugins.push( + require.resolve('./strip-data-test-properties-plugin6') + ); } else { this.ui.writeWarnLine( 'ember-test-selectors: You are using an unsupported ember-cli-babel version. data-test ' +