Add check for setting HasProbeCompleted

Signed-off-by: nagesh bansal <[email protected]>

pkg/probe/cmdprobe.go

@@ -337,6 +337,11 @@ loop:
 		select {
 		case <-chaosDetails.ProbeContext.Ctx.Done():
 			log.Info("Chaos Execution completed. Stopping Probes")
+			for index := range chaosresult.ProbeDetails {
+				if chaosresult.ProbeDetails[index].Name == probe.Name {
+					chaosresult.ProbeDetails[index].HasProbeCompleted = true
+				}
+			}
 			break loop
 		default:
 			err := triggerInlineCmdProbe(probe, chaosresult)

pkg/probe/k8sprobe.go

@@ -141,6 +141,11 @@ loop:
 		select {
 		case <-chaosDetails.ProbeContext.Ctx.Done():
 			log.Info("Chaos Execution completed. Stopping Probes")
+			for index := range chaosresult.ProbeDetails {
+				if chaosresult.ProbeDetails[index].Name == probe.Name {
+					chaosresult.ProbeDetails[index].HasProbeCompleted = true
+				}
+			}
 			break loop
 
 		default:

pkg/probe/probe.go

@@ -254,7 +254,7 @@ func getDescription(err error) string {
 }
 
 // CheckForErrorInContinuousProbe check for the error in the continuous probes
-func checkForErrorInContinuousProbe(resultDetails *types.ResultDetails, probeName string, timeout int, delay int) error {
+func checkForErrorInContinuousProbe(resultDetails *types.ResultDetails, probeName string, delay int, timeout int) error {
 
 	probe := getProbeByName(probeName, resultDetails.ProbeDetails)
 	timeoutSignal := time.After(time.Duration(timeout) * time.Second)

pkg/probe/promProbe.go

@@ -249,6 +249,11 @@ loop:
 		select {
 		case <-chaosDetails.ProbeContext.Ctx.Done():
 			log.Info("Chaos Execution completed. Stopping Probes")
+			for index := range chaosresult.ProbeDetails {
+				if chaosresult.ProbeDetails[index].Name == probe.Name {
+					chaosresult.ProbeDetails[index].HasProbeCompleted = true
+				}
+			}
 			break loop
 		default:
 			err = triggerPromProbe(probe, chaosresult)