1 package oauth2
2
3 import (
4 "context"
5 "encoding/json"
6 "fmt"
7 "strings"
8 "testing"
9 "time"
10
11 "github.com/google/go-cmp/cmp"
12 "github.com/google/go-cmp/cmp/cmpopts"
13 )
14
15 func TestDeviceAuthResponseMarshalJson(t *testing.T) {
16 tests := []struct {
17 name string
18 response DeviceAuthResponse
19 want string
20 }{
21 {
22 name: "empty",
23 response: DeviceAuthResponse{},
24 want: `{"device_code":"","user_code":"","verification_uri":""}`,
25 },
26 {
27 name: "soon",
28 response: DeviceAuthResponse{
29 Expiry: time.Now().Add(100*time.Second + 999*time.Millisecond),
30 },
31 want: `{"expires_in":100,"device_code":"","user_code":"","verification_uri":""}`,
32 },
33 }
34 for _, tc := range tests {
35 t.Run(tc.name, func(t *testing.T) {
36 begin := time.Now()
37 gotBytes, err := json.Marshal(tc.response)
38 if err != nil {
39 t.Fatal(err)
40 }
41 if strings.Contains(tc.want, "expires_in") && time.Since(begin) > 999*time.Millisecond {
42 t.Skip("test ran too slowly to compare `expires_in`")
43 }
44 got := string(gotBytes)
45 if got != tc.want {
46 t.Errorf("want=%s, got=%s", tc.want, got)
47 }
48 })
49 }
50 }
51
52 func TestDeviceAuthResponseUnmarshalJson(t *testing.T) {
53 tests := []struct {
54 name string
55 data string
56 want DeviceAuthResponse
57 }{
58 {
59 name: "empty",
60 data: `{}`,
61 want: DeviceAuthResponse{},
62 },
63 {
64 name: "soon",
65 data: `{"expires_in":100}`,
66 want: DeviceAuthResponse{Expiry: time.Now().UTC().Add(100 * time.Second)},
67 },
68 }
69 for _, tc := range tests {
70 t.Run(tc.name, func(t *testing.T) {
71 begin := time.Now()
72 got := DeviceAuthResponse{}
73 err := json.Unmarshal([]byte(tc.data), &got)
74 if err != nil {
75 t.Fatal(err)
76 }
77 if !cmp.Equal(got, tc.want, cmpopts.IgnoreUnexported(DeviceAuthResponse{}), cmpopts.EquateApproxTime(time.Second+time.Since(begin))) {
78 t.Errorf("want=%#v, got=%#v", tc.want, got)
79 }
80 })
81 }
82 }
83
84 func ExampleConfig_DeviceAuth() {
85 var config Config
86 ctx := context.Background()
87 response, err := config.DeviceAuth(ctx)
88 if err != nil {
89 panic(err)
90 }
91 fmt.Printf("please enter code %s at %s\n", response.UserCode, response.VerificationURI)
92 token, err := config.DeviceAccessToken(ctx, response)
93 if err != nil {
94 panic(err)
95 }
96 fmt.Println(token)
97 }
98
View as plain text