Add deadline to TestCreateDisk: controller context deadline exceeded

kubernetes-sigs · Feb 21, 2024 · e1e567f · e1e567f
1 parent 81be013
commit e1e567f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/cloud/cloud_test.go b/pkg/cloud/cloud_test.go
@@ -46,6 +46,10 @@ const (
 	defaultPath     = "/dev/xvdaa"
 )
 
+const (
+	provisionerSidecarDeadline = time.Second * 5
+)
+
 func generateVolumes(volIdCount, volTagCount int) []*ec2.Volume {
 	volumes := make([]*ec2.Volume, 0, volIdCount+volTagCount)
 
@@ -807,7 +811,8 @@ func TestCreateDisk(t *testing.T) {
 				VolumeId:   aws.String("snap-test-volume"),
 				State:      aws.String("completed"),
 			}
-			ctx := context.Background()
+			ctx, ctxCancel := context.WithDeadline(context.Background(), time.Now().Add(provisionerSidecarDeadline))
+			defer ctxCancel()
 
 			if tc.expCreateVolumeInput != nil {
 				matcher := eqCreateVolume(tc.expCreateVolumeInput)