Function TestExchangeCodeForToken

internal/auth/authutil/exchange_test.go:14–78

(t *testing.T)

)

func TestExchangeCodeForToken(t *testing.T) {
	t.Run("Successfully call token endpoint", 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, `{
				"access_token": "access-token-here",
				"id_token": "id-token-here",
				"token_type": "token-type-here",
				"expires_in": 1000
			}`)
			require.NoError(t, err)
		}))

		defer ts.Close()
		parsedURL, err := url.Parse(ts.URL)
		assert.NoError(t, err)

		token, err := ExchangeCodeForToken(ts.Client(), parsedURL.Host, "some-client-id", "some-client-secret", "some-code", "http://localhost:8484")

		assert.NoError(t, err)
		assert.Equal(t, "access-token-here", token.AccessToken)
		assert.Equal(t, "id-token-here", token.IDToken)
		assert.Equal(t, "token-type-here", token.TokenType)
		assert.Equal(t, int64(1000), token.ExpiresIn)
	})

	testCases := []struct {
		name       string
		expect     string
		httpStatus int
		response   string
	}{
		{
			name:       "Bad status code",
			expect:     "unable to exchange code for token: 404 Not Found",
			httpStatus: http.StatusNotFound,
		},
		{
			name:       "Malformed JSON",
			expect:     "cannot decode response: unexpected EOF",
			httpStatus: http.StatusOK,
			response:   `{ "foo": "bar" `,
		},
	}

	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)
					require.NoError(t, err)
				}
			}))

			defer ts.Close()
			parsedURL, err := url.Parse(ts.URL)
			assert.NoError(t, err)