test : kubectl rollout history for stafefulset

Author: NeverRaR

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

@@ -25,6 +25,7 @@
 import io.kubernetes.client.openapi.ApiClient;
 import io.kubernetes.client.openapi.models.V1DaemonSet;
 import io.kubernetes.client.openapi.models.V1Deployment;
+import io.kubernetes.client.openapi.models.V1StatefulSet;
 import io.kubernetes.client.util.ClientBuilder;
 import java.io.IOException;
 import java.nio.file.Files;
@@ -58,6 +59,15 @@ public class KubectlRolloutHistoryTest {
           .getResource("daemonset-controllerrevision-list.json")
           .getPath();
 
+  private static final String STATEFUL_SET =
+      KubectlRolloutHistoryTest.class.getClassLoader().getResource("statefulset.json").getPath();
+
+  private static final String STATEFUL_SET_CONTROLLER_REVISION_LIST =
+      KubectlRolloutHistoryTest.class
+          .getClassLoader()
+          .getResource("statefulset-controllerrevision-list.json")
+          .getPath();
+
   @Before
   public void setup() throws IOException {
     apiClient = new ClientBuilder().setBasePath("http://localhost:" + wireMockRule.port()).build();
@@ -187,6 +197,73 @@ public void testKubectlRolloutHistoryDaemonSetWithRevisionShouldWork()
     Assert.assertNotNull(rolloutHistory.getTemplate());
   }
 
+  @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))))));
+    KubectlRolloutHistory<V1StatefulSet> rolloutHistory =
+        Kubectl.rolloutHistory(V1StatefulSet.class)
+            .apiClient(apiClient)
+            .name("foo")
+            .namespace("default");
+    rolloutHistory.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, rolloutHistory.getHistories().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))))));
+    KubectlRolloutHistory<V1StatefulSet> rolloutHistory =
+        Kubectl.rolloutHistory(V1StatefulSet.class)
+            .apiClient(apiClient)
+            .name("foo")
+            .namespace("default")
+            .revision(2);
+    rolloutHistory.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.assertNotNull(rolloutHistory.getTemplate());
+  }
+
   @Test
   public void testKubectlRolloutHistoryWithInvalidRevisionShouldThrow() throws IOException {
     wireMockRule.stubFor(