Add --all flag

Author: tiagobcx

src/main/wrapper/CxConstants.ts

@@ -37,6 +37,7 @@ export enum CxConstants {
     CMD_TRIAGE = "triage",
     SUB_CMD_UPDATE = "update",
     SUB_CMD_GET_STATES = "get-states",
+    ALL_STATES_FLAG = "--all",
     CMD_RESULT = "results",
     SUB_CMD_BFL = "bfl",
     CMD_CODE_BASHING = "codebashing",

src/main/wrapper/CxWrapper.ts

@@ -200,15 +200,16 @@ export class CxWrapper {
         return await exec.executeCommands(this.config.pathToExecutable, commands, CxConstants.PREDICATE_TYPE);
     }
 
-    async triageUpdate(projectId: string, similarityId: string, scanType: string, state: string, comment: string, severity: string, stateId=""): Promise<CxCommandOutput> {
+    async triageUpdate(projectId: string, similarityId: string, scanType: string, state: string, comment: string, severity: string, stateId = ""): Promise<CxCommandOutput> {
         const commands: string[] = [CxConstants.CMD_TRIAGE, CxConstants.SUB_CMD_UPDATE, CxConstants.PROJECT_ID, projectId, CxConstants.SIMILARITY_ID, similarityId, CxConstants.SCAN_TYPES_SUB_CMD, scanType, CxConstants.STATE, state, CxConstants.STATE_ID, stateId, CxConstants.COMMENT, comment, CxConstants.SEVERITY, severity];
         commands.push(...this.initializeCommands(false));
         const exec = new ExecutionService();
         return await exec.executeCommands(this.config.pathToExecutable, commands);
     }
 
-    async triageGetStates(): Promise<CxCommandOutput> {
+    async triageGetStates(all: boolean): Promise<CxCommandOutput> {
         const commands: string[] = [CxConstants.CMD_TRIAGE, CxConstants.SUB_CMD_GET_STATES];
+        if (all) commands.push(CxConstants.ALL_STATES_FLAG)
         commands.push(...this.initializeCommands(false));
         const exec = new ExecutionService();
         return await exec.executeCommands(this.config.pathToExecutable, commands);

src/tests/PredicateTest.test.ts

@@ -31,14 +31,21 @@ describe("Triage cases", () => {
         expect(cxShow.exitCode).toEqual(0);
     }
 
-    const handleTriageUpdate = async (scan: any, result: CxResult, newState: string, newSeverity: string, newStateId="") => {
+    const handleTriageUpdate = async (scan: any, result: CxResult, newState: string, newSeverity: string, newStateId = "") => {
         const cxUpdate: CxCommandOutput = await auth.triageUpdate(
             scan.projectID, result.similarityId, result.type, newState,
             "Edited via JavascriptWrapper",
             newSeverity, newStateId
         );
         expect(cxUpdate.exitCode).toEqual(0);
     };
+    const handlegetStates = async () => {
+        const cxCommandOutput: CxCommandOutput = await auth.triageGetStates(false);
+        console.log("Json object from states successful case: " + JSON.stringify(cxCommandOutput));
+        expect(cxCommandOutput.payload.length).toBeGreaterThanOrEqual(1);
+        expect(cxCommandOutput.exitCode).toBe(0);
+        return cxCommandOutput
+    };
 
     it('Triage Successful case', async () => {
         const { scan, result } = await getScanAndResult();
@@ -49,10 +56,7 @@ describe("Triage cases", () => {
     it.skip('Triage with custom state Successful case', async () => {
         const { scan, result } = await getScanAndResult();
 
-        const cxCommandOutput: CxCommandOutput = await auth.triageGetStates();
-        console.log("Json object from states successful case: " + JSON.stringify(cxCommandOutput));
-        expect(cxCommandOutput.payload.length).toBeGreaterThan(1);
-        expect(cxCommandOutput.exitCode).toBe(0);
+        const cxCommandOutput = await handlegetStates();
 
         let customState = cxCommandOutput.payload[0].name
 
@@ -70,10 +74,8 @@ describe("Triage cases", () => {
     it.skip('Triage with custom state id Successful case', async () => {
         const { scan, result } = await getScanAndResult();
 
-        const cxCommandOutput: CxCommandOutput = await auth.triageGetStates();
-        console.log("Json object from states successful case: " + JSON.stringify(cxCommandOutput));
-        expect(cxCommandOutput.payload.length).toBeGreaterThan(1);
-        expect(cxCommandOutput.exitCode).toBe(0);
+        const cxCommandOutput = await handlegetStates();
+
         const allStates = cxCommandOutput.payload;
         let customStateId = allStates[0].id
         const customStateName = allStates[0].name