oauth2/oidc_endpoint_provider_test.go

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

package oauth2

import (
	"encoding/json"
	"net/http"
	"net/http/httptest"

	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
)

var _ = Describe("GetOIDCWellKnownEndpointsFromIssuerURL", func() {
	It("calls and gets the well known data from the correct endpoint for the issuer", func() {
		var req *http.Request
		wkEndpointsResp := OIDCWellKnownEndpoints{
			AuthorizationEndpoint: "the-auth-endpoint", TokenEndpoint: "the-token-endpoint"}
		responseBytes, err := json.Marshal(wkEndpointsResp)
		Expect(err).ToNot(HaveOccurred())

		ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			req = r

			w.WriteHeader(http.StatusOK)
			w.Write(responseBytes)

		}))
		defer ts.Close()

		endpoints, err := GetOIDCWellKnownEndpointsFromIssuerURL(ts.URL)

		Expect(err).ToNot(HaveOccurred())
		Expect(*endpoints).To(Equal(wkEndpointsResp))
		Expect(req.URL.Path).To(Equal("/.well-known/openid-configuration"))
	})

	It("errors when url.Parse errors", func() {
		endpoints, err := GetOIDCWellKnownEndpointsFromIssuerURL("://")

		Expect(err).To(HaveOccurred())
		Expect(err.Error()).To(Equal(
			"could not parse issuer url to build well known endpoints: parse ://: missing protocol scheme"))
		Expect(endpoints).To(BeNil())
	})

	It("errors when the get errors", func() {
		endpoints, err := GetOIDCWellKnownEndpointsFromIssuerURL("https://")

		Expect(err).To(HaveOccurred())
		Expect(err.Error()).To(Equal(
			"could not get well known endpoints from url https://.well-known/openid-configuration: " +
				"Get https://.well-known/openid-configuration: dial tcp: lookup .well-known: no such host"))
		Expect(endpoints).To(BeNil())
	})

	It("errors when the json decoder errors", func() {
		var req *http.Request

		ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			req = r

			w.WriteHeader(http.StatusOK)
			w.Write([]byte("<"))

		}))
		defer ts.Close()

		endpoints, err := GetOIDCWellKnownEndpointsFromIssuerURL(ts.URL)

		Expect(err).To(HaveOccurred())
		Expect(err.Error()).To(Equal("could not decode json body when getting well" +
			" known endpoints: invalid character '<' looking for beginning of value"))
		Expect(endpoints).To(BeNil())
		Expect(req.URL.Path).To(Equal("/.well-known/openid-configuration"))
	})
})