Fix New Style testing helpers usage to support integration tests

README.md

@@ -152,7 +152,7 @@ module('Acceptance | Awesome page', function(hooks) {
   setupApplicationTest(hooks);
 
   test('links go to the new homepage', async function (assert) {
-    enableFeature(this.owner, 'new-homepage');
+    enableFeature('new-homepage');
 
     await visit('/');
     await click('a.home');

addon-test-support/-private/enable-feature.js

@@ -0,0 +1,4 @@
+export function _enableFeature(owner, featureName) {
+  let featuresService = owner.lookup('service:features');
+  featuresService.enable(featureName);
+}

addon-test-support/helpers/with-feature.js

@@ -1,6 +1,6 @@
 import { registerHelper } from '@ember/test';
-import { enableFeature } from '../index';
+import { _enableFeature } from '../-private/enable-feature';
 
 registerHelper('withFeature', function withFeature(app, featureName) {
-  enableFeature(app.__container__, featureName);
+  _enableFeature(app.__deprecatedInstance__, featureName);
 });

addon-test-support/index.js

@@ -1,4 +1,7 @@
-export function enableFeature(owner, featureName) {
-  let featuresService = owner.lookup('service:features');
-  featuresService.enable(featureName);
+import { getContext } from '@ember/test-helpers';
+import { _enableFeature } from './-private/enable-feature';
+
+export function enableFeature(featureName) {
+  let { owner } = getContext();
+  _enableFeature(owner, featureName);
 }

tests/acceptance/feature-flags-new-style-test.js

@@ -7,7 +7,7 @@ module('Acceptance | feature flags with new-style acceptance tests', function(ho
   setupApplicationTest(hooks);
 
   test('visiting / with acceptance-feature on', async function(assert) {
-    enableFeature(this.owner, 'acceptance-feature');
+    enableFeature('acceptance-feature');
 
     await visit('/');