retry test fix

Author: csviri

operator-framework/src/main/java/io/javaoperatorsdk/operator/processing/PostExecutionControl.java

@@ -55,4 +55,13 @@ public boolean exceptionDuringExecution() {
   public Optional<RuntimeException> getRuntimeException() {
     return Optional.ofNullable(runtimeException);
   }
+
+  @Override
+  public String toString() {
+    return "PostExecutionControl{" +
+            "onlyFinalizerHandled=" + onlyFinalizerHandled +
+            ", updatedCustomResource=" + updatedCustomResource +
+            ", runtimeException=" + runtimeException +
+            '}';
+  }
 }

operator-framework/src/test/java/io/javaoperatorsdk/operator/processing/DefaultEventHandlerTest.java

@@ -34,7 +34,7 @@ class DefaultEventHandlerTest {
   private static final Logger log = LoggerFactory.getLogger(DefaultEventHandlerTest.class);
 
   public static final int FAKE_CONTROLLER_EXECUTION_DURATION = 250;
-  public static final int SEPARATE_EXECUTION_TIMEOUT = 450;
+  public static final int SEPARATE_EXECUTION_TIMEOUT = 150;
   private EventDispatcher eventDispatcherMock = mock(EventDispatcher.class);
   private CustomResourceCache customResourceCache = new CustomResourceCache();
   private DefaultEventSourceManager defaultEventSourceManagerMock =
@@ -174,21 +174,26 @@ public void successfulExecutionResetsTheRetry() {
         PostExecutionControl.exceptionDuringExecution(new RuntimeException("test"));
     PostExecutionControl defaultDispatchControl = PostExecutionControl.defaultDispatch();
 
-    defaultEventHandlerWithRetry.handleEvent(event);
-    defaultEventHandlerWithRetry.eventProcessingFinished(
-        executionScope, postExecutionControlWithException);
+    when(eventDispatcherMock.handleExecution(any()))
+            .thenReturn(postExecutionControlWithException)
+            .thenReturn(defaultDispatchControl);
+
+    ArgumentCaptor<ExecutionScope> executionScopeArgumentCaptor =
+            ArgumentCaptor.forClass(ExecutionScope.class);
 
     defaultEventHandlerWithRetry.handleEvent(event);
-    defaultEventHandlerWithRetry.eventProcessingFinished(executionScope, defaultDispatchControl);
 
+    verify(eventDispatcherMock, timeout(SEPARATE_EXECUTION_TIMEOUT).times(1))
+            .handleExecution(any());
     defaultEventHandlerWithRetry.handleEvent(event);
 
-    log.info("Finished successfulExecutionResetsTheRetry");
+    verify(eventDispatcherMock, timeout(SEPARATE_EXECUTION_TIMEOUT).times(2))
+            .handleExecution(any());
+    defaultEventHandlerWithRetry.handleEvent(event);
 
-    ArgumentCaptor<ExecutionScope> executionScopeArgumentCaptor =
-        ArgumentCaptor.forClass(ExecutionScope.class);
     verify(eventDispatcherMock, timeout(SEPARATE_EXECUTION_TIMEOUT).times(3))
-        .handleExecution(executionScopeArgumentCaptor.capture());
+            .handleExecution(executionScopeArgumentCaptor.capture());
+    log.info("Finished successfulExecutionResetsTheRetry");
 
     List<ExecutionScope> executionScopes = executionScopeArgumentCaptor.getAllValues();