Make image preload timeout configurable

hakuna-matatah · hakuna-matatah · commit 98245913fcfa · 2025-07-23T18:14:17.000Z
diff --git a/clusterloader2/pkg/imagepreload/imagepreload.go b/clusterloader2/pkg/imagepreload/imagepreload.go
@@ -44,17 +44,18 @@ const (
 	namespace       = "preload"
 	daemonsetName   = "preload"
 	pollingInterval = 5 * time.Second
-	pollingTimeout  = 15 * time.Minute
 )
 
 var (
-	images []string
+	images         []string
+	pollingTimeout = 15 * time.Minute
 	//go:embed manifests
 	manifestsFS embed.FS
 )
 
 func InitFlags() {
 	flags.StringSliceEnvVar(&images, "node-preload-images", "NODE_PRELOAD_IMAGES", []string{}, "List of images to preload on each node in the test cluster before executing tests")
+	flags.DurationEnvVar(&pollingTimeout, "node-preload-images-timeout", "NODE_PRELOAD_IMAGES_TIMEOUT", 15*time.Minute, "Timeout for waiting for all nodes to preload images (e.g. 10m, 1h)")
 }
 
 type controller struct {

Original file line number	Diff line number	Diff line change
`@@ -44,17 +44,18 @@ const (`
`44`	`44`	`namespace = "preload"`
`45`	`45`	`daemonsetName = "preload"`
`46`	`46`	`pollingInterval = 5 * time.Second`
`47`		`- pollingTimeout = 15 * time.Minute`
`48`	`47`	`)`
`49`	`48`
`50`	`49`	`var (`
`51`		`- images []string`
	`50`	`+ images []string`
	`51`	`+ pollingTimeout = 15 * time.Minute`
`52`	`52`	`//go:embed manifests`
`53`	`53`	`manifestsFS embed.FS`
`54`	`54`	`)`
`55`	`55`
`56`	`56`	`func InitFlags() {`
`57`	`57`	`flags.StringSliceEnvVar(&images, "node-preload-images", "NODE_PRELOAD_IMAGES", []string{}, "List of images to preload on each node in the test cluster before executing tests")`
	`58`	`+ flags.DurationEnvVar(&pollingTimeout, "node-preload-images-timeout", "NODE_PRELOAD_IMAGES_TIMEOUT", 15*time.Minute, "Timeout for waiting for all nodes to preload images (e.g. 10m, 1h)")`
`58`	`59`	`}`
`59`	`60`
`60`	`61`	`type controller struct {`