use invariant

graphql · Mar 21, 2021 · ace3509 · ace3509
1 parent e272823
commit ace3509
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/src/execution/__tests__/stream-test.js b/src/execution/__tests__/stream-test.js
@@ -136,14 +136,13 @@ async function completeAsync(document, numCalls) {
 
   const result = await execute({ schema, document, rootValue: {} });
 
-  if (isAsyncIterable(result)) {
-    const promises = [];
-    for (let i = 0; i < numCalls; i++) {
-      promises.push(result.next());
-    }
-    return Promise.all(promises);
+  invariant(isAsyncIterable(result));
+
+  const promises = [];
+  for (let i = 0; i < numCalls; i++) {
+    promises.push(result.next());
   }
-  return result;
+  return Promise.all(promises);
 }
 
 describe('Execute: stream directive', () => {