oauth2: remove unneeded TokenSource implementation in transport test
Change-Id: I2dca3a211f7b7e30ae023635c5ac44b6237ee01c
Reviewed-on: https://go-review.googlesource.com/c/oauth2/+/114957
Reviewed-by: Michael Pratt <mpratt@google.com>
Auto-Submit: Sean Liao <sean@liao.dev>
Reviewed-by: Sean Liao <sean@liao.dev>
LUCI-TryBot-Result: Go LUCI <golang-scoped@luci-project-accounts.iam.gserviceaccount.com>
Reviewed-by: Junyang Shao <shaojunyang@google.com>
diff --git a/transport_test.go b/transport_test.go
index faa87d5..a8e6ea2 100644
--- a/transport_test.go
+++ b/transport_test.go
@@ -9,12 +9,6 @@
"time"
)
-type tokenSource struct{ token *Token }
-
-func (t *tokenSource) Token() (*Token, error) {
- return t.token, nil
-}
-
func TestTransportNilTokenSource(t *testing.T) {
tr := &Transport{}
server := newMockServer(func(w http.ResponseWriter, r *http.Request) {})
@@ -88,13 +82,10 @@
}
func TestTransportTokenSource(t *testing.T) {
- ts := &tokenSource{
- token: &Token{
- AccessToken: "abc",
- },
- }
tr := &Transport{
- Source: ts,
+ Source: StaticTokenSource(&Token{
+ AccessToken: "abc",
+ }),
}
server := newMockServer(func(w http.ResponseWriter, r *http.Request) {
if got, want := r.Header.Get("Authorization"), "Bearer abc"; got != want {
@@ -123,14 +114,11 @@
{key: "basic", val: val, want: "Basic abc"},
}
for _, tc := range tests {
- ts := &tokenSource{
- token: &Token{
+ tr := &Transport{
+ Source: StaticTokenSource(&Token{
AccessToken: tc.val,
TokenType: tc.key,
- },
- }
- tr := &Transport{
- Source: ts,
+ }),
}
server := newMockServer(func(w http.ResponseWriter, r *http.Request) {
if got, want := r.Header.Get("Authorization"), tc.want; got != want {
