added client and cofiguration

Signed-off-by: Oum Kale <[email protected]>

Author: oumkale

.gitignore

@@ -1,7 +1,6 @@
 **/.idea
 **/*.egg-info
 **/.DS_Store
-**/build
 **/*intuit*/**
 **/*intuit.md
 **/*intuit*

.travis.yml

@@ -24,6 +24,7 @@ jobs:
       script:
       - make chaostest
 
+
 notifications:
   email:
     recipients:

Makefile

@@ -16,15 +16,17 @@ DOCKER_TAG ?= ci
 help:
 	@echo ""
 	@echo "Usage:-"
+	@echo "\tmake deps          --  sets up dependencies for image build"
 	@echo "\tmake litmuspython  --  will build and push the litmus-python image"
-	@echo "\tmake chaostest     -- will build and push python experiment images
+	@echo "\tmake chaostest     --  will build and push python experiment images
 	@echo ""
 
 
 deps: _build_check_docker
 	@echo ""
 	@echo "INFO:\tverifying dependencies for litmus-python ..."
 
+.PHONY: litmuspython
 _build_tests_litmus_python_image:
 	@echo "INFO: Building container image for performing chaostoolkit tests"
 	docker build -t litmuschaos/litmus-python:ci .
@@ -36,7 +38,7 @@ _push_tests_litmus_python_image:
 
 litmuspython: deps _build_tests_litmus_python_image _push_tests_litmus_python_image
 
-
+.PHONY: chaostest
 _build_tests_chaostest_image:
 	@echo "INFO: Building container image for performing chaostoolkit tests"
 	cd byoc && docker build -t litmuschaos/chaostoolkit:ci .

bin/experiment/chaosresult.yaml

@@ -0,0 +1,20 @@
+---
+apiVersion: litmuschaos.io/v1alpha1
+kind: ChaosResult
+metadata:
+    name: "pod-delete-pod-delete"
+    namespace: "litmus"
+    labels: {'component': 'litmusportal-frontend', 'pod-template-hash': '64898fdcc8', 'name': 'pod-delete-pod-delete'}
+spec:
+    engine: "pod-delete"
+    experiment: "pod-delete"
+status:
+    experimentStatus: 
+        phase: "Completed"
+        verdict: "Pass"
+        probeSuccessPercentage: "100"
+        failStep: "N/A"
+    history:
+        passedRuns: 1
+        failedRuns: 0
+        stoppedRuns: 0

bin/experiment/experiment.py

@@ -4,6 +4,8 @@
 import argparse
 import os
 import logging
+from pkg.utils.client.client import Configs, Client
+
 logging.basicConfig(format='time=%(asctime)s level=%(levelname)s  msg=%(message)s', level=logging.INFO)  
 def main():
 	parser = argparse.ArgumentParser()
@@ -21,11 +23,13 @@ def main():
                     help="Kubernetes client ignore SSL")
 	args = parser.parse_args()
 
+	config = Configs(kubeContext=args.kubeContext)
+	clients = Client(conf = config.get_config())
 	logging.info("Experiment Name: {}".format(args.experimentName))
 
 	# invoke the corresponding experiment based on the the (-name) flag
 	if args.experimentName == "pod-delete":
-		podDelete.PodDelete()
+		podDelete.PodDelete(clients)
 	else:
 		logging.warning("Unsupported -name {}, please provide the correct value of -name args".format(args.experimentName))
 	return

build/lib/bin/__init__.py

@@ -0,0 +1,20 @@
+---
+apiVersion: litmuschaos.io/v1alpha1
+kind: ChaosResult
+metadata:
+    name: "pod-delete-pod-delete"
+    namespace: "litmus"
+    labels: {'component': 'litmusportal-frontend', 'pod-template-hash': '64898fdcc8', 'name': 'pod-delete-pod-delete'}
+spec:
+    engine: "pod-delete"
+    experiment: "pod-delete"
+status:
+    experimentStatus: 
+        phase: "Completed"
+        verdict: "Pass"
+        probeSuccessPercentage: "100"
+        failStep: "N/A"
+    history:
+        passedRuns: 1
+        failedRuns: 0
+        stoppedRuns: 0

build/lib/bin/experiment/__init__.py

@@ -0,0 +1,38 @@
+#!/usr/bin/env python3
+
+import experimentList.generic.podDelete.podDelete as podDelete
+import argparse
+import os
+import logging
+from pkg.utils.client.client import Configs, Client
+
+logging.basicConfig(format='time=%(asctime)s level=%(levelname)s  msg=%(message)s', level=logging.INFO)  
+def main():
+	parser = argparse.ArgumentParser()
+	
+	# parse the experiment name
+	parser.add_argument("-experimentName", action='store',
+                    default="pod-delete",
+                    dest="experimentName",
+                    help="Chaos experiment to chose for execution"
+                    )
+	parser.add_argument("-kubeContext",
+                    required=False,
+                    default=os.environ.get("KUBECONFIG", ""),
+                    dest='kubeContext',
+                    help="Kubernetes client ignore SSL")
+	args = parser.parse_args()
+	
+	config = Configs(kubeContext=args.kubeContext)
+	clients = Client(conf = config.get_config())
+	logging.info("Experiment Name: {}".format(args.experimentName))
+
+	# invoke the corresponding experiment based on the the (-name) flag
+	if args.experimentName == "pod-delete":
+		podDelete.PodDelete(clients)
+	else:
+		logging.warning("Unsupported -name {}, please provide the correct value of -name args".format(args.experimentName))
+	return
+if __name__ == "__main__":
+	main()
+	

build/lib/bin/experiment/chaosresult.yaml

@@ -0,0 +1,171 @@
+
+import pkg.types.types  as types
+import pkg.events.events as events
+from kubernetes import client, config
+import time
+import logging
+logging.basicConfig(format='time=%(asctime)s level=%(levelname)s  msg=%(message)s', level=logging.INFO)  
+import pkg.utils.common.common as common
+from pkg.utils.common.pods import Pods
+from datetime import datetime
+from pkg.status.application import Application
+import pkg.maths.maths as maths
+import os
+
+#PreparePodDelete contains the prepration steps before chaos injection
+def PreparePodDelete(experimentsDetails , resultDetails, eventsDetails, chaosDetails, clients):
+
+	#Waiting for the ramp time before chaos injection
+	if experimentsDetails.RampTime != 0 :
+		logging.info("[Ramp]: Waiting for the %s ramp time before injecting chaos",(experimentsDetails.RampTime))
+		common.WaitForDuration(experimentsDetails.RampTime)
+	
+	if experimentsDetails.Sequence.lower() == "serial":
+		err = injectChaosInSerialMode(experimentsDetails, chaosDetails, eventsDetails, resultDetails, clients)
+		if err != None:
+			return err
+	elif experimentsDetails.Sequence.lower() == "parallel":
+		err = injectChaosInParallelMode(experimentsDetails, chaosDetails, eventsDetails, resultDetails, clients)
+		if err != None:
+			return err
+	else:
+		logging.warning("[Warning]: %s sequence is not supported",(experimentsDetails.Sequence))
+
+	#Waiting for the ramp time after chaos injection
+	if experimentsDetails.RampTime != 0 :
+		logging.info("[Ramp]: Waiting for the %s ramp time after injecting chaos",(experimentsDetails.RampTime))
+		common.WaitForDuration(experimentsDetails.RampTime)
+	return None
+
+# injectChaosInSerialMode delete the target application pods serial mode(one by one)
+def injectChaosInSerialMode(experimentsDetails , chaosDetails , eventsDetails , resultDetails, clients): 
+	
+	status = Application()
+	
+	pods = Pods()
+	GracePeriod = 0
+	
+	#ChaosStartTimeStamp contains the start timestamp, when the chaos injection begin
+	ChaosStartTimeStamp = datetime.now()
+	duration = (datetime.now() - ChaosStartTimeStamp).seconds
+	
+	while duration < experimentsDetails.ChaosDuration:
+		# Get the target pod details for the chaos execution
+		# if the target pod is not defined it will derive the random target pod list using pod affected percentage
+		if experimentsDetails.TargetPods == "" and chaosDetails.AppDetail.Label == "" :
+			return logging.warning("[Warning]: Please provide one of the appLabel or TARGET_PODS")
+		
+		targetPodList, err = pods.GetPodList(experimentsDetails.TargetPods, experimentsDetails.PodsAffectedPerc, chaosDetails, clients)
+		if err != None: 
+			return err
+		
+		podNames = []
+		for pod in targetPodList.items:
+			podNames.append(pod.metadata.name)
+		
+		logging.info("[Info]: Target pods list, %s",(podNames))
+
+		if experimentsDetails.EngineName != "" :
+			msg = "Injecting " + experimentsDetails.ExperimentName + " chaos on application pod"
+			types.SetEngineEventAttributes(eventsDetails, types.ChaosInject, msg, "Normal", chaosDetails)
+			events.GenerateEvents(eventsDetails, chaosDetails, "ChaosEngine", clients)
+		
+		#Deleting the application pod
+		for pod in targetPodList.items :
+
+			logging.info("[Info]: Killing the following pods, PodName : %s", pod.metadata.name)
+			try:
+				if experimentsDetails.Force == True:
+					err = clients.clientk8s.delete_namespaced_pod(pod.metadata.name, experimentsDetails.AppNS, grace_period_seconds=GracePeriod)
+				else:
+					err = clients.clientk8s.delete_namespaced_pod(pod.metadata.name, experimentsDetails.AppNS)
+			except Exception as e:
+				return e
+
+			if chaosDetails.Randomness == True :
+				err = common.RandomInterval(experimentsDetails.ChaosInterval)
+				if err != None:
+					return err
+			else:
+				#Waiting for the chaos interval after chaos injection
+				if experimentsDetails.ChaosInterval != "":
+					logging.info("[Wait]: Wait for the chaos interval %s",(experimentsDetails.ChaosInterval))
+					waitTime = maths.atoi(experimentsDetails.ChaosInterval)
+					common.WaitForDuration(waitTime)
+
+			#Verify the status of pod after the chaos injection
+			logging.info("[Status]: Verification for the recreation of application pod")
+			err = status.CheckApplicationStatus(experimentsDetails.AppNS, experimentsDetails.AppLabel, experimentsDetails.Timeout, experimentsDetails.Delay,clients)
+			if err != None:
+				return err
+			
+			duration = (datetime.now() - ChaosStartTimeStamp).seconds
+
+	logging.info("[Completion]: %s chaos is done",(experimentsDetails.ExperimentName))
+
+	return None
+
+# injectChaosInParallelMode delete the target application pods in parallel mode (all at once)
+def injectChaosInParallelMode(experimentsDetails , chaosDetails , eventsDetails , resultDetails, clients):
+	
+	status = Application()
+	pods = Pods()
+
+	GracePeriod = 0
+	
+	#ChaosStartTimeStamp contains the start timestamp, when the chaos injection begin
+	ChaosStartTimeStamp = datetime.now()
+	duration = (datetime.now() - ChaosStartTimeStamp).seconds
+	
+	while duration < experimentsDetails.ChaosDuration:
+		# Get the target pod details for the chaos execution
+		# if the target pod is not defined it will derive the random target pod list using pod affected percentage
+		if experimentsDetails.TargetPods == "" and chaosDetails.AppDetail.Label == "" :
+			return logging.info("[Warning]: Please provide one of the appLabel or TARGET_PODS")
+		
+		targetPodList, err = pods.GetPodList(experimentsDetails.TargetPods, experimentsDetails.PodsAffectedPerc, chaosDetails, clients)
+		if err != None:
+			return err
+		podNames = []
+		for pod in targetPodList.items:
+			podNames.append(str(pod.metadata.name))
+		
+		logging.info("[Info]: Target pods list for chaos, %s",(podNames))
+		
+		if experimentsDetails.EngineName != "" :
+			msg = "Injecting " + experimentsDetails.ExperimentName + " chaos on application pod"
+			types.SetEngineEventAttributes(eventsDetails, types.ChaosInject, msg, "Normal", chaosDetails)
+			events.GenerateEvents(eventsDetails, chaosDetails, "ChaosEngine",clients)
+
+		#Deleting the application pod
+		for pod in targetPodList.items:
+			logging.info("[Info]: Killing the following pods, PodName : %s", pod.metadata.name)
+			try:
+				if experimentsDetails.Force == True:
+					clients.clientk8s.delete_namespaced_pod(pod.metadata.name, experimentsDetails.AppNS, grace_period_seconds=GracePeriod)
+				else:
+					clients.clientk8s.delete_namespaced_pod(pod.metadata.name, experimentsDetails.AppNS)
+			except Exception as err:
+				return err	
+		
+		if chaosDetails.Randomness == True:
+			err = common.RandomInterval(experimentsDetails.ChaosInterval)
+			if err != None:
+				return err
+		else:
+			#Waiting for the chaos interval after chaos injection
+			if experimentsDetails.ChaosInterval != "" :
+				logging.info("[Wait]: Wait for the chaos interval %s", experimentsDetails.ChaosInterval)
+				waitTime = maths.atoi(experimentsDetails.ChaosInterval)
+				common.WaitForDuration(waitTime)
+
+		#Verify the status of pod after the chaos injection
+		logging.info("[Status]: Verification for the recreation of application pod")
+		err = status.CheckApplicationStatus(experimentsDetails.AppNS, experimentsDetails.AppLabel, experimentsDetails.Timeout, experimentsDetails.Delay, clients)
+		if err != None:
+			return err
+		duration = (datetime.now() - ChaosStartTimeStamp).seconds
+
+	logging.info("[Completion]: %s chaos is done",(experimentsDetails.ExperimentName))
+
+	return None