@@ -72,6 +72,31 @@ func SetDefaultEnvVarsType(envVarType string) {
7272 }
7373}
7474
75+ func GetRandomEnvVarType (index int ) []corev1.EnvVar {
76+ switch index {
77+ case 1 :
78+ return newEnvVars (defaultEnvVarCount , 100 , "SOME_ENV_VAR_NANO" )
79+ case 2 :
80+ return newEnvVars (defaultEnvVarCount , 200 , "SOME_ENV_VAR_MICRO" )
81+ case 3 :
82+ return newEnvVars (defaultEnvVarCount , 500 , "SOME_ENV_VAR_XSMALL" )
83+ case 4 :
84+ return newEnvVars (defaultEnvVarCount , 1024 , "SOME_ENV_VAR_SMALL" )
85+ case 5 :
86+ return newEnvVars (defaultEnvVarCount , 2 * 1024 , "SOME_ENV_VAR_MEDIUM" )
87+ case 6 :
88+ return newEnvVars (defaultEnvVarCount , 4 * 1024 , "SOME_ENV_VAR_LARGE" )
89+ case 7 :
90+ return newEnvVars (defaultEnvVarCount , 8 * 1024 , "SOME_ENV_VAR_XLARGE" )
91+ case 8 :
92+ return newEnvVars (defaultEnvVarCount , 10 * 1024 , "SOME_ENV_VAR_XLARGE2" )
93+ case 9 :
94+ return newEnvVars (defaultEnvVarCount , 40 * 1024 , "SOME_ENV_VAR_XLARGE8" )
95+ default :
96+ return newEnvVars (defaultEnvVarCount , 2 * 1024 , "SOME_ENV_VAR_MEDIUM" )
97+ }
98+ }
99+
75100// envVarsByType is a slice of different envvar slice types.
76101var envVarsByType = [][]corev1.EnvVar {nano , micro , xsmall , small , medium , large , xlarge , xlarge2 }
77102
@@ -243,7 +268,7 @@ func newPodSpec(randomEnvVars bool) corev1.PodSpec {
243268
244269// getRandomEnvVarType returns a random envvar slice from the envVarsByType.
245270func getRandomEnvVarType () []corev1.EnvVar {
246- return envVarsByType [ rand .Intn (len (envVarsByType ))]
271+ return GetRandomEnvVarType ( rand .Intn (len (envVarsByType )))
247272}
248273
249274// newAffinity creates a new affinity which matches nodes with the type kwok.
0 commit comments