1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
package zts
import (
"testing"
"time"
"github.com/vespa-engine/vespa/client/go/internal/mock"
)
type manualClock struct{ t time.Time }
func (c *manualClock) now() time.Time { return c.t }
func (c *manualClock) advance(d time.Duration) { c.t = c.t.Add(d) }
func TestAccessToken(t *testing.T) {
httpClient := mock.HTTPClient{}
client, err := NewClient(&httpClient, "vespa.vespa", "http://example.com")
if err != nil {
t.Fatal(err)
}
clock := &manualClock{t: time.Now()}
client.now = clock.now
httpClient.NextResponseString(400, `{"message": "bad request"}`)
_, err = client.AccessToken()
if err == nil {
t.Fatal("want error for non-ok response status")
}
httpClient.NextResponseString(200, `{"access_token": "foo", "expires_in": 3600}`)
token, err := client.AccessToken()
if err != nil {
t.Fatal(err)
}
// Token is cached
expiresAt := clock.now().Add(time.Hour)
assertToken(t, Token{Value: "foo", ExpiresAt: expiresAt}, token)
clock.advance(54 * time.Minute)
assertToken(t, Token{Value: "foo", ExpiresAt: expiresAt}, token)
// Token is renewed when nearing expiry
clock.advance(time.Minute + time.Second)
httpClient.NextResponseString(200, `{"access_token": "bar", "expires_in": 1800}`)
token, err = client.AccessToken()
if err != nil {
t.Fatal(err)
}
expiresAt = clock.now().Add(30 * time.Minute)
assertToken(t, Token{Value: "bar", ExpiresAt: expiresAt}, token)
// Request body is included in error
httpClient.NextResponseString(503, "broken!")
_, err = client.AccessToken()
want := "zts: got status 503 (broken!) from http://example.com/zts/v1/oauth2/token"
if got := err.Error(); got != want {
t.Errorf("got err=%q, want %q", got, want)
}
}
func assertToken(t *testing.T, want, got Token) {
if want.Value != got.Value {
t.Errorf("got Value=%q, want %q", got.Value, want.Value)
}
if want.ExpiresAt != got.ExpiresAt {
t.Errorf("got ExpiresAt=%s, want %s", got.ExpiresAt, want.ExpiresAt)
}
}
|
