Function TestGetDeviceCode

internal/auth/auth_test.go:113–187

(t *testing.T)

}

func TestGetDeviceCode(t *testing.T) {
	t.Run("successfully retrieve state from response", func(t *testing.T) {
		ts := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			w.Header().Set("Content-Type", "application/json")
			_, err := io.WriteString(w, `{
				"device_code": "device-code-here",
				"user_code": "user-code-here",
				"verification_uri_complete": "verification-uri-here",
				"expires_in": 1000,
				"interval": 1
			}`)
			require.NoError(t, err)
		}))

		defer ts.Close()

		parsedURL, err := url.Parse(ts.URL)
		assert.NoError(t, err)
		u := url.URL{Scheme: "https", Host: parsedURL.Host, Path: "/oauth/device/code"}
		credentials.DeviceCodeEndpoint = u.String()

		state, err := GetDeviceCode(context.Background(), ts.Client(), []string{}, "")

		assert.NoError(t, err)
		assert.Equal(t, "device-code-here", state.DeviceCode)
		assert.Equal(t, "user-code-here", state.UserCode)
		assert.Equal(t, "verification-uri-here", state.VerificationURI)
		assert.Equal(t, 1000, state.ExpiresIn)
		assert.Equal(t, 1, state.Interval)
		assert.Equal(t, time.Duration(4000000000), state.IntervalDuration())
	})

	testCases := []struct {
		name       string
		httpStatus int
		response   string
		expect     string
	}{
		{
			name:       "handle HTTP status errors",
			httpStatus: http.StatusNotFound,
			response:   "Test response return",
			expect:     "received a 404 response: Test response return",
		},
		{
			name:       "handle bad JSON response",
			httpStatus: http.StatusOK,
			response:   "foo",
			expect:     "failed to decode the response: invalid character 'o' in literal false (expecting 'a')",
		},
	}

	for _, testCase := range testCases {
		t.Run(testCase.name, func(t *testing.T) {
			ts := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				w.WriteHeader(testCase.httpStatus)
				if testCase.response != "" {
					_, err := io.WriteString(w, testCase.response)