Fix mock AIDA client to increment callId before yielding final data

Bug: none
Change-Id: I7499d0a1638f07559ab57539d2fa46c7d23613b6
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/6286305
Reviewed-by: Ergün Erdoğmuş <[email protected]>
Auto-Submit: Alex Rudenko <[email protected]>
Commit-Queue: Alex Rudenko <[email protected]>

Author: OrKoN

front_end/testing/AiAssistanceHelpers.ts

@@ -57,14 +57,16 @@ export function mockAidaClient(data: Array<[MockAidaResponse, ...MockAidaRespons
         yield {...chunk, metadata, completed: true};
         break;
       }
+      const completed = idx === data[callId].length - 1;
+      if (completed) {
+        callId++;
+      }
       yield {
         ...chunk,
         metadata,
-        completed: idx === data[callId].length - 1,
+        completed,
       };
     }
-
-    callId++;
   }
 
   return createMockAidaClient(provideAnswer);