예제 #1
0
func oauthClient(source Source) (*github.Client, error) {
	ts := oauth2.StaticTokenSource(&oauth2.Token{
		AccessToken: source.AccessToken,
	})
	oauthClient := oauth2.NewClient(oauth2.NoContext, ts)

	apiHost := "api.github.com"
	if source.GitHubAPIURL != "" {
		uri, err := url.Parse(source.GitHubAPIURL)
		if err != nil {
			return nil, err
		}

		apiHost = uri.Host
	}

	uploadHost := "uploads.github.com"
	if source.GitHubUploadsURL != "" {
		uri, err := url.Parse(source.GitHubUploadsURL)
		if err != nil {
			return nil, err
		}

		uploadHost = uri.Host
	}

	// The google/go-github library uses the same http.Client to perform
	// requests to both github.com and the S3 download API (for downloading
	// release assets). We don't want it to user the same OAuth transport for
	// both.
	transport := statham.NewTransport(http.DefaultTransport, statham.Mapping{
		apiHost:    oauthClient.Transport,
		uploadHost: oauthClient.Transport,
	})

	httpClient := &http.Client{
		Transport: transport,
	}

	return github.NewClient(httpClient), nil
}
예제 #2
0
var _ = Describe("A per-host transport mapping", func() {
	var (
		defaultTransport *fakeRoundTripper
		orgTransport     *fakeRoundTripper
		comTransport     *fakeRoundTripper

		transport http.RoundTripper
	)

	BeforeEach(func() {
		defaultTransport = &fakeRoundTripper{}
		orgTransport = &fakeRoundTripper{}
		comTransport = &fakeRoundTripper{}

		transport = statham.NewTransport(defaultTransport, statham.Mapping{
			"example.org": orgTransport,
			"example.com": comTransport,
		})
	})

	Context("when the host doesn't match any domain in the mapping", func() {
		It("uses the default transport", func() {
			req, _ := http.NewRequest("GET", "http://example.net", nil)

			transport.RoundTrip(req)

			Expect(defaultTransport.Calls).To(Equal(1))
			Expect(orgTransport.Calls).To(Equal(0))
			Expect(comTransport.Calls).To(Equal(0))
		})
	})