From 62d347d7f3b8c10a8b83a906abc81e67543df9bb Mon Sep 17 00:00:00 2001
From: Yaacov Rydzinski <yaacovCR@gmail.com>
Date: Mon, 3 Jun 2024 17:26:08 +0300
Subject: [PATCH] incremental(stream): revert test logic (#4101)

Test logic was inadvertently altered in #4026.

Tests still pass, just fixing logic.
---
 src/execution/__tests__/stream-test.ts | 34 +++++++++++++-------------
 1 file changed, 17 insertions(+), 17 deletions(-)

diff --git a/src/execution/__tests__/stream-test.ts b/src/execution/__tests__/stream-test.ts
index 905b00be9e..f0a103b935 100644
--- a/src/execution/__tests__/stream-test.ts
+++ b/src/execution/__tests__/stream-test.ts
@@ -1939,7 +1939,9 @@ describe('Execute: stream directive', () => {
       hasNext: true,
     });
 
-    const result2 = await iterator.next();
+    const result2Promise = iterator.next();
+    resolveIterableCompletion(null);
+    const result2 = await result2Promise;
     expectJSON(result2).toDeepEqual({
       value: {
         pending: [{ id: '2', path: ['friendList', 1], label: 'DeferName' }],
@@ -1960,7 +1962,7 @@ describe('Execute: stream directive', () => {
     });
 
     const result3Promise = iterator.next();
-    resolveIterableCompletion(null);
+    resolveSlowField('Han');
     const result3 = await result3Promise;
     expectJSON(result3).toDeepEqual({
       value: {
@@ -1969,9 +1971,7 @@ describe('Execute: stream directive', () => {
       },
       done: false,
     });
-    const result4Promise = iterator.next();
-    resolveSlowField('Han');
-    const result4 = await result4Promise;
+    const result4 = await iterator.next();
     expectJSON(result4).toDeepEqual({
       value: {
         incremental: [
@@ -2062,19 +2062,8 @@ describe('Execute: stream directive', () => {
       done: false,
     });
 
-    const result3Promise = iterator.next();
-    resolveIterableCompletion(null);
-    const result3 = await result3Promise;
+    const result3 = await iterator.next();
     expectJSON(result3).toDeepEqual({
-      value: {
-        completed: [{ id: '1' }],
-        hasNext: true,
-      },
-      done: false,
-    });
-
-    const result4 = await iterator.next();
-    expectJSON(result4).toDeepEqual({
       value: {
         incremental: [
           {
@@ -2083,6 +2072,17 @@ describe('Execute: stream directive', () => {
           },
         ],
         completed: [{ id: '2' }],
+        hasNext: true,
+      },
+      done: false,
+    });
+
+    const result4Promise = iterator.next();
+    resolveIterableCompletion(null);
+    const result4 = await result4Promise;
+    expectJSON(result4).toDeepEqual({
+      value: {
+        completed: [{ id: '1' }],
         hasNext: false,
       },
       done: false,