TS: improve types mapAsyncIterator-test (#3038)

graphql · Apr 12, 2021 · 33e3a33 · 33e3a33
1 parent f103924
commit 33e3a33
Showing 1 changed file with 8 additions and 6 deletions.
diff --git 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 });
       },
     };