From 4bea474ae60a09f70944b348aaab350e04a0262b Mon Sep 17 00:00:00 2001 From: Joona Hoikkala Date: Mon, 28 Dec 2020 13:53:09 +0200 Subject: [PATCH] Fix filter tests (#368) --- pkg/filter/lines_test.go | 2 +- pkg/filter/regexp_test.go | 8 +++++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/pkg/filter/lines_test.go b/pkg/filter/lines_test.go index c6808a4..92c36d4 100644 --- a/pkg/filter/lines_test.go +++ b/pkg/filter/lines_test.go @@ -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) diff --git a/pkg/filter/regexp_test.go b/pkg/filter/regexp_test.go index f6cd770..20033b4 100644 --- a/pkg/filter/regexp_test.go +++ b/pkg/filter/regexp_test.go @@ -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)