fix(testing): async/fakeAsync/inject/withModule helpers should pass through context to callback functions (#13718)

Make sure that context (`this`) that is passed to functions generated by test helpers is passed through to the callback functions. Enables usage of Jasmine's variable sharing system to prevent accidental memory leaks during test runs.

modules/@angular/core/testing/async.ts

@@ -31,14 +31,15 @@ export function async(fn: Function): (done: any) => any {
   // If we're running using the Jasmine test framework, adapt to call the 'done'
   // function when asynchronous activity is finished.
   if (_global.jasmine) {
-    return (done: any) => {
+    // Not using an arrow function to preserve context passed from call site
+    return function(done: any) {
       if (!done) {
         // if we run beforeEach in @angular/core/testing/testing_internal then we get no done
         // fake it here and assume sync.
         done = function() {};
         done.fail = function(e: any) { throw e; };
       }
-      runInTestZone(fn, done, (err: any) => {
+      runInTestZone(fn, this, done, (err: any) => {
         if (typeof err === 'string') {
           return done.fail(new Error(<string>err));
         } else {
@@ -50,12 +51,16 @@ export function async(fn: Function): (done: any) => any {
   // Otherwise, return a promise which will resolve when asynchronous activity
   // is finished. This will be correctly consumed by the Mocha framework with
   // it('...', async(myFn)); or can be used in a custom framework.
-  return () => new Promise<void>((finishCallback, failCallback) => {
-           runInTestZone(fn, finishCallback, failCallback);
-         });
+  // Not using an arrow function to preserve context passed from call site
+  return function() {
+    return new Promise<void>((finishCallback, failCallback) => {
+      runInTestZone(fn, this, finishCallback, failCallback);
+    });
+  };
 }
 
-function runInTestZone(fn: Function, finishCallback: Function, failCallback: Function) {
+function runInTestZone(
+    fn: Function, context: any, finishCallback: Function, failCallback: Function) {
   const currentZone = Zone.current;
   const AsyncTestZoneSpec = (Zone as any)['AsyncTestZoneSpec'];
   if (AsyncTestZoneSpec === undefined) {
@@ -103,5 +108,5 @@ function runInTestZone(fn: Function, finishCallback: Function, failCallback: Fun
         'test');
     proxyZoneSpec.setDelegate(testZoneSpec);
   });
-  return Zone.current.runGuarded(fn);
+  return Zone.current.runGuarded(fn, context);
 }

modules/@angular/core/testing/fake_async.ts

@@ -48,6 +48,7 @@ let _inFakeAsyncCall = false;
  * @experimental
  */
 export function fakeAsync(fn: Function): (...args: any[]) => any {
+  // Not using an arrow function to preserve context passed from call site
   return function(...args: any[]) {
     const proxyZoneSpec = ProxyZoneSpec.assertPresent();
     if (_inFakeAsyncCall) {
@@ -67,7 +68,7 @@ export function fakeAsync(fn: Function): (...args: any[]) => any {
       const lastProxyZoneSpec = proxyZoneSpec.getDelegate();
       proxyZoneSpec.setDelegate(_fakeAsyncTestZoneSpec);
       try {
-        res = fn(...args);
+        res = fn.apply(this, args);
         flushMicrotasks();
       } finally {
         proxyZoneSpec.setDelegate(lastProxyZoneSpec);

modules/@angular/core/testing/test_bed.ts

@@ -325,10 +325,10 @@ export class TestBed implements Injector {
     return result === UNDEFINED ? this._compiler.injector.get(token, notFoundValue) : result;
   }
 
-  execute(tokens: any[], fn: Function): any {
+  execute(tokens: any[], fn: Function, context?: any): any {
     this._initIfNeeded();
     const params = tokens.map(t => this.get(t));
-    return fn(...params);
+    return fn.apply(context, params);
   }
 
   overrideModule(ngModule: Type<any>, override: MetadataOverride<NgModule>): void {
@@ -413,17 +413,19 @@ export function getTestBed() {
 export function inject(tokens: any[], fn: Function): () => any {
   const testBed = getTestBed();
   if (tokens.indexOf(AsyncTestCompleter) >= 0) {
-    return () =>
-               // Return an async test method that returns a Promise if AsyncTestCompleter is one of
-        // the
-        // injected tokens.
-        testBed.compileComponents().then(() => {
-          const completer: AsyncTestCompleter = testBed.get(AsyncTestCompleter);
-          testBed.execute(tokens, fn);
-          return completer.promise;
-        });
+    // Not using an arrow function to preserve context passed from call site
+    return function() {
+      // Return an async test method that returns a Promise if AsyncTestCompleter is one of
+      // the injected tokens.
+      return testBed.compileComponents().then(() => {
+        const completer: AsyncTestCompleter = testBed.get(AsyncTestCompleter);
+        testBed.execute(tokens, fn, this);
+        return completer.promise;
+      });
+    };
   } else {
-    return () => testBed.execute(tokens, fn);
+    // Not using an arrow function to preserve context passed from call site
+    return function() { return testBed.execute(tokens, fn, this); };
   }
 }
 
@@ -441,9 +443,11 @@ export class InjectSetupWrapper {
   }
 
   inject(tokens: any[], fn: Function): () => any {
-    return () => {
-      this._addModule();
-      return inject(tokens, fn)();
+    const self = this;
+    // Not using an arrow function to preserve context passed from call site
+    return function() {
+      self._addModule();
+      return inject(tokens, fn).call(this);
     };
   }
 }
@@ -456,12 +460,13 @@ export function withModule(moduleDef: TestModuleMetadata, fn: Function): () => a
 export function withModule(moduleDef: TestModuleMetadata, fn: Function = null): (() => any)|
     InjectSetupWrapper {
   if (fn) {
-    return () => {
+    // Not using an arrow function to preserve context passed from call site
+    return function() {
       const testBed = getTestBed();
       if (moduleDef) {
         testBed.configureTestingModule(moduleDef);
       }
-      return fn();
+      return fn.apply(this);
     };
   }
   return new InjectSetupWrapper(() => moduleDef);