fix(incrementalDelivery): remove extra then from AsyncRecord completion

extracted from #3786

Author: yaacovCR

src/execution/__tests__/stream-test.ts

@@ -483,11 +483,6 @@ describe('Execute: stream directive', () => {
               },
             ],
           },
-        ],
-        hasNext: true,
-      },
-      {
-        incremental: [
           {
             items: [{ name: 'Leia', id: '3' }],
             path: ['friendList', 2],

src/execution/execute.ts

@@ -1846,16 +1846,21 @@ function executeDeferredFragment(
       fields,
       asyncPayloadRecord,
     );
-
-    if (isPromise(promiseOrData)) {
-      promiseOrData = promiseOrData.then(null, (e) => {
-        asyncPayloadRecord.errors.push(e);
-        return null;
-      });
-    }
   } catch (e) {
     asyncPayloadRecord.errors.push(e);
-    promiseOrData = null;
+    asyncPayloadRecord.addData(null);
+    return;
+  }
+
+  if (isPromise(promiseOrData)) {
+    promiseOrData.then(
+      (value) => asyncPayloadRecord.addData(value),
+      (error) => {
+        asyncPayloadRecord.errors.push(error);
+        asyncPayloadRecord.addData(null);
+      },
+    );
+    return;
   }
   asyncPayloadRecord.addData(promiseOrData);
 }
@@ -1938,21 +1943,19 @@ function executeStreamField(
     return asyncPayloadRecord;
   }
 
-  let completedItems: PromiseOrValue<Array<unknown> | null>;
   if (isPromise(completedItem)) {
-    completedItems = completedItem.then(
-      (value) => [value],
+    completedItem.then(
+      (resolvedItem) => asyncPayloadRecord.addItems([resolvedItem]),
       (error) => {
         asyncPayloadRecord.errors.push(error);
         filterSubsequentPayloads(exeContext, path, asyncPayloadRecord);
-        return null;
+        asyncPayloadRecord.addItems(null);
       },
     );
   } else {
-    completedItems = [completedItem];
+    asyncPayloadRecord.addItems([completedItem]);
   }
 
-  asyncPayloadRecord.addItems(completedItems);
   return asyncPayloadRecord;
 }
 
@@ -2063,22 +2066,19 @@ async function executeStreamIterator(
 
     const { done, value: completedItem } = iteration;
 
-    let completedItems: PromiseOrValue<Array<unknown> | null>;
     if (isPromise(completedItem)) {
-      completedItems = completedItem.then(
-        (value) => [value],
+      completedItem.then(
+        (resolvedItem) => asyncPayloadRecord.addItems([resolvedItem]),
         (error) => {
           asyncPayloadRecord.errors.push(error);
           filterSubsequentPayloads(exeContext, path, asyncPayloadRecord);
-          return null;
+          asyncPayloadRecord.addItems(null);
         },
       );
     } else {
-      completedItems = [completedItem];
+      asyncPayloadRecord.addItems([completedItem]);
     }
 
-    asyncPayloadRecord.addItems(completedItems);
-
     if (done) {
       break;
     }
@@ -2257,7 +2257,7 @@ class DeferredFragmentRecord {
     });
   }
 
-  addData(data: PromiseOrValue<ObjMap<unknown> | null>) {
+  addData(data: ObjMap<unknown> | null) {
     const parentData = this.parentContext?.promise;
     if (parentData) {
       this._resolve?.(parentData.then(() => data));
@@ -2308,7 +2308,7 @@ class StreamRecord {
     });
   }
 
-  addItems(items: PromiseOrValue<Array<unknown> | null>) {
+  addItems(items: Array<unknown> | null) {
     const parentData = this.parentContext?.promise;
     if (parentData) {
       this._resolve?.(parentData.then(() => items));