feature: update observed generation on updateResource

Author: csviri

operator-framework-core/src/main/java/io/javaoperatorsdk/operator/processing/event/ReconciliationDispatcher.java

@@ -157,6 +157,9 @@ private PostExecutionControl<R> reconcileExecution(ExecutionScope<R> executionSc
       updatedCustomResource = updateStatusGenerationAware(updateControl.getResource());
     } else if (updateControl.isUpdateResource()) {
       updatedCustomResource = updateCustomResource(updateControl.getResource());
+      if (shouldUpdateObservedGenerationAutomatically(updatedCustomResource)) {
+        updatedCustomResource = updateStatusGenerationAware(originalResource);
+      }
     } else if (updateControl.isNoUpdate()
         && shouldUpdateObservedGenerationAutomatically(resourceForExecution)) {
       updatedCustomResource = updateStatusGenerationAware(originalResource);

operator-framework-core/src/test/java/io/javaoperatorsdk/operator/processing/event/ReconciliationDispatcherTest.java

@@ -61,7 +61,6 @@ private <R extends HasMetadata> ReconciliationDispatcher<R> init(R customResourc
         .thenReturn(UpdateControl.updateResource(customResource));
     when(reconciler.cleanup(eq(customResource), any()))
         .thenReturn(DeleteControl.defaultDelete());
-    when(customResourceFacade.replaceWithLock(any())).thenReturn(null);
     Controller<R> controller =
         new Controller<>(reconciler, configuration, null);
 
@@ -345,6 +344,27 @@ void updatesObservedGenerationOnNoUpdateUpdateControl() {
         .isEqualTo(1L);
   }
 
+  @Test
+  void updateObservedGenerationOnCustomResourceUpdate() {
+    var observedGenResource = createObservedGenCustomResource();
+
+    Reconciler<ObservedGenCustomResource> reconciler = mock(Reconciler.class);
+    ControllerConfiguration<ObservedGenCustomResource> config =
+            mock(ControllerConfiguration.class);
+    CustomResourceFacade<ObservedGenCustomResource> facade = mock(CustomResourceFacade.class);
+    when(config.isGenerationAware()).thenReturn(true);
+    when(reconciler.reconcile(any(), any()))
+            .thenReturn(UpdateControl.updateResource(observedGenResource));
+    when(facade.replaceWithLock(any())).thenReturn(observedGenResource);
+    when(facade.updateStatus(observedGenResource)).thenReturn(observedGenResource);
+    var dispatcher = init(observedGenResource, reconciler, config, facade);
+
+    PostExecutionControl<ObservedGenCustomResource> control = dispatcher.handleExecution(
+            executionScopeWithCREvent(observedGenResource));
+    assertThat(control.getUpdatedCustomResource().get().getStatus().getObservedGeneration())
+            .isEqualTo(1L);
+  }
+
   @Test
   void callErrorStatusHandlerIfImplemented() {
     testCustomResource.addFinalizer(DEFAULT_FINALIZER);