diff options
Diffstat (limited to 'request/request_test.go')
| -rw-r--r-- | request/request_test.go | 27 |
1 files changed, 23 insertions, 4 deletions
diff --git a/request/request_test.go b/request/request_test.go index f99e5cd87..c58612605 100644 --- a/request/request_test.go +++ b/request/request_test.go @@ -14,21 +14,21 @@ func TestRequestDo(t *testing.T) { st.Do() if st.do == nil { - t.Fatalf("Expected st.do to be set") + t.Errorf("Expected st.do to be set") } } func TestRequestRemote(t *testing.T) { st := testRequest() if st.IP() != "10.240.0.1" { - t.Fatalf("Wrong IP from request") + t.Errorf("Wrong IP from request") } p := st.Port() if p == "" { - t.Fatalf("Failed to get Port from request") + t.Errorf("Failed to get Port from request") } if p != "40212" { - t.Fatalf("Wrong port from request") + t.Errorf("Wrong port from request") } } @@ -202,3 +202,22 @@ func testRequest() Request { m.SetEdns0(4097, true) return Request{W: &test.ResponseWriter{}, Req: m} } + +func TestRequestClear(t *testing.T) { + st := testRequest() + if st.IP() != "10.240.0.1" { + t.Errorf("Wrong IP from request") + } + p := st.Port() + if p == "" { + t.Errorf("Failed to get Port from request") + } + st.Clear() + if st.ip != "" { + t.Errorf("Expected st.ip to be cleared after Clear") + } + + if st.port != "" { + t.Errorf("Expected st.port to be cleared after Clear") + } +} |