...
1
18
19 package tokenmanager
20
21 import (
22 "os"
23 "testing"
24
25 commonpb "github.com/google/s2a-go/internal/proto/common_go_proto"
26 )
27
28 const testAccessToken = "test_access_token"
29
30 func TestNewSingleTokenAccessTokenManagerFailure(t *testing.T) {
31 os.Unsetenv(s2aAccessTokenEnvironmentVariable)
32 if _, err := NewSingleTokenAccessTokenManager(); err == nil {
33 t.Errorf("expected NewSingleTokenAccessTokenManager() to return non-nil error")
34 }
35 }
36
37 func TestNewSingleTokenAccessTokenManagerDefaultTokenSuccess(t *testing.T) {
38 os.Setenv(s2aAccessTokenEnvironmentVariable, testAccessToken)
39 tokenManager, err := NewSingleTokenAccessTokenManager()
40 if err != nil {
41 t.Errorf("NewSingleTokenAccessTokenManager() returned unexpected error: %v", err)
42 }
43
44 token, err := tokenManager.DefaultToken()
45 if err != nil {
46 t.Errorf("tokenManager.DefaultToken() returned unexpected error: %v", err)
47 }
48 if got, want := token, testAccessToken; got != want {
49 t.Errorf("tokenManager.DefaultToken()= %v, want %s", got, want)
50 }
51 }
52
53 func TestNewSingleTokenAccessTokenManagerTokenSuccess(t *testing.T) {
54 os.Setenv(s2aAccessTokenEnvironmentVariable, testAccessToken)
55 tokenManager, err := NewSingleTokenAccessTokenManager()
56 if err != nil {
57 t.Errorf("NewSingleTokenAccessTokenManager() returned unexpected error: %v", err)
58 }
59
60 token, err := tokenManager.Token(&commonpb.Identity{})
61 if err != nil {
62 t.Errorf("tokenManager.Token() returned unexpected error: %v", err)
63 }
64 if got, want := token, testAccessToken; got != want {
65 t.Errorf("tokenManager.Token(%v)= %v, want %s", &commonpb.Identity{}, got, want)
66 }
67 }
68
View as plain text