add fetch results in finalize build for LT Caps

Author: parthlambdatest

src/lib/ctx.ts

@@ -136,6 +136,7 @@ export default (options: Record<string, string>): Context => {
         sessionCapabilitiesMap: new Map<string, any[]>(),
         sessionToBuildMap: new Map<string, string>(),
         buildToSnapshotCountMap: new Map<string, number>(),
-        buildToProjectTokenMap: new Map<string, string>()
+        buildToProjectTokenMap: new Map<string, string>(),
+        fetchResultsForBuild: new Array<string>
     }
 }

src/lib/server.ts

@@ -62,15 +62,6 @@ export default async (ctx: Context): Promise<FastifyInstance<Server, IncomingMes
 							ctx.sessionCapabilitiesMap.set(sessionId, fetchedCapabilitiesResp);
 							ctx.sessionToBuildMap.set(sessionId, capsBuildId);
 							ctx.buildToProjectTokenMap.set(capsBuildId, fetchedCapabilitiesResp?.projectToken || '');
-							if (ctx.options.fetchResults) {
-								let is_baseline;
-								if (fetchedCapabilitiesResp.baseline) {
-									is_baseline = true; 
-								} else {
-									is_baseline = false
-								}
-								startPolling(ctx, capsBuildId, is_baseline, fetchedCapabilitiesResp.projectToken)
-							}
 						}
 						ctx.log.debug(`fetch caps for sessionId: ${sessionId} are ${JSON.stringify(fetchedCapabilitiesResp)}`)
 					} catch (error: any) {

src/lib/snapshotQueue.ts

@@ -308,7 +308,7 @@ export default class Queue {
                         capsBuildId = cachedCapabilities?.buildId || '';
                         useKafkaFlowCaps = cachedCapabilities?.useKafkaFlow || false;
                         if (capsBuildId != '' && capsProjectToken != '') {
-                            useCapsBuildId = true; // Set to true if capsBuildId is not empty
+                            useCapsBuildId = true;
                         }
                     }
 

src/lib/utils.ts

@@ -1,9 +1,9 @@
-import fs from 'fs'
 import { Context } from '../types.js'
 import { chromium, firefox, webkit, Browser } from '@playwright/test'
 import constants from './constants.js';
 import chalk from 'chalk';
 import axios from 'axios';
+import fs from 'fs/promises';
 
 import { globalAgent } from 'http';
 import { promisify } from 'util'
@@ -229,17 +229,10 @@ export function getRenderViewportsForOptions(options: any): Array<Record<string,
 
 // Background polling function
 export async function startPolling(ctx: Context, build_id: string, baseline: boolean, projectToken: string): Promise<void> {
-    let filename = '';
     let isPollingActive = true;
     if (build_id) {
-        filename = `${build_id}.json`
         ctx.log.info(`Fetching results for buildId ${build_id} in progress....`);
     } else if (ctx.build && ctx.build.id) {
-        if (ctx.options.fetchResultsFileName) {
-            filename = ctx.options.fetchResultsFileName
-        } else {
-            filename = `${ctx.build.id}.json`
-        }
         ctx.log.info(`Fetching results for buildId ${ctx.build.id} in progress....`);
     }
 
@@ -262,16 +255,19 @@ export async function startPolling(ctx: Context, build_id: string, baseline: boo
             if (!resp.build) {
                 ctx.log.info("Error: Build data is null.");
                 clearInterval(intervalId);
-                // isPollingActive = false;
+                return;
             }
 
-            fs.writeFileSync(filename, JSON.stringify(resp, null, 2));
-            ctx.log.debug(`Updated results in ${filename}`);
+            let fileName = `${resp.build.build_id}.json`
+            if (ctx.options.fetchResults && ctx.options.fetchResultsFileName && ctx.build && ctx.build.id && resp.build.build_id === ctx.build.id) {
+                fileName = `${ctx.options.fetchResultsFileName}`
+            }
+            await fs.writeFile(`${fileName}`, JSON.stringify(resp, null, 2));
+            ctx.log.debug(`Updated results in ${fileName}`);
 
             if (resp.build.build_status_ind === constants.BUILD_COMPLETE || resp.build.build_status_ind === constants.BUILD_ERROR) {
                 clearInterval(intervalId);
-                ctx.log.info(`Fetching results completed. Final results written to ${filename}`);
-                // isPollingActive = false;
+                ctx.log.info(`Fetching results completed. Final results written to ${fileName}`);
 
 
                 // Evaluating Summary
@@ -309,7 +305,7 @@ export async function startPolling(ctx: Context, build_id: string, baseline: boo
                         `${chalk.yellow('Project Name:')} ${chalk.white(resp.project.name)}\n` +
                         `${chalk.yellow('Build ID:')} ${chalk.white(resp.build.build_id)}\n`
                     )
-                );                            
+                );
             }
         } catch (error: any) {
             if (error.message.includes('ENOTFOUND')) {
@@ -319,7 +315,6 @@ export async function startPolling(ctx: Context, build_id: string, baseline: boo
                 ctx.log.error(`Error fetching screenshot data: ${error.message}`);
             }
             clearInterval(intervalId);
-            // isPollingActive = false;
         }
     }, 5000);
 }

src/tasks/finalizeBuild.ts

@@ -2,6 +2,7 @@ import { ListrTask, ListrRendererFactory } from 'listr2';
 import { Context } from '../types.js'
 import { updateLogContext } from '../lib/logger.js';
 import chalk from 'chalk';
+import { startPolling } from '../lib/utils.js';
 import { unlinkSync } from 'fs';
 import constants from '../lib/constants.js';
 import fs from 'fs';
@@ -30,6 +31,21 @@ export default (ctx: Context): ListrTask<Context, ListrRendererFactory, ListrRen
                     const totalSnapshots = capabilities?.snapshotCount || 0;
                     const sessionBuildUrl = capabilities?.buildURL || '';
                     const testId = capabilities?.id || '';
+
+                    if (ctx.options.fetchResults && ctx.fetchResultsForBuild) {
+                        if (!ctx.fetchResultsForBuild.includes(buildId)) {
+                            let is_baseline;
+                            if (capabilities.baseline) {
+                                is_baseline = true;
+                            } else {
+                                is_baseline = false;
+                            }
+                            console.log(`start polling was called at finalize build for buildId: ${buildId}`)
+                            startPolling(ctx, buildId, is_baseline, capabilities.projectToken);
+                            await new Promise(resolve => setTimeout(resolve, 7000));
+                            ctx.fetchResultsForBuild.push(buildId);
+                        }
+                    }
 
                     if (buildId && projectToken) {
                         await ctx.client.finalizeBuildForCapsWithToken(buildId, totalSnapshots, projectToken, ctx.log);

src/types.ts

@@ -63,6 +63,7 @@ export interface Context {
     buildToSnapshotCountMap?: Map<string, number>;
     sessionToBuildMap?: Map<string, string>;
     buildToProjectTokenMap?: Map<string, string>;
+    fetchResultsForBuild?: Array<string>
 }
 
 export interface Env {