Get correct leader election holder ID in upgrade test

nginxinc · May 22, 2024 · 9503bf3 · 9503bf3
1 parent 206f136
commit 9503bf3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/suite/upgrade_test.go b/tests/suite/upgrade_test.go
@@ -208,11 +208,12 @@ var _ = Describe("Upgrade testing", Label("nfr", "upgrade"), func() {
 			500*time.Millisecond,
 			true, /* poll immediately */
 			func(_ context.Context) (bool, error) {
+				defer GinkgoRecover()
 				Expect(k8sClient.Get(leaseCtx, key, &lease)).To(Succeed())
 
 				if lease.Spec.HolderIdentity != nil {
 					for _, podName := range podNames {
-						if podName == *lease.Spec.HolderIdentity {
+						if podName == strings.Split(*lease.Spec.HolderIdentity, "_")[0] {
 							return true, nil
 						}
 					}