TrueCloudLab/coredns
16
0
Fork 2
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

request: add LocalIP (#1925)

Browse source 
* request: add LocalIP

Fix TODO that was added: Add LocalIP and test the Clear() method.

Signed-off-by: Miek Gieben <miek@miek.nl>

* Move to Errorf

PR feedback and move to Errorf instead Fatalf.

Signed-off-by: Miek Gieben <miek@miek.nl>
This commit is contained in:
Miek Gieben 2018-07-03 09:47:26 +01:00 • committed by GitHub
parent 6dd2cf8c4b
commit 908d4fbd23
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 34 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

27
request/request_test.go
View file

@ -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")
}
}