tweak test

graphql · Mar 21, 2021 · 3fd57a8 · 3fd57a8
1 parent 9680b06
commit 3fd57a8
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/src/execution/__tests__/stream-test.js b/src/execution/__tests__/stream-test.js
@@ -402,16 +402,14 @@ describe('Execute: stream directive', () => {
 
     const result = await execute({ schema, document, rootValue: {} });
 
-    const results = [];
-    if (isAsyncIterable(result)) {
-      const asyncResults = await Promise.all([
-        result.next(),
-        result.next(),
-        result.next(),
-        result.next(),
-      ]);
-      results.push(...asyncResults);
-    }
+    expect(isAsyncIterable(result)).to.equal(true);
+
+    const results = await Promise.all([
+      result.next(),
+      result.next(),
+      result.next(),
+      result.next(),
+    ]);
 
     expect(results).to.deep.equal([
       {