From 69259435000731a47e541687817119245587ce53 Mon Sep 17 00:00:00 2001
From: tamarleviCm <tamar.levi@checkmarx.com>
Date: Thu, 16 May 2024 12:19:33 +0300
Subject: [PATCH] AST-37667 adding ability to send the agent to results command

---
 src/main/wrapper/CxWrapper.ts | 10 +++++++---
 src/tests/ResultTest.test.ts  | 10 ++++++++++
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/src/main/wrapper/CxWrapper.ts b/src/main/wrapper/CxWrapper.ts
index 4247185b..52bd82ff 100644
--- a/src/main/wrapper/CxWrapper.ts
+++ b/src/main/wrapper/CxWrapper.ts
@@ -205,8 +205,8 @@ export class CxWrapper {
         return exec.executeResultsCommandsFile(scanId, CxConstants.FORMAT_HTML, CxConstants.FORMAT_HTML_FILE, commands, this.config.pathToExecutable, fileName);
     }
 
-    async getResults(scanId: string, resultType: string, outputFileName: string, outputFilePath: string) {
-        const commands = this.resultsShow(scanId, resultType, outputFileName, outputFilePath)
+    async getResults(scanId: string, resultType: string, outputFileName: string, outputFilePath: string, agent?: string | null) {
+        const commands = this.resultsShow(scanId, resultType, outputFileName, outputFilePath, agent)
         const exec = new ExecutionService();
         return await exec.executeCommands(this.config.pathToExecutable, commands);
     }
@@ -218,7 +218,7 @@ export class CxWrapper {
         return await exec.executeCommands(this.config.pathToExecutable, commands, CxConstants.CODE_BASHING_TYPE);
     }
 
-    resultsShow(scanId: string, reportFormat: string, outputFileName: string, outputPath: string): string[] {
+    resultsShow(scanId: string, reportFormat: string, outputFileName: string, outputPath: string, agent?: string | null): string[] {
         const commands: string[] = [CxConstants.CMD_RESULT, CxConstants.SUB_CMD_SHOW, CxConstants.SCAN_ID, scanId, CxConstants.REPORT_FORMAT, reportFormat];
         if (outputFileName) {
             commands.push(CxConstants.OUTPUT_NAME);
@@ -228,6 +228,10 @@ export class CxWrapper {
             commands.push(CxConstants.OUTPUT_PATH);
             commands.push(outputPath);
         }
+        if (agent) {
+            commands.push(CxConstants.AGENT);
+            commands.push(agent);
+        }
         commands.push(...this.initializeCommands(false));
         return commands;
     }
diff --git a/src/tests/ResultTest.test.ts b/src/tests/ResultTest.test.ts
index 98a9af86..05d28a58 100644
--- a/src/tests/ResultTest.test.ts
+++ b/src/tests/ResultTest.test.ts
@@ -15,6 +15,16 @@ describe("Results cases",() => {
         });
     });
 
+    it('Result Test With Agent Flug Successful case', async () => {
+        const auth = new CxWrapper(cxScanConfig);
+        const cxCommandOutput: CxCommandOutput  = await auth.scanList("statuses=Completed");
+        const sampleId  = cxCommandOutput.payload.pop().id;
+        
+        auth.getResults(sampleId,"json","jsonList", ".", "jswrapper").then(() => {
+           fileExists("./jsonList.json").then(file => expect(file).toBe(true));
+        });
+    });
+
     it('Result List Successful case', async () => {
         const auth = new CxWrapper(cxScanConfig);
         const scanList: CxCommandOutput = await auth.scanList("statuses=Completed");