From a0f19878a3952782aed0cd4ec7346735169cf5ac Mon Sep 17 00:00:00 2001
From: UDIT GAURAV <35391335+uditgaurav@users.noreply.github.com>
Date: Thu, 15 Oct 2020 12:30:37 +0530
Subject: [PATCH] chore(check): Add option for custom kill command in CPU and
 Memory hog (#348)

Signed-off-by: Udit Gaurav <udit.gaurav@mayadata.io>
---
 charts/generic/pod-cpu-hog/experiment.yaml    | 6 ++++++
 charts/generic/pod-memory-hog/experiment.yaml | 3 +++
 2 files changed, 9 insertions(+)

diff --git a/charts/generic/pod-cpu-hog/experiment.yaml b/charts/generic/pod-cpu-hog/experiment.yaml
index 866f842..d1b8a38 100644
--- a/charts/generic/pod-cpu-hog/experiment.yaml
+++ b/charts/generic/pod-cpu-hog/experiment.yaml
@@ -68,6 +68,12 @@ spec:
     - name: TARGET_POD
       value: ''
 
+    - name: CHAOS_INJECT_COMMAND
+      value: 'md5sum /dev/zero'
+
+    - name: CHAOS_KILL_COMMAND
+      value: "kill -9 $(ps afx | grep \"[md5sum] /dev/zero\" | awk '{print$1}' | tr '\n' ' ')"
+
     ## it defines the sequence of chaos execution for multiple target pods
     ## supported values: serial, parallel
     - name: SEQUENCE
diff --git a/charts/generic/pod-memory-hog/experiment.yaml b/charts/generic/pod-memory-hog/experiment.yaml
index 79d4ed4..d63ff69 100644
--- a/charts/generic/pod-memory-hog/experiment.yaml
+++ b/charts/generic/pod-memory-hog/experiment.yaml
@@ -64,6 +64,9 @@ spec:
       ## It is used in pumba lib only    
       - name: LIB_IMAGE
         value: 'gaiaadm/pumba'
+        
+      - name: CHAOS_KILL_COMMAND
+        value: "kill -9 $(ps afx | grep \"[dd] if /dev/zero\" | awk '{print $1}' | tr '\n' ' ')"
 
       ## it defines the sequence of chaos execution for multiple target pods
       ## supported values: serial, parallel