Merge pull request #578 from oracle/develop-feature-rolling-restart-tests

Tests for rolling restart

Author: rjeberhard

operator/src/test/java/oracle/kubernetes/operator/helpers/AdminPodHelperTest.java

@@ -417,6 +417,18 @@ public void whenPodHasCustomLabelConflictWithInternal_createAdminPodWithInternal
     assertThat(podLabels, hasEntry("label1", "server-label-value1"));
   }
 
+  @Test
+  public void whenDomainAndAdminHasRestartVersion_createAdminPodWithRestartVersionLabel() {
+    getConfigurator()
+        .withRestartVersion("domainRestartV1")
+        .configureAdminServer((ADMIN_SERVER))
+          .withRestartVersion("adminRestartV1");
+
+    Map<String, String> podLabels = getCreatedPod().getMetadata().getLabels();
+    assertThat(podLabels, hasEntry(LabelConstants.DOMAINRESTARTVERSION_LABEL, "domainRestartV1"));
+    assertThat(podLabels, hasEntry(LabelConstants.SERVERRESTARTVERSION_LABEL, "adminRestartV1"));
+  }
+
   @Override
   protected void onAdminExpectListPersistentVolume() {
     expectListPersistentVolume().returning(createPersistentVolumeList());

operator/src/test/java/oracle/kubernetes/operator/helpers/ManagedPodHelperTest.java

@@ -209,6 +209,14 @@ public void whenExistingManagedPodSpecContainerHasWrongEnvFrom_replaceIt() {
     verifyRollManagedPodWhen((pod) -> getSpecContainer(pod).envFrom(Collections.emptyList()));
   }
 
+  @Test
+  public void whenExistingManagedPodRestartVersionChange() {
+    verifyRollManagedPodWhen(
+        (pod) ->
+            pod.getMetadata()
+                .putLabelsItem(LabelConstants.SERVERRESTARTVERSION_LABEL, "serverRestartV1"));
+  }
+
   @Test
   public void whenDomainHasAdditionalVolumes_createManagedPodWithThem() {
     getConfigurator()
@@ -362,6 +370,24 @@ public void whenClusterHasLabels_createManagedPodWithThem() {
     assertThat(podLabels, hasEntry("label2", "cluster-label-value2"));
   }
 
+  @Test
+  public void whenClusterHasRestartVersion_createManagedPodWithRestartVersionLabel() {
+    testSupport.addToPacket(ProcessingConstants.CLUSTER_NAME, CLUSTER_NAME);
+    getConfigurator().configureCluster(CLUSTER_NAME).withRestartVersion("clusterRestartV1");
+
+    Map<String, String> podLabels = getCreatedPod().getMetadata().getLabels();
+    assertThat(podLabels, hasEntry(LabelConstants.CLUSTERRESTARTVERSION_LABEL, "clusterRestartV1"));
+  }
+
+  @Test
+  public void whenDomainHasRestartVersion_createManagedPodWithRestartVersionLabel() {
+    testSupport.addToPacket(ProcessingConstants.CLUSTER_NAME, CLUSTER_NAME);
+    getConfigurator().withRestartVersion("domainRestartV1");
+
+    Map<String, String> podLabels = getCreatedPod().getMetadata().getLabels();
+    assertThat(podLabels, hasEntry(LabelConstants.DOMAINRESTARTVERSION_LABEL, "domainRestartV1"));
+  }
+
   @Test
   public void whenClusterHasAnnotations_createManagedPodWithThem() {
     testSupport.addToPacket(ProcessingConstants.CLUSTER_NAME, CLUSTER_NAME);