-
Notifications
You must be signed in to change notification settings - Fork 1
/
dupefilters_test.go
67 lines (56 loc) · 2.33 KB
/
dupefilters_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package tegenaria
import (
"testing"
)
func TestDoDupeFilter(t *testing.T) {
server := newTestServer()
// downloader := NewDownloader()
headers := map[string]string{
"Params1": "params1",
"Intparams": "1",
"Boolparams": "false",
}
request1 := NewRequest(server.URL+"/testHeader", GET, testParser, RequestWithRequestHeader(headers))
request2 := NewRequest(server.URL+"/testHeader", GET, testParser, RequestWithRequestHeader(headers))
request3 := NewRequest(server.URL+"/testHeader2", GET, testParser, RequestWithRequestHeader(headers))
duplicates := NewRFPDupeFilter(1024*1024, 5)
if r1, _ := duplicates.DoDupeFilter(request1); r1 {
t.Errorf("Request1 error expected=%v, get=%v", false, true)
}
if r2, _ := duplicates.DoDupeFilter(request2); !r2 {
t.Errorf("Request2 error expected=%v, get=%v", true, false)
}
if r3, _ := duplicates.DoDupeFilter(request3); r3 {
t.Errorf("Request3 error expected=%v, get=%v", false, true)
}
}
func TestDoBodyDupeFilter(t *testing.T) {
server := newTestServer()
// downloader := NewDownloader()
headers := map[string]string{
"Params1": "params1",
"Intparams": "1",
"Boolparams": "false",
}
body := make(map[string]interface{})
body["test"] = "tegenaria"
body1 := make(map[string]interface{})
body1["test"] = "tegenaria2"
request1 := NewRequest(server.URL+"/testHeader", GET, testParser, RequestWithRequestHeader(headers),RequestWithRequestBody(body))
request2 := NewRequest(server.URL+"/testHeader", GET, testParser, RequestWithRequestHeader(headers),RequestWithRequestBody(body))
request3 := NewRequest(server.URL+"/testHeader2", GET, testParser, RequestWithRequestHeader(headers))
request4 := NewRequest(server.URL+"/testHeader", GET, testParser, RequestWithRequestHeader(headers),RequestWithRequestBody(body))
duplicates := NewRFPDupeFilter(1024*1024, 5)
if r1, err := duplicates.DoDupeFilter(request1); r1||err!=nil {
t.Errorf("Request1 error expected=%v, get=%v", false, true)
}
if r2, err := duplicates.DoDupeFilter(request2); !r2||err!=nil {
t.Errorf("Request2 error expected=%v, get=%v", true, false)
}
if r3, err := duplicates.DoDupeFilter(request3); r3||err!=nil {
t.Errorf("Request3 error expected=%v, get=%v", false, true)
}
if r4, err := duplicates.DoDupeFilter(request4); !r4||err!=nil {
t.Errorf("Request4 error expected=%v, get=%v", true, false)
}
}