test: Added UTs for ReadyPodsCount and WaitForAllToBeReady methods

dell · Nov 18, 2024 · 5117acb · 5117acb
1 parent 936fcc3
commit 5117acb
Showing 1 changed file with 47 additions and 0 deletions.
diff --git a/pkg/k8sclient/resources/pod/pod_test.go b/pkg/k8sclient/resources/pod/pod_test.go
@@ -219,6 +219,53 @@ func (suite *PodTestSuite) TestDeleteAll() {
 	})
 }
 
+func (suite *PodTestSuite) TestReadyPodsCount() {
+	podconf := &pod.Config{
+		NamePrefix:     "pod-prov-test-",
+		PvcNames:       []string{"pvc1", "pvc2", "pvc3"},
+		VolumeName:     "vol",
+		MountPath:      "/data",
+		ContainerName:  "prov-test",
+		ContainerImage: "quay.io/centos/centos:latest",
+		Command:        []string{"/app/run.sh"},
+	}
+
+	client, err := suite.kubeClient.CreatePodClient("test-namespace")
+	suite.NoError(err)
+
+	podTmpl := client.MakePod(podconf)
+	suite.Equal("test-namespace", podTmpl.Namespace)
+
+	suite.Run("Ready Pods Count test", func() {
+		readyCount, err := client.ReadyPodsCount(context.Background())
+		suite.Equal(readyCount, 0)
+		suite.NoError(err)
+	})
+}
+
+func (suite *PodTestSuite) TestWaitForAllToBeReady() {
+	podconf := &pod.Config{
+		NamePrefix:     "pod-prov-test-",
+		PvcNames:       []string{"pvc1", "pvc2", "pvc3"},
+		VolumeName:     "vol",
+		MountPath:      "/data",
+		ContainerName:  "prov-test",
+		ContainerImage: "quay.io/centos/centos:latest",
+		Command:        []string{"/app/run.sh"},
+	}
+
+	client, err := suite.kubeClient.CreatePodClient("test-namespace")
+	suite.NoError(err)
+
+	podTmpl := client.MakePod(podconf)
+	suite.Equal("test-namespace", podTmpl.Namespace)
+
+	suite.Run("waits for all Pods to be in Ready state test", func() {
+		err := client.WaitForAllToBeReady(context.Background())
+		suite.NoError(err)
+	})
+}
+
 func TestPodTestSuite(t *testing.T) {
 	suite.Run(t, new(PodTestSuite))
 }