From b44fc7811abf39ab9cc7425a1df63c82da12e83e Mon Sep 17 00:00:00 2001
From: Karan Kumar <karan.kumar6@ibm.com>
Date: Thu, 7 Mar 2024 11:12:06 +0530
Subject: [PATCH] fixup! Add e2e tests to verify vulnerability scanning

---
 test/e2e/v1beta1/common_cbs_test.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/e2e/v1beta1/common_cbs_test.go b/test/e2e/v1beta1/common_cbs_test.go
index c65b25224e..81f0ae9280 100644
--- a/test/e2e/v1beta1/common_cbs_test.go
+++ b/test/e2e/v1beta1/common_cbs_test.go
@@ -58,7 +58,7 @@ func (c *clusterBuildStrategyPrototype) Create() (cbs *buildapi.ClusterBuildStra
 		return nil, err
 	}
 
-	err = wait.PollImmediate(pollCreateInterval, pollCreateTimeout, func() (done bool, err error) {
+	err = wait.PollUntilContextTimeout(ctx, pollCreateInterval, pollCreateTimeout, true, func(context.Context) (done bool, err error) {
 		cbs, err = testBuild.BuildClientSet.ShipwrightV1beta1().ClusterBuildStrategies().Get(ctx, c.clusterBuildStrategy.Name, meta.GetOptions{})
 		if err != nil {
 			return false, err