chore(push): Update chaos-ci-lib with latest litmus-1.11.1 (#5)

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

Author: uditgaurav

pkg/file.go

@@ -55,6 +55,35 @@ func EditKeyValue(filepath, key, oldvalue, newvalue string) error {
 	return nil
 }
 
+// AddAfterMatch will add a new line when a match is found
+func AddAfterMatch(filepath, key, newvalue string) error {
+	failFlag := true
+	fileData, err := ioutil.ReadFile(filepath)
+	if err != nil {
+		return errors.Wrapf(err, "Failed to read the given file, due to:%v", err)
+	}
+	lines := strings.Split(string(fileData), "\n")
+
+	for i, line := range lines {
+		if strings.Contains(line, key) {
+			lines = append(lines, "")
+			copy(lines[i+2:], lines[i+1:])
+			lines[i+1] = newvalue
+			failFlag = false
+		}
+	}
+	if failFlag {
+		return errors.Errorf("Error in adding \"%v\", \"%v\" not found ", newvalue, key)
+	}
+	output := strings.Join(lines, "\n")
+	err = ioutil.WriteFile(filepath, []byte(output), 0644)
+	if err != nil {
+		return errors.Wrapf(err, "Failed to write the data in the given file, due to:%v", err)
+	}
+
+	return nil
+}
+
 // DownloadFile will download a url to a local file. It's efficient because it will
 // write as it downloads and not load the whole file into memory.
 func DownloadFile(filepath string, url string) error {

tests/container-kill_test.go

@@ -81,7 +81,7 @@ var _ = Describe("BDD of container-kill experiment", func() {
 			Expect(err).To(BeNil(), "Failed to update the total chaos duration")
 			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "CONTAINER_RUNTIME", "value: 'docker'", "value: '"+pkg.GetEnv("CONTAINER_RUNTIME", "docker")+"'")
 			Expect(err).To(BeNil(), "Fail to update the network runtime")
-			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "SOCKET_PATH", "value: '/run/containerd/containerd.sock'", "value: '"+pkg.GetEnv("SOCKET_PATH", "/run/containerd/containerd.sock")+"'")
+			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "SOCKET_PATH", "value: '/var/run/docker.sock'", "value: '"+pkg.GetEnv("SOCKET_PATH", "/run/containerd/containerd.sock")+"'")
 			Expect(err).To(BeNil(), "Fail to update the socket path env")
 
 			//Creating ChaosEngine

tests/pod-cpu-hog_test.go

@@ -77,12 +77,10 @@ var _ = Describe("BDD of pod-cpu-hog experiment", func() {
 			Expect(err).To(BeNil(), "Failed to update total chaos duration")
 			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "NODE_CPU_CORE", "value: ''", "value: '"+pkg.GetEnv("CPU_CORES", "1")+"'")
 			Expect(err).To(BeNil(), "Failed to update the cpu cores")
-			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "TARGET_CONTAINER", "value: 'nginx'", "value: '"+pkg.GetEnv("TARGET_CONTAINER", "nginx")+"'")
-			Expect(err).To(BeNil(), "Failed to update the target container name")
-			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "CHAOS_KILL_COMMAND", "value: \"kill -9 $(ps afx | grep \"[md5sum] /dev/zero\" | awk '{print$1}' | tr '\n' ' ')\"", "value: '"+pkg.GetEnv("CHAOS_KILL_COMMAND", "\"kill -9 $(ps afx | grep \"[md5sum] /dev/zero\" | awk '{print$1}' | tr '\n' ' ')\"")+"'")
-			Expect(err).To(BeNil(), "Failed to update the chaos kill command")
-			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "CHAOS_INJECT_COMMAND", "value: 'md5sum /dev/zero'", "value: '"+pkg.GetEnv("CHAOS_INJECT_COMMAND", "md5sum /dev/zero")+"'")
-			Expect(err).To(BeNil(), "Failed to update the chaos inject command")
+			err = pkg.AddAfterMatch(experimentName+"-ce.yaml", "value: '60'", "\n            - name: CHAOS_INJECT_COMMAND\n              value: '"+pkg.GetEnv("CHAOS_INJECT_COMMAND", "md5sum /dev/zero")+"'")
+			Expect(err).To(BeNil(), "Failed to add the chaos inject command")
+			err = pkg.AddAfterMatch(experimentName+"-ce.yaml", "value: '60'", "\n            - name: CHAOS_KILL_COMMAND\n              value: "+pkg.GetEnv("CHAOS_KILL_COMMAND", "kill -9 $(ps afx | grep \"[md5sum] /dev/zero\" | awk '{print$1}' | tr '\\n' ' ')")+"")
+			Expect(err).To(BeNil(), "Failed to add the chaos kill command")
 
 			//Creating ChaosEngine
 			By("Creating ChaosEngine")

tests/pod-memory-hog_test.go

@@ -77,10 +77,8 @@ var _ = Describe("BDD of pod-memory-hog experiment", func() {
 			Expect(err).To(BeNil(), "Failed to update total chaos duration")
 			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "NODE_CPU_CORE", "value: ''", "value: '"+pkg.GetEnv("MEMORY_CONSUMPTION", "500")+"'")
 			Expect(err).To(BeNil(), "Failed to update the memory consumption in megabytes")
-			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "TARGET_CONTAINER", "value: 'nginx'", "value: '"+pkg.GetEnv("TARGET_CONTAINER", "nginx")+"'")
-			Expect(err).To(BeNil(), "Failed to update the target container name")
-			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "CHAOS_KILL_COMMAND", "value: \"kill -9 $(ps afx | grep \"[md5sum] /dev/zero\" | awk '{print$1}' | tr '\n' ' ')\"", "value: '"+pkg.GetEnv("CHAOS_KILL_COMMAND", "\"kill -9 $(ps afx | grep \"[md5sum] /dev/zero\" | awk '{print$1}' | tr '\n' ' ')\"")+"'")
-			Expect(err).To(BeNil(), "Failed to update the chaos kill command")
+			err = pkg.AddAfterMatch(experimentName+"-ce.yaml", "value: '60'", "\n            - name: CHAOS_KILL_COMMAND\n              value: "+pkg.GetEnv("CHAOS_KILL_COMMAND", "kill -9 $(ps afx | grep \"[md5sum] /dev/zero\" | awk '{print$1}' | tr '\\n' ' ')")+"")
+			Expect(err).To(BeNil(), "Failed to add the chaos kill command")
 
 			//Creating ChaosEngine
 			By("Creating ChaosEngine")

tests/pod-network-corruption_test.go

@@ -77,11 +77,9 @@ var _ = Describe("BDD of pod-network-corruption experiment", func() {
 			Expect(err).To(BeNil(), "Failed to update total chaos duration")
 			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "NETWORK_INTERFACE", "value: ''", "value: '"+pkg.GetEnv("NETWORK_INTERFACE", "eth0")+"'")
 			Expect(err).To(BeNil(), "Failed to update the network interface")
-			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "TARGET_CONTAINER", "value: 'nginx'", "value: '"+pkg.GetEnv("TARGET_CONTAINER", "nginx")+"'")
-			Expect(err).To(BeNil(), "Failed to update the target container name")
 			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "CONTAINER_RUNTIME", "value: 'docker'", "value: '"+pkg.GetEnv("CONTAINER_RUNTIME", "docker")+"'")
 			Expect(err).To(BeNil(), "Fail to update the network runtime")
-			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "SOCKET_PATH", "value: '/run/containerd/containerd.sock'", "value: '"+pkg.GetEnv("SOCKET_PATH", "/run/containerd/containerd.sock")+"'")
+			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "SOCKET_PATH", "value: '/var/run/docker.sock'", "value: '"+pkg.GetEnv("SOCKET_PATH", "/run/containerd/containerd.sock")+"'")
 			Expect(err).To(BeNil(), "Fail to update the socket path env")
 
 			//Creating ChaosEngine

tests/pod-network-duplication_test.go

@@ -77,11 +77,9 @@ var _ = Describe("BDD of pod-network-duplication experiment", func() {
 			Expect(err).To(BeNil(), "Failed to update total chaos duration")
 			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "NETWORK_INTERFACE", "value: ''", "value: '"+pkg.GetEnv("NETWORK_INTERFACE", "eth0")+"'")
 			Expect(err).To(BeNil(), "Failed to update the network interface")
-			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "TARGET_CONTAINER", "value: 'nginx'", "value: '"+pkg.GetEnv("TARGET_CONTAINER", "nginx")+"'")
-			Expect(err).To(BeNil(), "Failed to update the target container name")
 			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "CONTAINER_RUNTIME", "value: 'docker'", "value: '"+pkg.GetEnv("CONTAINER_RUNTIME", "docker")+"'")
 			Expect(err).To(BeNil(), "Fail to update the network runtime")
-			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "SOCKET_PATH", "value: '/run/containerd/containerd.sock'", "value: '"+pkg.GetEnv("SOCKET_PATH", "/run/containerd/containerd.sock")+"'")
+			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "SOCKET_PATH", "value: '/var/run/docker.sock'", "value: '"+pkg.GetEnv("SOCKET_PATH", "/run/containerd/containerd.sock")+"'")
 			Expect(err).To(BeNil(), "Fail to update the socket path env")
 			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "NETWORK_PACKET_DUPLICATION_PERCENTAGE", "value: '100'", "value: '"+pkg.GetEnv("NETWORK_PACKET_DUPLICATION_PERCENTAGE", "100")+"'")
 			Expect(err).To(BeNil(), "Fail to update the network packet duplication percentage env")

tests/pod-network-latency_test.go

@@ -77,13 +77,11 @@ var _ = Describe("BDD of pod-network-latency experiment", func() {
 			Expect(err).To(BeNil(), "Failed to update total chaos duration")
 			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "NETWORK_INTERFACE", "value: ''", "value: '"+pkg.GetEnv("NETWORK_INTERFACE", "eth0")+"'")
 			Expect(err).To(BeNil(), "Failed to update the network interface")
-			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "TARGET_CONTAINER", "value: 'nginx'", "value: '"+pkg.GetEnv("TARGET_CONTAINER", "nginx")+"'")
-			Expect(err).To(BeNil(), "Failed to update the target container name")
 			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "NETWORK_LATENCY", "value: 'nginx'", "value: '"+pkg.GetEnv("NETWORK_LATENCY", "60000")+"'")
 			Expect(err).To(BeNil(), "Failed to update the network latency value")
 			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "CONTAINER_RUNTIME", "value: 'docker'", "value: '"+pkg.GetEnv("CONTAINER_RUNTIME", "docker")+"'")
 			Expect(err).To(BeNil(), "Fail to update the network runtime")
-			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "SOCKET_PATH", "value: '/run/containerd/containerd.sock'", "value: '"+pkg.GetEnv("SOCKET_PATH", "/run/containerd/containerd.sock")+"'")
+			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "SOCKET_PATH", "value: '/var/run/docker.sock'", "value: '"+pkg.GetEnv("SOCKET_PATH", "/run/containerd/containerd.sock")+"'")
 			Expect(err).To(BeNil(), "Fail to update the socket path env")
 
 			//Creating ChaosEngine

tests/pod-network-loss_test.go

@@ -77,13 +77,11 @@ var _ = Describe("BDD of pod-network-loss experiment", func() {
 			Expect(err).To(BeNil(), "Failed to update total chaos duration")
 			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "NETWORK_INTERFACE", "value: ''", "value: '"+pkg.GetEnv("NETWORK_INTERFACE", "eth0")+"'")
 			Expect(err).To(BeNil(), "Failed to update the network interface")
-			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "TARGET_CONTAINER", "value: 'nginx'", "value: '"+pkg.GetEnv("TARGET_CONTAINER", "nginx")+"'")
-			Expect(err).To(BeNil(), "Failed to update the target container name")
 			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "NETWORK_PACKET_LOSS_PERCENTAGE", "value: 'nginx'", "value: '"+pkg.GetEnv("NETWORK_PACKET_LOSS_PERCENTAGE", "100")+"'")
 			Expect(err).To(BeNil(), "Failed to update the network packet loss percentage")
 			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "CONTAINER_RUNTIME", "value: 'docker'", "value: '"+pkg.GetEnv("CONTAINER_RUNTIME", "docker")+"'")
 			Expect(err).To(BeNil(), "Fail to update the network runtime")
-			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "SOCKET_PATH", "value: '/run/containerd/containerd.sock'", "value: '"+pkg.GetEnv("SOCKET_PATH", "/run/containerd/containerd.sock")+"'")
+			err = pkg.EditKeyValue(experimentName+"-ce.yaml", "SOCKET_PATH", "value: '/var/run/docker.sock'", "value: '"+pkg.GetEnv("SOCKET_PATH", "/run/containerd/containerd.sock")+"'")
 			Expect(err).To(BeNil(), "Fail to update the socket path env")
 
 			//Creating ChaosEngine