fix flake seen in pod_shutdown_test.go

This commit is contained in:
Ryan Richard 2023-09-26 11:00:35 -07:00
parent 7513092432
commit 87b7ea14d5

View File

@ -120,7 +120,7 @@ func testShutdownAllPodsOfApp(t *testing.T, env *testlib.TestEnv, namespace stri
// The leader election lease should already contain the empty string as the holder, because the old leader // The leader election lease should already contain the empty string as the holder, because the old leader
// pod should have given up the lease during its graceful shutdown. // pod should have given up the lease during its graceful shutdown.
waitForLeaderElectionLeaseToHaveHolderIdentity(t, namespace, appName, waitForLeaderElectionLeaseToHaveHolderIdentity(t, namespace, appName,
func(holder string) bool { return holder == "" }, 200*time.Millisecond) func(holder string) bool { return holder == "" }, 1*time.Minute)
} }
// Given a list of pods, return a list of their names. // Given a list of pods, return a list of their names.
@ -220,13 +220,14 @@ func waitForLeaderElectionLeaseToHaveHolderIdentity(
waitDuration time.Duration, waitDuration time.Duration,
) { ) {
t.Helper() t.Helper()
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Minute) ctx, cancel := context.WithTimeout(context.Background(), waitDuration*2)
defer cancel() defer cancel()
client := testlib.NewKubernetesClientset(t) client := testlib.NewKubernetesClientset(t)
testlib.RequireEventually(t, func(requireEventually *require.Assertions) { testlib.RequireEventually(t, func(requireEventually *require.Assertions) {
lease, err := client.CoordinationV1().Leases(namespace).Get(ctx, leaseName, metav1.GetOptions{}) lease, err := client.CoordinationV1().Leases(namespace).Get(ctx, leaseName, metav1.GetOptions{})
requireEventually.NoError(err) requireEventually.NoError(err)
requireEventually.True(holderIdentityPredicate(*lease.Spec.HolderIdentity)) requireEventually.Truef(holderIdentityPredicate(*lease.Spec.HolderIdentity),
"leader election lease had holder %s", *lease.Spec.HolderIdentity)
}, waitDuration, 200*time.Millisecond) }, waitDuration, 200*time.Millisecond)
} }