fixing tests

surface-security · Jun 22, 2023 · 4cfaadc · 4cfaadc
1 parent 4c8ac67
commit 4cfaadc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/runner/runner.go b/runner/runner.go
@@ -709,8 +709,13 @@ func (r *Runner) RunEnumeration() {
 					gologger.Warning().Msgf("Could not decode response: %s\n", err)
 					continue
 				}
-				dslVars, _ := dslVariables()
+				dslVars, err := dslVariables()
+				if err != nil {
+					gologger.Warning().Msgf("Could not retrieve dsl variables: %s\n", err)
+					continue
+				}
 				flatMap := make(map[string]interface{})
+
 				for _, v := range dslVars {
 					flatMap[v] = rawMap[v]
 				}

diff --git a/runner/types.go b/runner/types.go
@@ -82,7 +82,7 @@ type Result struct {
 // function to get dsl variables from result struct
 func dslVariables() ([]string, error) {
 	fakeResult := Result{}
-	fieldsToIgnore := []string{"Hashes", "ResponseHeader", "Err"}
+	fieldsToIgnore := []string{"Hashes", "ResponseHeader", "Err", "KnowledgeBase"}
 	if err := faker.FakeData(&fakeResult, options.WithFieldsToIgnore(fieldsToIgnore...)); err != nil {
 		return nil, err
 	}