fix tests

projectdiscovery · Oct 11, 2023 · 128cad8 · 128cad8
1 parent 38ef937
commit 128cad8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/inputs_test.go b/inputs_test.go
@@ -62,9 +62,10 @@ func TestExtractVar(t *testing.T) {
 	}{
 		{statement: "{{sub}}.something.{{tld}}", expected: []string{"sub", "tld"}},
 		{statement: "{{sub}}.{{sub1}}.{{sub2}}.{{root}}", expected: []string{"sub", "sub1", "sub2", "root"}},
-		{statement: "no variables", expected: []string{}},
+		{statement: "no variables", expected: nil},
 	}
 	for _, v := range testcases {
-		require.Equal(t, v.expected, getAllVars(v.statement))
+		actual := getAllVars(v.statement)
+		require.Equal(t, v.expected, actual)
 	}
 }
diff --git a/mutator_test.go b/mutator_test.go
@@ -2,6 +2,7 @@ package alterx
 
 import (
 	"bytes"
+	"math"
 	"strings"
 	"testing"
 
@@ -39,6 +40,7 @@ func TestMutatorResults(t *testing.T) {
 	}
 	opts.Patterns = testConfig.Patterns
 	opts.Payloads = testConfig.Payloads
+	opts.MaxSize = math.MaxInt
 	m, err := New(opts)
 	require.Nil(t, err)
 	var buff bytes.Buffer