From 33e3a33107ba59a99123332b7f227bf36016219c Mon Sep 17 00:00:00 2001
From: Saihajpreet Singh <saihajpreet.singh@gmail.com>
Date: Mon, 12 Apr 2021 18:02:18 -0500
Subject: [PATCH] TS: improve types mapAsyncIterator-test (#3038)

---
 .../__tests__/mapAsyncIterator-test.js             | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/subscription/__tests__/mapAsyncIterator-test.js b/src/subscription/__tests__/mapAsyncIterator-test.js
index 7341595ada..1f7a6b60a5 100644
--- a/src/subscription/__tests__/mapAsyncIterator-test.js
+++ b/src/subscription/__tests__/mapAsyncIterator-test.js
@@ -25,15 +25,17 @@ describe('mapAsyncIterator', () => {
   it('maps over async iterator', async () => {
     const items = [1, 2, 3];
 
-    const iterator: any = {
+    const iterator: $FlowFixMe = {
       [Symbol.asyncIterator]() {
         return this;
       },
-      next() {
-        return Promise.resolve({
-          done: items.length === 0,
-          value: items.shift(),
-        });
+
+      next(): Promise<IteratorResult<number, void>> {
+        if (items.length > 0) {
+          return Promise.resolve({ done: false, value: items.shift() });
+        }
+
+        return Promise.resolve({ done: true, value: undefined });
       },
     };