Chore(stress-chaos): Support stress-chaos experiment with custom experiment name (#474)

Signed-off-by: udit <[email protected]>

Author: uditgaurav

chaoslib/litmus/stress-chaos/helper/stress-helper.go

@@ -230,16 +230,16 @@ func prepareStressor(experimentDetails *experimentTypes.ExperimentDetails) []str
 		strconv.Itoa(experimentDetails.ChaosDuration) + "s",
 	}
 
-	switch experimentDetails.ExperimentName {
-	case "pod-cpu-hog":
+	switch experimentDetails.StressType {
+	case "pod-cpu-stress":
 
 		log.InfoWithValues("[Info]: Details of Stressor:", logrus.Fields{
 			"CPU Core": experimentDetails.CPUcores,
 			"Timeout":  experimentDetails.ChaosDuration,
 		})
 		stressArgs = append(stressArgs, "--cpu "+strconv.Itoa(experimentDetails.CPUcores))
 
-	case "pod-memory-hog":
+	case "pod-memory-stress":
 
 		log.InfoWithValues("[Info]: Details of Stressor:", logrus.Fields{
 			"Number of Workers":  experimentDetails.NumberOfWorkers,
@@ -515,6 +515,7 @@ func getENV(experimentDetails *experimentTypes.ExperimentDetails) {
 	experimentDetails.NumberOfWorkers, _ = strconv.Atoi(types.Getenv("NUMBER_OF_WORKERS", ""))
 	experimentDetails.MemoryConsumption, _ = strconv.Atoi(types.Getenv("MEMORY_CONSUMPTION", ""))
 	experimentDetails.VolumeMountPath = types.Getenv("VOLUME_MOUNT_PATH", "")
+	experimentDetails.StressType = types.Getenv("STRESS_TYPE", "")
 }
 
 // abortWatcher continuously watch for the abort signals

chaoslib/litmus/stress-chaos/lib/stress-chaos.go

@@ -296,6 +296,7 @@ func getPodEnv(experimentsDetails *experimentTypes.ExperimentDetails, podName st
 		SetEnv("NUMBER_OF_WORKERS", strconv.Itoa(experimentsDetails.NumberOfWorkers)).
 		SetEnv("MEMORY_CONSUMPTION", strconv.Itoa(experimentsDetails.MemoryConsumption)).
 		SetEnv("VOLUME_MOUNT_PATH", experimentsDetails.VolumeMountPath).
+		SetEnv("STRESS_TYPE", experimentsDetails.StressType).
 		SetEnv("INSTANCE_ID", experimentsDetails.InstanceID).
 		SetEnvFromDownwardAPI("v1", "metadata.name")
 

pkg/generic/stress-chaos/environment/environment.go

@@ -39,16 +39,19 @@ func GetENV(experimentDetails *experimentTypes.ExperimentDetails, expName string
 	switch expName {
 	case "pod-cpu-hog":
 		experimentDetails.CPUcores, _ = strconv.Atoi(types.Getenv("CPU_CORES", "1"))
+		experimentDetails.StressType = "pod-cpu-stress"
 
 	case "pod-memory-hog":
 		experimentDetails.MemoryConsumption, _ = strconv.Atoi(types.Getenv("MEMORY_CONSUMPTION", "500"))
 		experimentDetails.NumberOfWorkers, _ = strconv.Atoi(types.Getenv("NUMBER_OF_WORKERS", "4"))
+		experimentDetails.StressType = "pod-memory-stress"
 
 	case "pod-io-stress":
 		experimentDetails.FilesystemUtilizationPercentage, _ = strconv.Atoi(types.Getenv("FILESYSTEM_UTILIZATION_PERCENTAGE", ""))
 		experimentDetails.FilesystemUtilizationBytes, _ = strconv.Atoi(types.Getenv("FILESYSTEM_UTILIZATION_BYTES", ""))
 		experimentDetails.NumberOfWorkers, _ = strconv.Atoi(types.Getenv("NUMBER_OF_WORKERS", "4"))
 		experimentDetails.VolumeMountPath = types.Getenv("VOLUME_MOUNT_PATH", "")
 		experimentDetails.CPUcores, _ = strconv.Atoi(types.Getenv("CPU_CORES", "0"))
+		experimentDetails.StressType = "pod-io-stress"
 	}
 }

pkg/generic/stress-chaos/types/types.go

@@ -38,4 +38,5 @@ type ExperimentDetails struct {
 	NumberOfWorkers                 int
 	MemoryConsumption               int
 	VolumeMountPath                 string
+	StressType                      string
 }