[backend] Fix enum comparison on inject status

OpenBAS-Platform · Nov 14, 2024 · b0bea51 · b0bea51
1 parent 9d322a1
commit b0bea51
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/openbas-model/src/main/java/io/openbas/database/model/ExecutionStatus.java b/openbas-model/src/main/java/io/openbas/database/model/ExecutionStatus.java
@@ -4,7 +4,6 @@ public enum ExecutionStatus {
   SUCCESS,
   ERROR,
   MAYBE_PREVENTED,
-
   DRAFT,
   QUEUING,
   EXECUTING,

diff --git a/openbas-model/src/main/java/io/openbas/database/model/InjectStatus.java b/openbas-model/src/main/java/io/openbas/database/model/InjectStatus.java
@@ -29,12 +29,12 @@ public static InjectStatus fromExecution(Execution execution, Inject executedInj
     int numberOfError =
         (int)
             execution.getTraces().stream()
-                .filter(ex -> ex.getStatus().equals(ExecutionStatus.ERROR))
+                .filter(ex -> ExecutionTraceStatus.ERROR.equals(ex.getStatus()))
                 .count();
     int numberOfSuccess =
         (int)
             execution.getTraces().stream()
-                .filter(ex -> ex.getStatus().equals(ExecutionStatus.SUCCESS))
+                .filter(ex -> ExecutionTraceStatus.SUCCESS.equals(ex.getStatus()))
                 .count();
     injectStatus.setTrackingTotalError(numberOfError);
     injectStatus.setTrackingTotalSuccess(numberOfSuccess);