Skip to content

Commit 72f6b08

Browse files
committed
[backend] Batching execution traces
1 parent 31c2857 commit 72f6b08

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

openbas-api/src/test/java/io/openbas/rest/inject/InjectApiTest.java

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -938,7 +938,7 @@ void shouldAddTraceAndComputeAgentStatusWhenOneAgentFinishes() throws Exception
938938
Inject injectSaved = injectRepository.findById(inject.getId()).orElseThrow();
939939
InjectStatus injectStatusSaved = injectSaved.getStatus().orElseThrow();
940940
// Check inject status
941-
assertEquals(ExecutionStatus.PENDING, injectStatusSaved.getName());
941+
assertEquals(ExecutionStatus.ERROR, injectStatusSaved.getName());
942942
assertEquals(2, injectStatusSaved.getTraces().size());
943943
// The status of the complete trace should be ERROR
944944
List<ExecutionTrace> completeTraces =
@@ -984,10 +984,8 @@ void shouldAddTraceComputeAgentStatusAndUpdateInjectStatusWhenAllAgentsFinish()
984984
.injectExecutionInput(input2)
985985
.agentId(secondAgentId)
986986
.injectId(inject.getId())
987-
.emissionDate(Instant.now().toEpochMilli())
987+
.emissionDate(Instant.now().toEpochMilli() + 1)
988988
.build();
989-
batchingInjectStatusService.handleInjectExecutionCallbackList(
990-
List.of(injectExecutionCallback1, injectExecutionCallback2));
991989

992990
InjectExecutionInput input3 = new InjectExecutionInput();
993991
String lastLogMessage = "Complete log received";
@@ -999,17 +997,21 @@ void shouldAddTraceComputeAgentStatusAndUpdateInjectStatusWhenAllAgentsFinish()
999997
.injectExecutionInput(input3)
1000998
.agentId(firstAgentId)
1001999
.injectId(inject.getId())
1002-
.emissionDate(Instant.now().toEpochMilli() + 1)
1000+
.emissionDate(Instant.now().toEpochMilli() + 2)
10031001
.build();
10041002
InjectExecutionCallback injectExecutionCallback4 =
10051003
InjectExecutionCallback.builder()
10061004
.injectExecutionInput(input3)
10071005
.agentId(secondAgentId)
10081006
.injectId(inject.getId())
1009-
.emissionDate(Instant.now().toEpochMilli() + 2)
1007+
.emissionDate(Instant.now().toEpochMilli() + 3)
10101008
.build();
10111009
batchingInjectStatusService.handleInjectExecutionCallbackList(
1012-
List.of(injectExecutionCallback3, injectExecutionCallback4));
1010+
List.of(
1011+
injectExecutionCallback1,
1012+
injectExecutionCallback2,
1013+
injectExecutionCallback3,
1014+
injectExecutionCallback4));
10131015

10141016
// -- ASSERT --
10151017
Inject injectSaved = injectRepository.findById(inject.getId()).orElseThrow();

0 commit comments

Comments
 (0)