start working on cleaning up the output in yml , xml and json

Signed-off-by: Fokion Sotiropoulos <[email protected]>

process_testcase.go

@@ -141,22 +141,25 @@ func (v *Venom) runTestCase(ctx context.Context, ts *TestSuite, tc *TestCase) {
 	tc.Vars.AddAll(ts.ComputedVars)
 	tc.computedVars = H{}
 
-	computedSecrets := []string{}
+	ctx = v.processSecrets(ctx, ts, tc)
+
+	Info(ctx, "Starting testcase")
+
+	defer Info(ctx, "Ending testcase")
+	// ##### RUN Test Steps Here
+	v.runTestSteps(ctx, tc, nil)
+}
 
+func (v *Venom) processSecrets(ctx context.Context, ts *TestSuite, tc *TestCase) context.Context {
+	computedSecrets := []string{}
 	for k, v := range tc.Vars {
 		for _, s := range ts.Secrets {
 			if strings.Compare(k, s) == 0 {
 				computedSecrets = append(computedSecrets, fmt.Sprint(v))
 			}
 		}
 	}
-	ctx = context.WithValue(ctx, ContextKey("secrets"), computedSecrets)
-
-	Info(ctx, "Starting testcase")
-
-	defer Info(ctx, "Ending testcase")
-	// ##### RUN Test Steps Here
-	v.runTestSteps(ctx, tc, nil)
+	return context.WithValue(ctx, ContextKey("secrets"), computedSecrets)
 }
 
 func (v *Venom) runTestSteps(ctx context.Context, tc *TestCase, tsIn *TestStepResult) {

venom_output.go

@@ -2,6 +2,7 @@ package venom
 
 import (
 	"bytes"
+	"context"
 	"encoding/json"
 	"encoding/xml"
 	"fmt"
@@ -23,6 +24,23 @@ func init() {
 	}
 }
 
+func (v *Venom) CleanUpSecrets(testSuite *TestSuite) *TestSuite {
+	for _, testCase := range testSuite.TestCases {
+		ctx := v.processSecrets(context.Background(), testSuite, &testCase)
+		for _, result := range testCase.TestStepResults {
+			for i, v := range result.ComputedVars {
+				result.ComputedVars[i] = HideSensitive(ctx, v)
+			}
+			for i, v := range result.InputVars {
+				result.ComputedVars[i] = HideSensitive(ctx, v)
+			}
+			result.Systemout = HideSensitive(ctx, result.Systemout)
+			result.Systemerr = HideSensitive(ctx, result.Systemerr)
+		}
+	}
+	return testSuite
+}
+
 // OutputResult output result to sdtout, files...
 func (v *Venom) OutputResult() error {
 	if v.OutputDir == "" {