Skip to content

Commit 8a57ed9

Browse files
odeke-embradfitz
authored andcommitted
transport_test: added TestNilTokenSource + close res.Body per GET
+ TestTransportNilTokenSource added. + res.Body.Close() in tests to cut leaks and also to increase test coverage for *onEOFReader. Change-Id: I1b962b7ecbc4f6f8b5bcccc6870c025e89ad9c1a Reviewed-on: https://go-review.googlesource.com/18141 Reviewed-by: Burcu Dogan <[email protected]>
1 parent 191c617 commit 8a57ed9

File tree

1 file changed

+26
-4
lines changed

1 file changed

+26
-4
lines changed

transport_test.go

Lines changed: 26 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,20 @@ func (t *tokenSource) Token() (*Token, error) {
1313
return t.token, nil
1414
}
1515

16+
func TestTransportNilTokenSource(t *testing.T) {
17+
tr := &Transport{}
18+
server := newMockServer(func(w http.ResponseWriter, r *http.Request) {})
19+
defer server.Close()
20+
client := &http.Client{Transport: tr}
21+
res, err := client.Get(server.URL)
22+
if err == nil {
23+
t.Errorf("a nil Source was passed into the transport expected an error")
24+
}
25+
if res != nil {
26+
t.Errorf("expected a nil response, got %v", res)
27+
}
28+
}
29+
1630
func TestTransportTokenSource(t *testing.T) {
1731
ts := &tokenSource{
1832
token: &Token{
@@ -28,8 +42,12 @@ func TestTransportTokenSource(t *testing.T) {
2842
}
2943
})
3044
defer server.Close()
31-
client := http.Client{Transport: tr}
32-
client.Get(server.URL)
45+
client := &http.Client{Transport: tr}
46+
res, err := client.Get(server.URL)
47+
if err != nil {
48+
t.Fatal(err)
49+
}
50+
res.Body.Close()
3351
}
3452

3553
// Test for case-sensitive token types, per https://github.com/golang/oauth2/issues/113
@@ -60,8 +78,12 @@ func TestTransportTokenSourceTypes(t *testing.T) {
6078
}
6179
})
6280
defer server.Close()
63-
client := http.Client{Transport: tr}
64-
client.Get(server.URL)
81+
client := &http.Client{Transport: tr}
82+
res, err := client.Get(server.URL)
83+
if err != nil {
84+
t.Fatal(err)
85+
}
86+
res.Body.Close()
6587
}
6688
}
6789

0 commit comments

Comments
 (0)