ff-automationv2 2.1.2 → 2.1.3-beta.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ff-automationv2",
-  "version": "2.1.2",
+  "version": "2.1.3-beta.0",
   "private": false,
   "type": "module",
   "description": "This lib is used to automate the manual testcase",

package/src/core/main/runAutomationScript.ts

@@ -14,12 +14,64 @@ import { DomProcessingEngine } from "../../domAnalysis/getRelaventElements.js"
 import { getAnnotatedDOM } from "../../utils/DomExtraction/jsForAttributeInjection.js"
 import { logger } from "../../utils/logger/logData.js"
 export class AutomationRunner implements IAutomationRunner {
+  static sessionTerminationDetails: Record<string, boolean> = {};
   constructor(
     private readonly request: AutomationRequest,
     private pageLoad: number = 20000,
     private implicit: number = 15000,
     private tokensConsumed: number = 0
-  ) { }
+
+  ) { AutomationRunner.sessionTerminationDetails[request.testCaseId] = false }
+
+  static getSessionTerminationInfo(testCaseId: string): boolean {
+    return AutomationRunner.sessionTerminationDetails[testCaseId]
+  }
+  static updateSessionTerminationInfo(testCaseId: string): void {
+    AutomationRunner.sessionTerminationDetails[testCaseId] = true
+  }
+
+  private async cleanup(
+    context: ExecutionContext,
+    domInfo: any,
+    extractedRelevantDom: any,
+    stepProcessor: StepProcessor,
+    scriptRunner: ScriptRunner
+  ) {
+    try {
+      logger.info("Starting cleanup process...");
+      if (context?.session) {
+        try {
+          const browser = await context.session.getCurrentBrowser();
+          if (browser) {
+            await browser.deleteSession?.();
+          }
+        } catch (e) {
+          logger.error("Browser cleanup failed:", e);
+        }
+      }
+
+      domInfo = null;
+      extractedRelevantDom = null;
+
+      if (stepProcessor) {
+        (stepProcessor as any).llm = null;
+      }
+
+      scriptRunner = null as any;
+
+      delete AutomationRunner.sessionTerminationDetails[this.request.testCaseId];
+
+      if (global.gc) {
+        global.gc();
+      }
+
+      logger.info("Cleanup completed successfully.");
+    }
+    catch (error) {
+      logger.error("Cleanup error:", error);
+    }
+  }
+
 
   async run(): Promise<void> {
     const apiService = new FireFlinkApiService();
@@ -47,6 +99,8 @@ export class AutomationRunner implements IAutomationRunner {
     }
     for (const step of listOfSteps) {
       try {
+        if (AutomationRunner.getSessionTerminationInfo(this.request.testCaseId)) { break; }
+
         const start = Math.max(0, stepCount - 3);
         const end = Math.min(listOfSteps.length, stepCount + 3);
         const priorAndNextSteps = listOfSteps.slice(start, end);
@@ -184,5 +238,14 @@ export class AutomationRunner implements IAutomationRunner {
         "Failed to send payload to FireFlink API:", { cause: error }
       );
     }
+    finally {
+      await this.cleanup(
+        context,
+        domInfo,
+        extractedRelevantDom,
+        stepProcessor,
+        scriptRunner
+      );
+    }
   }
 }