github.com/cloudfoundry-attic/cli-with-i18n@v6.32.1-0.20171002233121-7401370d3b85+incompatible/api/uaa/client_test.go (about)

     1  package uaa_test
     2  
     3  import (
     4  	"fmt"
     5  	"net/http"
     6  	"runtime"
     7  
     8  	. "code.cloudfoundry.org/cli/api/uaa"
     9  
    10  	. "github.com/onsi/ginkgo"
    11  	. "github.com/onsi/gomega"
    12  	. "github.com/onsi/gomega/ghttp"
    13  )
    14  
    15  var _ = Describe("UAA Client", func() {
    16  	var (
    17  		client *Client
    18  	)
    19  
    20  	BeforeEach(func() {
    21  		client = NewTestUAAClientAndStore()
    22  	})
    23  
    24  	Describe("Request Headers", func() {
    25  		Describe("User-Agent", func() {
    26  			var userAgent string
    27  			BeforeEach(func() {
    28  				userAgent = fmt.Sprintf("CF CLI UAA API Test/Unknown (%s; %s %s)",
    29  					runtime.Version(),
    30  					runtime.GOARCH,
    31  					runtime.GOOS,
    32  				)
    33  				server.AppendHandlers(
    34  					CombineHandlers(
    35  						VerifyRequest(http.MethodPost, "/oauth/token"),
    36  						VerifyHeaderKV("User-Agent", userAgent),
    37  						RespondWith(http.StatusOK, "{}"),
    38  					))
    39  			})
    40  
    41  			It("adds the User-Agent header to requests", func() {
    42  				_, err := client.RefreshAccessToken("")
    43  				Expect(err).ToNot(HaveOccurred())
    44  
    45  				Expect(server.ReceivedRequests()).To(HaveLen(2))
    46  			})
    47  		})
    48  
    49  		Describe("Conection", func() {
    50  			BeforeEach(func() {
    51  				server.AppendHandlers(
    52  					CombineHandlers(
    53  						VerifyRequest(http.MethodPost, "/oauth/token"),
    54  						VerifyHeaderKV("Connection", "close"),
    55  						RespondWith(http.StatusOK, "{}"),
    56  					))
    57  			})
    58  
    59  			It("forcefully closes the connection after each request", func() {
    60  				_, err := client.RefreshAccessToken("")
    61  				Expect(err).ToNot(HaveOccurred())
    62  
    63  				Expect(server.ReceivedRequests()).To(HaveLen(2))
    64  			})
    65  		})
    66  	})
    67  })