@@ -938,7 +938,7 @@ void shouldAddTraceAndComputeAgentStatusWhenOneAgentFinishes() throws Exception
938
938
Inject injectSaved = injectRepository .findById (inject .getId ()).orElseThrow ();
939
939
InjectStatus injectStatusSaved = injectSaved .getStatus ().orElseThrow ();
940
940
// Check inject status
941
- assertEquals (ExecutionStatus .PENDING , injectStatusSaved .getName ());
941
+ assertEquals (ExecutionStatus .ERROR , injectStatusSaved .getName ());
942
942
assertEquals (2 , injectStatusSaved .getTraces ().size ());
943
943
// The status of the complete trace should be ERROR
944
944
List <ExecutionTrace > completeTraces =
@@ -984,10 +984,8 @@ void shouldAddTraceComputeAgentStatusAndUpdateInjectStatusWhenAllAgentsFinish()
984
984
.injectExecutionInput (input2 )
985
985
.agentId (secondAgentId )
986
986
.injectId (inject .getId ())
987
- .emissionDate (Instant .now ().toEpochMilli ())
987
+ .emissionDate (Instant .now ().toEpochMilli () + 1 )
988
988
.build ();
989
- batchingInjectStatusService .handleInjectExecutionCallbackList (
990
- List .of (injectExecutionCallback1 , injectExecutionCallback2 ));
991
989
992
990
InjectExecutionInput input3 = new InjectExecutionInput ();
993
991
String lastLogMessage = "Complete log received" ;
@@ -999,17 +997,21 @@ void shouldAddTraceComputeAgentStatusAndUpdateInjectStatusWhenAllAgentsFinish()
999
997
.injectExecutionInput (input3 )
1000
998
.agentId (firstAgentId )
1001
999
.injectId (inject .getId ())
1002
- .emissionDate (Instant .now ().toEpochMilli () + 1 )
1000
+ .emissionDate (Instant .now ().toEpochMilli () + 2 )
1003
1001
.build ();
1004
1002
InjectExecutionCallback injectExecutionCallback4 =
1005
1003
InjectExecutionCallback .builder ()
1006
1004
.injectExecutionInput (input3 )
1007
1005
.agentId (secondAgentId )
1008
1006
.injectId (inject .getId ())
1009
- .emissionDate (Instant .now ().toEpochMilli () + 2 )
1007
+ .emissionDate (Instant .now ().toEpochMilli () + 3 )
1010
1008
.build ();
1011
1009
batchingInjectStatusService .handleInjectExecutionCallbackList (
1012
- List .of (injectExecutionCallback3 , injectExecutionCallback4 ));
1010
+ List .of (
1011
+ injectExecutionCallback1 ,
1012
+ injectExecutionCallback2 ,
1013
+ injectExecutionCallback3 ,
1014
+ injectExecutionCallback4 ));
1013
1015
1014
1016
// -- ASSERT --
1015
1017
Inject injectSaved = injectRepository .findById (inject .getId ()).orElseThrow ();
0 commit comments