chore: test MinAvailable update for PodDisruptionBudgets

Signed-off-by: 久米拓馬 <takuma.kume@pepabo.com>
IBM · Apr 5, 2023 · 32031a7 · 32031a7
1 parent ef93f50
commit 32031a7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/test/e2e/framework/redis_cluster_handlers.go b/test/e2e/framework/redis_cluster_handlers.go
@@ -477,14 +477,19 @@ func CreateRedisNodeServiceAccountFunc(kubeClient kclient.Client, redisCluster *
 
 // IsPodDisruptionBudgetCreatedFunc returns the func that checks if the PodDisruptionBudget
 // associated with the RedisCluster has been created properly.
-func IsPodDisruptionBudgetCreatedFunc(kubeClient kclient.Client, redisCluster *rapi.RedisCluster) func() error {
+func IsPodDisruptionBudgetCreatedFunc(kubeClient kclient.Client, redisCluster *rapi.RedisCluster, minAvailable int32) func() error {
 	return func() error {
+		pdb := &policyv1.PodDisruptionBudget{}
 		pdbName := types.NamespacedName{Namespace: redisCluster.Namespace, Name: redisCluster.Name}
-		err := kubeClient.Get(context.Background(), pdbName, &policyv1.PodDisruptionBudget{})
+		err := kubeClient.Get(context.Background(), pdbName, pdb)
 		if err != nil {
 			Logf("Cannot get PodDisruptionBudget associated to the redisCluster:%s/%s, err:%v", redisCluster.Namespace, redisCluster.Name, err)
 			return err
 		}
+
+		if minAvailable != pdb.Spec.MinAvailable.IntVal {
+			return LogAndReturnErrorf("PodDisruptionBudget minAvailable mismatch: %v, but got: %v", minAvailable, pdb.Spec.MinAvailable.IntVal)
+		}
 		return nil
 	}
 }
diff --git a/test/e2e/operator.go b/test/e2e/operator.go
@@ -45,7 +45,7 @@ var _ = Describe("RedisCluster CRUD operations", func() {
 
 		Eventually(framework.CreateRedisClusterConfigMapFunc(kubeClient, cluster), "5s", "1s").ShouldNot(HaveOccurred())
 
-		Eventually(framework.IsPodDisruptionBudgetCreatedFunc(kubeClient, cluster), "5s", "1s").ShouldNot(HaveOccurred())
+		Eventually(framework.IsPodDisruptionBudgetCreatedFunc(kubeClient, cluster, int32(5)), "5s", "1s").ShouldNot(HaveOccurred())
 
 		Eventually(framework.IsRedisClusterStartedFunc(kubeClient, cluster), "8m", "5s").ShouldNot(HaveOccurred())
 
@@ -87,6 +87,8 @@ lazyfree-lazy-expire yes`,
 
 			Eventually(framework.IsRedisClusterStartedFunc(kubeClient, cluster), "5m", "5s").ShouldNot(HaveOccurred())
 
+			Eventually(framework.IsPodDisruptionBudgetCreatedFunc(kubeClient, cluster, int32(7)), "5s", "1s").ShouldNot(HaveOccurred())
+
 			Eventually(framework.ZonesBalancedFunc(kubeClient, cluster), "10s", "1s").ShouldNot(HaveOccurred())
 		})
 		Context("a RedisCluster is running", func() {