Fixed Label issue for Info (#1363)

checkmarx-ltd · Jul 26, 2024 · 1d9fefa · 1d9fefa
1 parent 1767caa
commit 1d9fefa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/src/main/java/com/checkmarx/flow/custom/GitHubIssueTracker.java b/src/main/java/com/checkmarx/flow/custom/GitHubIssueTracker.java
@@ -26,10 +26,7 @@
 import org.springframework.web.client.HttpStatusCodeException;
 import org.springframework.web.client.RestTemplate;
 import java.time.Duration;
-import java.util.ArrayList;
-import java.util.List;
-import java.util.Map;
-import java.util.Objects;
+import java.util.*;
 
 @Service("GitHub")
 public class GitHubIssueTracker implements IssueTracker {
@@ -337,7 +334,7 @@ private String[] getString(ScanResults.XIssue resultIssue) {
         try {
             Map<FindingSeverity, String> findingsPerSeverity = properties.getIssueslabel();
             for (Map.Entry<FindingSeverity, String> entry : findingsPerSeverity.entrySet()) {
-                if(resultIssue.getSeverity().equalsIgnoreCase(entry.getKey().toString())){
+                if(resultIssue.getSeverity().equalsIgnoreCase(entry.getKey().toString()) || resultIssue.getSeverity().toLowerCase(Locale.ROOT).contains(entry.getKey().toString().toLowerCase(Locale.ROOT))){
 //converting using String.split() method with whitespace as a delimiter
                     strArray = entry.getValue().split(",");
                     break;

diff --git a/src/main/java/com/checkmarx/flow/custom/GitLabIssueTracker.java b/src/main/java/com/checkmarx/flow/custom/GitLabIssueTracker.java
@@ -404,7 +404,7 @@ private String getString(ScanResults.XIssue resultIssue) {
         try {
             Map<FindingSeverity, String> findingsPerSeverity = properties.getIssueslabel();
             for (Map.Entry<FindingSeverity, String> entry : findingsPerSeverity.entrySet()) {
-                if (resultIssue.getSeverity().equalsIgnoreCase(entry.getKey().toString())) {
+                if (resultIssue.getSeverity().equalsIgnoreCase(entry.getKey().toString()) || resultIssue.getSeverity().toLowerCase(Locale.ROOT).contains(entry.getKey().toString().toLowerCase(Locale.ROOT))) {
                     label = entry.getValue();
                     break;
                 }