test: add test cases for returning promise

Author: JiaLiPassion

test/jasmine-patch.spec.ts

@@ -48,5 +48,24 @@ ifEnvSupports(supportJasmineSpec, () => {
       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,5 +546,19 @@ 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');
+      }));
+    });
   });
 });

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;