fix(httpx): correctly combine paths (#706)

See https://github.com/ooni/probe/issues/2010

Co-authored-by: Simone Basso <bassosimone@gmail.com>
This commit is contained in:
Yeganathan S 2022-05-09 19:32:49 +00:00 committed by GitHub
parent 36ca28d673
commit 3d81845614
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 85 additions and 5 deletions

View File

@ -10,6 +10,7 @@ import (
"io" "io"
"net/http" "net/http"
"net/url" "net/url"
"strings"
"github.com/ooni/probe-cli/v3/internal/model" "github.com/ooni/probe-cli/v3/internal/model"
"github.com/ooni/probe-cli/v3/internal/netxlite" "github.com/ooni/probe-cli/v3/internal/netxlite"
@ -70,9 +71,13 @@ const DefaultMaxBodySize = 1 << 22
// APIClient is a client configured to call a given API identified // APIClient is a client configured to call a given API identified
// by a given baseURL and using a given model.HTTPClient. // by a given baseURL and using a given model.HTTPClient.
//
// The resource path argument passed to APIClient methods is appended
// to the base URL's path for determining the full URL's path.
type APIClient interface { type APIClient interface {
// GetJSON reads the JSON resource at resourcePath and unmarshals the // GetJSON reads the JSON resource whose path is obtained concatenating
// results into output. The request is bounded by the lifetime of the // the baseURL's path with `resourcePath` and unmarshals the results
// into `output`. The request is bounded by the lifetime of the
// context passed as argument. Returns the error that occurred. // context passed as argument. Returns the error that occurred.
GetJSON(ctx context.Context, resourcePath string, output interface{}) error GetJSON(ctx context.Context, resourcePath string, output interface{}) error
@ -80,8 +85,9 @@ type APIClient interface {
GetJSONWithQuery(ctx context.Context, resourcePath string, GetJSONWithQuery(ctx context.Context, resourcePath string,
query url.Values, output interface{}) error query url.Values, output interface{}) error
// PostJSON creates a JSON subresource of the resource at resourcePath // PostJSON creates a JSON subresource of the resource whose
// using the JSON document at input and returning the result into the // path is obtained concatenating the baseURL'spath with `resourcePath` using
// the JSON document at `input` as value and returning the result into the
// JSON document at output. The request is bounded by the context's // JSON document at output. The request is bounded by the context's
// lifetime. Returns the error that occurred. // lifetime. Returns the error that occurred.
PostJSON(ctx context.Context, resourcePath string, input, output interface{}) error PostJSON(ctx context.Context, resourcePath string, input, output interface{}) error
@ -142,6 +148,23 @@ func (c *apiClient) newRequestWithJSONBody(
return request, nil return request, nil
} }
// joinURLPath appends the path of resource URL to the baseURL taking
// care of multiple forward slashes gracefully.
func (c *apiClient) joinURLPath(origPath string, newPath string) string {
// If the BaseURL path doesn't end with a slash, added one
if !strings.HasSuffix(origPath, "/") {
origPath += "/"
}
// If the resourceURL path has a leading slash, it is removed
if strings.HasPrefix(newPath, "/") {
newPath = newPath[1:]
}
return origPath + newPath
}
// newRequest creates a new request. // newRequest creates a new request.
func (c *apiClient) newRequest(ctx context.Context, method, resourcePath string, func (c *apiClient) newRequest(ctx context.Context, method, resourcePath string,
query url.Values, body io.Reader) (*http.Request, error) { query url.Values, body io.Reader) (*http.Request, error) {
@ -149,7 +172,8 @@ func (c *apiClient) newRequest(ctx context.Context, method, resourcePath string,
if err != nil { if err != nil {
return nil, err return nil, err
} }
URL.Path = resourcePath // BaseURL and resource URL is joined if they have a path
URL.Path = c.joinURLPath(URL.Path, resourcePath)
if query != nil { if query != nil {
URL.RawQuery = query.Encode() URL.RawQuery = query.Encode()
} }

View File

@ -93,6 +93,62 @@ func newAPIClient() *apiClient {
} }
} }
func TestJoinURLPath(t *testing.T) {
t.Run("empty baseURL path and slash-prefixed resource path", func(t *testing.T) {
ac := newAPIClient()
ac.BaseURL = "https://example.com"
req, err := ac.newRequest(context.Background(), "GET", "/foo", nil, nil)
if req.URL.String() != "https://example.com/foo" {
t.Fatal("unexpected result", err)
}
})
t.Run("root baseURL path and slash-prefixed resource path", func(t *testing.T) {
ac := newAPIClient()
ac.BaseURL = "https://example.com/"
req, err := ac.newRequest(context.Background(), "GET", "/foo", nil, nil)
if req.URL.String() != "https://example.com/foo" {
t.Fatal("unexpected result", err)
}
})
t.Run("empty baseURL path and empty resource path", func(t *testing.T) {
ac := newAPIClient()
ac.BaseURL = "https://example.com"
req, err := ac.newRequest(context.Background(), "GET", "", nil, nil)
if req.URL.String() != "https://example.com/" {
t.Fatal("unexpected result", err)
}
})
t.Run("non-slash-terminated baseURL path and slash-prefixed resource path", func(t *testing.T) {
ac := newAPIClient()
ac.BaseURL = "http://example.com/foo"
req, err := ac.newRequest(context.Background(), "GET", "/bar", nil, nil)
if req.URL.String() != "http://example.com/foo/bar" {
t.Fatal("unexpected result", err)
}
})
t.Run("slash-terminated baseURL path and slash-prefixed resource path", func(t *testing.T) {
ac := newAPIClient()
ac.BaseURL = "http://example.com/foo/"
req, err := ac.newRequest(context.Background(), "GET", "/bar", nil, nil)
if req.URL.String() != "http://example.com/foo/bar" {
t.Fatal("unexpected result", err)
}
})
t.Run("slash-terminated baseURL path and non-slash-prefixed resource path", func(t *testing.T) {
ac := newAPIClient()
ac.BaseURL = "http://example.com/foo/"
req, err := ac.newRequest(context.Background(), "GET", "bar", nil, nil)
if req.URL.String() != "http://example.com/foo/bar" {
t.Fatal("unexpected result", err)
}
})
}
// fakeRequest is a fake request we serialize. // fakeRequest is a fake request we serialize.
type fakeRequest struct { type fakeRequest struct {
Name string Name string