test: add test cases for returning promise (#1036)

Author: JiaLiPassion

test/jasmine-patch.spec.ts

@@ -44,9 +44,28 @@ ifEnvSupports(supportJasmineSpec, () => {
     afterEach(() => {
       let zone = Zone.current;
       expect(zone.name).toEqual('ProxyZone');
-      expect(beforeEachZone).toBe(zone);
+      expect(beforeEachZone.name).toEqual(zone.name);
       expect(itZone).toBe(zone);
     });
+  });
+
+  describe('return promise', () => {
+    let log: string[];
+    beforeEach(() => {
+      log = [];
+    });
 
+    it('should wait for promise to resolve', () => {
+      return new Promise((res, _) => {
+        setTimeout(() => {
+          log.push('resolved');
+          res();
+        }, 100);
+      });
+    });
+
+    afterEach(() => {
+      expect(log).toEqual(['resolved']);
+    });
   });
 })();

test/mocha-patch.spec.ts

@@ -98,4 +98,23 @@ ifEnvSupports('Mocha', function() {
 
   });
 
+  describe('return promise', () => {
+    let log: string[];
+    beforeEach(() => {
+      log = [];
+    });
+
+    it('should wait for promise to resolve', () => {
+      return new Promise((res, _) => {
+        setTimeout(() => {
+          log.push('resolved');
+          res();
+        }, 100);
+      });
+    });
+
+    afterEach(() => {
+      expect(log).toEqual(['resolved']);
+    });
+  });
 })();

test/zone-spec/async-test.spec.ts

@@ -546,6 +546,20 @@ describe('AsyncTestZoneSpec', function() {
         expect(logs).toEqual(['beforeEach', 'timeout']);
       }));
     });
+
+    describe('return promise', () => {
+      let value = 'init';
+      it('should only call finish once', testAsync(() => {
+        return new Promise((resolve, _) => {
+          setTimeout(() => {
+            value = 'timeout';
+            resolve();
+          }, 100);
+        });
+      }, () => {
+        expect(value).toEqual('timeout');
+      }));
+    });
   });
 
   describe('should be able to handle async for both beforeEach and it', () => {

test/zone-spec/fake-async-test.spec.ts

@@ -833,6 +833,26 @@ describe('FakeAsyncTestZoneSpec', () => {
     });
   });
 
+  describe('return promise', () => {
+    let log: string[];
+    beforeEach(() => {
+      log = [];
+    });
+
+    it('should wait for promise to resolve', () => {
+      return new Promise((res, _) => {
+        setTimeout(() => {
+          log.push('resolved');
+          res();
+        }, 100);
+      });
+    });
+
+    afterEach(() => {
+      expect(log).toEqual(['resolved']);
+    });
+  });
+
   describe('fakeAsyncTest should patch Date', () => {
     let FakeAsyncTestZoneSpec = (Zone as any)['FakeAsyncTestZoneSpec'];
     let testZoneSpec: any;