Update triageGetStates method to return CustomState and remove test ignore (#403)

Co-authored-by: AlvoBen <[email protected]>

Author: cx-ben-alvo

src/main/java/com/checkmarx/ast/wrapper/CxWrapper.java

@@ -166,7 +166,7 @@ public List<Predicate> triageShow(@NonNull UUID projectId, String similarityId,
         return Execution.executeCommand(withConfigArguments(arguments), logger, Predicate::listFromLine, Predicate::validator);
     }
 
-    public List<Predicate> triageGetStates(boolean all) throws IOException, InterruptedException, CxException {
+    public List<CustomState> triageGetStates(boolean all) throws IOException, InterruptedException, CxException {
         this.logger.info("Executing 'triage get-states' command using the CLI.");
 
         List<String> arguments = new ArrayList<>();
@@ -187,7 +187,6 @@ public void triageUpdate(@NonNull UUID projectId, String similarityId, String sc
         this.logger.info("Executing 'triage update' command using the CLI.");
         this.logger.info("Updating the similarityId {} with state {} with customStateId {} and severity {}.", similarityId, state, customStateId, severity);
 
-        boolean emptyState = state == null || state.isEmpty();
         boolean emptyCustomStateId = customStateId == null || customStateId.isEmpty();
 
         List<String> arguments = new ArrayList<>();

src/test/java/com/checkmarx/ast/PredicateTest.java

@@ -1,11 +1,11 @@
 package com.checkmarx.ast;
 
+import com.checkmarx.ast.predicate.CustomState;
 import com.checkmarx.ast.predicate.Predicate;
 import com.checkmarx.ast.results.Results;
 import com.checkmarx.ast.results.result.Result;
 import com.checkmarx.ast.scan.Scan;
 import com.checkmarx.ast.wrapper.CxConstants;
-import org.junit.Ignore;
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Disabled;
 import org.junit.jupiter.api.Test;
@@ -48,9 +48,8 @@ void testTriage() throws Exception {
     }
 
     @Test
-    @Disabled("Ignore this tests until get states api will be in production")
     void testGetStates() throws Exception {
-        List<Predicate> states = wrapper.triageGetStates(false);
+        List<CustomState> states = wrapper.triageGetStates(false);
         Assertions.assertNotNull(states);
     }
 }