oauth2: don't use tokenRefresher directly in tests

This changes the test to use the exported API instead of implementation
internals.

Change-Id: I07753b053c1a2a8eb027bed647ab921d95afda4e
Reviewed-on: https://go-review.googlesource.com/85658
Reviewed-by: Andrew Bonventre <andybons@golang.org>
This commit is contained in:
Ross Light 2017-12-27 16:42:28 -08:00
parent 90155042cb
commit 542ae755da

View File

@ -453,18 +453,14 @@ func TestRefreshToken_RefreshTokenReplacement(t *testing.T) {
})) }))
defer ts.Close() defer ts.Close()
conf := newConf(ts.URL) conf := newConf(ts.URL)
tkr := &tokenRefresher{ tkr := conf.TokenSource(context.Background(), &Token{RefreshToken: "OLD_REFRESH_TOKEN"})
conf: conf,
ctx: context.Background(),
refreshToken: "OLD_REFRESH_TOKEN",
}
tk, err := tkr.Token() tk, err := tkr.Token()
if err != nil { if err != nil {
t.Errorf("got err = %v; want none", err) t.Errorf("got err = %v; want none", err)
return return
} }
if tk.RefreshToken != tkr.refreshToken { if want := "NEW_REFRESH_TOKEN"; tk.RefreshToken != want {
t.Errorf("tokenRefresher.refresh_token = %q; want %q", tkr.refreshToken, tk.RefreshToken) t.Errorf("RefreshToken = %q; want %q", tk.RefreshToken, want)
} }
} }
@ -477,17 +473,13 @@ func TestRefreshToken_RefreshTokenPreservation(t *testing.T) {
defer ts.Close() defer ts.Close()
conf := newConf(ts.URL) conf := newConf(ts.URL)
const oldRefreshToken = "OLD_REFRESH_TOKEN" const oldRefreshToken = "OLD_REFRESH_TOKEN"
tkr := &tokenRefresher{ tkr := conf.TokenSource(context.Background(), &Token{RefreshToken: oldRefreshToken})
conf: conf, tk, err := tkr.Token()
ctx: context.Background(),
refreshToken: oldRefreshToken,
}
_, err := tkr.Token()
if err != nil { if err != nil {
t.Fatalf("got err = %v; want none", err) t.Fatalf("got err = %v; want none", err)
} }
if tkr.refreshToken != oldRefreshToken { if tk.RefreshToken != oldRefreshToken {
t.Errorf("tokenRefresher.refreshToken = %q; want %q", tkr.refreshToken, oldRefreshToken) t.Errorf("RefreshToken = %q; want %q", tk.RefreshToken, oldRefreshToken)
} }
} }