test : kubectl rollout history for stafefulset

Author: NeverRaR

extended/src/test/java/io/kubernetes/client/extended/kubectl/KubectlRolloutTest.java

@@ -29,7 +29,10 @@
 import io.kubernetes.client.openapi.models.V1DaemonSetList;
 import io.kubernetes.client.openapi.models.V1Deployment;
 import io.kubernetes.client.openapi.models.V1DeploymentList;
+import io.kubernetes.client.openapi.models.V1PodTemplate;
 import io.kubernetes.client.openapi.models.V1PodTemplateSpec;
+import io.kubernetes.client.openapi.models.V1StatefulSet;
+import io.kubernetes.client.openapi.models.V1StatefulSetList;
 import io.kubernetes.client.util.ClientBuilder;
 import io.kubernetes.client.util.ModelMapper;
 import java.io.IOException;
@@ -65,6 +68,18 @@ public class KubectlRolloutTest {
           .getResource("daemonset-controllerrevision-list.json")
           .getPath();
 
+  private static final String STATEFUL_SET =
+      KubectlRolloutTest.class.getClassLoader().getResource("statefulset.json").getPath();
+
+  private static final String PATCHED_STATEFUL_SET =
+      KubectlRolloutTest.class.getClassLoader().getResource("patched-statefulset.json").getPath();
+
+  private static final String STATEFUL_SET_CONTROLLER_REVISION_LIST =
+      KubectlRolloutTest.class
+          .getClassLoader()
+          .getResource("statefulset-controllerrevision-list.json")
+          .getPath();
+
   @Before
   public void setup() throws IOException {
     ModelMapper.addModelMap(
@@ -77,6 +92,14 @@ public void setup() throws IOException {
         V1DeploymentList.class);
     ModelMapper.addModelMap(
         "apps", "v1", "DaemonSet", "daemonsets", true, V1DaemonSet.class, V1DaemonSetList.class);
+    ModelMapper.addModelMap(
+        "apps",
+        "v1",
+        "StatefulSet",
+        "statefulsets",
+        true,
+        V1StatefulSet.class,
+        V1StatefulSetList.class);
     apiClient = new ClientBuilder().setBasePath("http://localhost:" + wireMockRule.port()).build();
   }
 
@@ -223,6 +246,87 @@ public void testKubectlRolloutHistoryDaemonSetWithRevisionShouldWork()
     Assert.assertNotNull(template);
   }
 
+  @Test
+  public void testKubectlRolloutHistoryStatefulSetShouldWork()
+      throws KubectlException, IOException {
+    wireMockRule.stubFor(
+        get(urlPathEqualTo("/apis/apps/v1/namespaces/default/statefulsets/foo"))
+            .willReturn(
+                aResponse()
+                    .withStatus(200)
+                    .withBody(new String(Files.readAllBytes(Paths.get(STATEFUL_SET))))));
+    wireMockRule.stubFor(
+        get(urlPathEqualTo("/apis/apps/v1/namespaces/default/controllerrevisions"))
+            .willReturn(
+                aResponse()
+                    .withStatus(200)
+                    .withBody(
+                        new String(
+                            Files.readAllBytes(
+                                Paths.get(STATEFUL_SET_CONTROLLER_REVISION_LIST))))));
+    List<History> histories =
+        Kubectl.rollout(V1StatefulSet.class)
+            .history()
+            .apiClient(apiClient)
+            .name("foo")
+            .namespace("default")
+            .skipDiscovery()
+            .execute();
+    wireMockRule.verify(
+        1, getRequestedFor((urlPathEqualTo("/apis/apps/v1/namespaces/default/statefulsets/foo"))));
+    wireMockRule.verify(
+        1,
+        getRequestedFor((urlPathEqualTo("/apis/apps/v1/namespaces/default/controllerrevisions")))
+            .withQueryParam("labelSelector", new EqualToPattern("app = bar")));
+    Assert.assertEquals(3, histories.size());
+  }
+
+  @Test
+  public void testKubectlRolloutHistoryStatefulSetWithRevisionShouldWork()
+      throws KubectlException, IOException {
+    wireMockRule.stubFor(
+        get(urlPathEqualTo("/apis/apps/v1/namespaces/default/statefulsets/foo"))
+            .willReturn(
+                aResponse()
+                    .withStatus(200)
+                    .withBody(new String(Files.readAllBytes(Paths.get(STATEFUL_SET))))));
+    wireMockRule.stubFor(
+        get(urlPathEqualTo("/apis/apps/v1/namespaces/default/controllerrevisions"))
+            .willReturn(
+                aResponse()
+                    .withStatus(200)
+                    .withBody(
+                        new String(
+                            Files.readAllBytes(
+                                Paths.get(STATEFUL_SET_CONTROLLER_REVISION_LIST))))));
+    wireMockRule.stubFor(
+        patch(urlPathEqualTo("/apis/apps/v1/namespaces/default/statefulsets/foo"))
+            .willReturn(
+                aResponse()
+                    .withStatus(200)
+                    .withBody(new String(Files.readAllBytes(Paths.get(PATCHED_STATEFUL_SET))))));
+    V1PodTemplateSpec template =
+        Kubectl.rollout(V1StatefulSet.class)
+            .history()
+            .apiClient(apiClient)
+            .name("foo")
+            .namespace("default")
+            .revision(2)
+            .skipDiscovery()
+            .execute();
+    wireMockRule.verify(
+        1, getRequestedFor((urlPathEqualTo("/apis/apps/v1/namespaces/default/statefulsets/foo"))));
+    wireMockRule.verify(
+        1,
+        getRequestedFor((urlPathEqualTo("/apis/apps/v1/namespaces/default/controllerrevisions")))
+            .withQueryParam("labelSelector", new EqualToPattern("app = bar")));
+    wireMockRule.verify(
+        1,
+        patchRequestedFor((urlPathEqualTo("/apis/apps/v1/namespaces/default/statefulsets/foo")))
+            .withQueryParam("dryRun", new EqualToPattern("All")));
+    Assert.assertNotNull(template);
+  }
+
   @Test
   public void testKubectlRolloutHistoryWithInvalidRevisionShouldThrow() throws IOException {
     wireMockRule.stubFor(

extended/src/test/resources/patched-statefulset.json

@@ -0,0 +1,151 @@
+{
+  "apiVersion": "apps/v1",
+  "kind": "StatefulSet",
+  "metadata": {
+    "annotations": {
+      "kubernetes.io/change-cause": "kubectl.exe create --filename\u003dstatefulset.yaml --record\u003dtrue"
+    },
+    "creationTimestamp": "2021-08-13T09:44:24.000000Z",
+    "generation": 4,
+    "managedFields": [{
+      "apiVersion": "apps/v1",
+      "fieldsType": "FieldsV1",
+      "fieldsV1": {
+        "f:metadata": {
+          "f:annotations": {
+            ".": {},
+            "f:kubernetes.io/change-cause": {}
+          }
+        },
+        "f:spec": {
+          "f:podManagementPolicy": {},
+          "f:replicas": {},
+          "f:revisionHistoryLimit": {},
+          "f:selector": {},
+          "f:serviceName": {},
+          "f:template": {
+            "f:metadata": {
+              "f:labels": {
+                ".": {},
+                "f:app": {}
+              }
+            },
+            "f:spec": {
+              "f:containers": {
+                "k:{\"name\":\"busybox-host\"}": {
+                  ".": {},
+                  "f:args": {},
+                  "f:command": {},
+                  "f:image": {},
+                  "f:imagePullPolicy": {},
+                  "f:name": {},
+                  "f:resources": {},
+                  "f:terminationMessagePath": {},
+                  "f:terminationMessagePolicy": {}
+                }
+              },
+              "f:dnsPolicy": {},
+              "f:restartPolicy": {},
+              "f:schedulerName": {},
+              "f:securityContext": {},
+              "f:terminationGracePeriodSeconds": {}
+            }
+          },
+          "f:updateStrategy": {
+            "f:rollingUpdate": {
+              ".": {},
+              "f:partition": {}
+            },
+            "f:type": {}
+          }
+        }
+      },
+      "manager": "kubectl-create",
+      "operation": "Update",
+      "time": "2021-08-13T09:44:24.000000Z"
+    }, {
+      "apiVersion": "apps/v1",
+      "fieldsType": "FieldsV1",
+      "fieldsV1": {
+        "f:spec": {
+          "f:template": {
+            "f:metadata": {
+              "f:annotations": {}
+            }
+          }
+        }
+      },
+      "manager": "kubectl-rollout",
+      "operation": "Update",
+      "time": "2021-08-13T09:47:22.000000Z"
+    }, {
+      "apiVersion": "apps/v1",
+      "fieldsType": "FieldsV1",
+      "fieldsV1": {
+        "f:spec": {
+          "f:template": {
+            "f:metadata": {
+              "f:annotations": {
+                "f:kubectl.kubernetes.io/restartedAt": {}
+              }
+            }
+          }
+        }
+      },
+      "manager": "Kubernetes Java Client",
+      "operation": "Update",
+      "time": "2021-08-17T11:13:42.000000Z"
+    }],
+    "name": "foo",
+    "namespace": "default",
+    "resourceVersion": "103707",
+    "uid": "9a968c95-4b78-4b72-baf9-fa18bab00155"
+  },
+  "spec": {
+    "podManagementPolicy": "OrderedReady",
+    "replicas": 2,
+    "revisionHistoryLimit": 10,
+    "selector": {
+      "matchLabels": {
+        "app": "bar"
+      }
+    },
+    "serviceName": "busybox-service\"",
+    "template": {
+      "metadata": {
+        "annotations": {
+          "kubectl.kubernetes.io/restartedAt": "2021-08-13T17:47:22+08:00"
+        },
+        "labels": {
+          "app": "bar"
+        }
+      },
+      "spec": {
+        "containers": [{
+          "args": ["1000"],
+          "command": ["sleep"],
+          "image": "busybox:1.31.1",
+          "imagePullPolicy": "IfNotPresent",
+          "name": "busybox-host",
+          "resources": {},
+          "terminationMessagePath": "/dev/termination-log",
+          "terminationMessagePolicy": "File"
+        }],
+        "dnsPolicy": "ClusterFirst",
+        "restartPolicy": "Always",
+        "schedulerName": "default-scheduler",
+        "securityContext": {},
+        "terminationGracePeriodSeconds": 30
+      }
+    },
+    "updateStrategy": {
+      "rollingUpdate": {
+        "partition": 0
+      },
+      "type": "RollingUpdate"
+    }
+  },
+  "status": {
+    "replicas": 0
+  }
+}