Skip to content

Commit b29f168

Browse files
committed
add random envvars
1 parent 02a31de commit b29f168

File tree

1 file changed

+26
-1
lines changed

1 file changed

+26
-1
lines changed

internal/simulator/resources/resources.go

Lines changed: 26 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -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.
76101
var 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.
245270
func 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

Comments
 (0)