From 641e6606400068d707b554a0876210ed67e63d5f Mon Sep 17 00:00:00 2001
From: Vinayak Shinde <vinayakshnd@gmail.com>
Date: Thu, 3 Feb 2022 13:17:28 +0530
Subject: [PATCH] :test_tube: Added testcases for ResizeWorkerPool() API.

---
 api/container/containerv2/worker_pool_test.go | 49 +++++++++++++++++++
 1 file changed, 49 insertions(+)

diff --git a/api/container/containerv2/worker_pool_test.go b/api/container/containerv2/worker_pool_test.go
index 42df347c..17c322e0 100644
--- a/api/container/containerv2/worker_pool_test.go
+++ b/api/container/containerv2/worker_pool_test.go
@@ -253,6 +253,55 @@ var _ = Describe("workerpools", func() {
 				Expect(err).To(HaveOccurred())
 			})
 		})
+
+		//Resize
+		Describe("Resize", func() {
+			Context("When resizing workerpool is successful", func() {
+				BeforeEach(func() {
+					server = ghttp.NewServer()
+					server.AppendHandlers(
+						ghttp.CombineHandlers(
+							ghttp.VerifyRequest(http.MethodPost, "/v2/resizeWorkerPool"),
+							ghttp.VerifyJSON(`{"cluster":"bm64u3ed02o93vv36hb0","workerpool":"mywork211","size":5}`),
+						),
+					)
+				})
+				It("should resize Workerpool in a cluster", func() {
+					target := ClusterTargetHeader{}
+					params := ResizeWorkerPoolReq{
+						Cluster:    "bm64u3ed02o93vv36hb0",
+						Workerpool: "mywork211",
+						Size:       5,
+					}
+					err := newWorkerPool(server.URL()).ResizeWorkerPool(params, target)
+					Expect(err).NotTo(HaveOccurred())
+				})
+			})
+			Context("When resizing workerpool is unsuccessful", func() {
+				BeforeEach(func() {
+					server = ghttp.NewServer()
+					server.SetAllowUnhandledRequests(true)
+					server.AppendHandlers(
+						ghttp.CombineHandlers(
+							ghttp.VerifyRequest(http.MethodPost, "/v2/resizeWorkerPool"),
+							ghttp.VerifyJSON(`{"cluster":"bm64u3ed02o93vv36hb0","workerpool":"mywork211","size":5}`),
+							ghttp.RespondWith(http.StatusInternalServerError, `Failed to resize workerpool`),
+						),
+					)
+				})
+
+				It("should return error during resizing workerpool", func() {
+					params := ResizeWorkerPoolReq{
+						Cluster:    "bm64u3ed02o93vv36hb0",
+						Workerpool: "mywork211",
+						Size:       5,
+					}
+					target := ClusterTargetHeader{}
+					err := newWorkerPool(server.URL()).ResizeWorkerPool(params, target)
+					Expect(err).To(HaveOccurred())
+				})
+			})
+		})
 	})
 })