Fix filter tests (#368)

This commit is contained in:
Joona Hoikkala 2020-12-28 13:53:09 +02:00 committed by GitHub
parent bc5e61ecc3
commit 4bea474ae6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 2 deletions

View File

@ -43,7 +43,7 @@ func TestLineFiltering(t *testing.T) {
for i := int64(0); i < test.input; i++ {
data = append(data, "A")
}
resp := ffuf.Response{Data: []byte(strings.Join(data, " "))}
resp := ffuf.Response{Data: []byte(strings.Join(data, "\n"))}
filterReturn, _ := f.Filter(&resp)
if filterReturn != test.output {
t.Errorf("Filter test %d: Was expecing filter return value of %t but got %t", i, test.output, filterReturn)

View File

@ -35,7 +35,13 @@ func TestRegexpFiltering(t *testing.T) {
{"s1arch", false},
{"invalid", false},
} {
resp := ffuf.Response{Data: []byte(test.input)}
inp := make(map[string][]byte)
resp := ffuf.Response{
Data: []byte(test.input),
Request: &ffuf.Request{
Input: inp,
},
}
filterReturn, _ := f.Filter(&resp)
if filterReturn != test.output {
t.Errorf("Filter test %d: Was expecing filter return value of %t but got %t", i, test.output, filterReturn)