Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions hyperdrive_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ package hyperdrive

import (
"net/http"
"net/http/httptest"
"testing"

"github.com/stretchr/testify/suite"
Expand All @@ -14,6 +15,7 @@ type HyperdriveTestSuite struct {
TestHandler http.Handler
TestRoot *RootResource
TestEndpointResource EndpointResource
TestGetRequest *http.Request
}

func (suite *HyperdriveTestSuite) SetupTest() {
Expand All @@ -22,6 +24,7 @@ func (suite *HyperdriveTestSuite) SetupTest() {
suite.TestHandler = NewMethodHandler(suite.TestEndpoint)
suite.TestRoot = NewRootResource(suite.TestAPI)
suite.TestEndpointResource = NewEndpointResource(suite.TestEndpoint)
suite.TestGetRequest = httptest.NewRequest("GET", "/test/2?id=1&a=b", nil)
}

func (suite *HyperdriveTestSuite) TestNewAPI() {
Expand Down
59 changes: 59 additions & 0 deletions params.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,59 @@
package hyperdrive

import (
"net/http"
"net/url"

"github.com/gorilla/mux"
)

// QueryParams extracts the values from the request QueryString. It returns
// a url.Values object (essentially map[string][]string). If the
// request method is not GET, an empty url.Values is returned.
func QueryParams(r *http.Request) url.Values {
if r.Method == "GET" {
r.ParseForm()
return r.Form
}
return url.Values{}
}

// BodyParams deserializes the input, and extracts the values from the request
// body. It returns a url.Values object (essentially map[string][]string). If
// the request method is GET, an empty url.Values is returned.
func BodyParams(r *http.Request) url.Values {
if r.Method != "GET" {
return url.Values{}
}
return url.Values{}
}

// PathParams extracts the values from the request path which match named
// params in the route. They are returned as url.Values for consistincey
// with http.Request.Form's behaviour.
func PathParams(r *http.Request) url.Values {
var params = url.Values{}
for k, v := range mux.Vars(r) {
params.Add(k, v)
}
return params
}

// Params extracts the param values from all sources: query, body, and path -- in
// that order. Each subsequent source will overwrite values with the same key, to
// ensure API client intent is maintained in a consistent way.
func Params(r *http.Request) url.Values {
var params = QueryParams(r)

for k, values := range BodyParams(r) {
for _, v := range values {
params.Set(k, v)
}
}

for k, v := range mux.Vars(r) {
params.Set(k, v)
}

return params
}
52 changes: 52 additions & 0 deletions params_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
package hyperdrive

import (
"net/http"
"net/http/httptest"
"net/url"
)

func (suite *HyperdriveTestSuite) TestQueryParamsGet() {
suite.IsType(url.Values{}, QueryParams(suite.TestGetRequest), "expects an instance of url.Values")
}

func (suite *HyperdriveTestSuite) TestQueryParamsGetValues() {
suite.Equal(url.Values{"id": []string{"1"}, "a": []string{"b"}}, QueryParams(suite.TestGetRequest), "returns populated url.Values")
}

func (suite *HyperdriveTestSuite) TestBodyParamsGet() {
suite.IsType(url.Values{}, BodyParams(suite.TestGetRequest), "expects an instance of url.Values")
}

func (suite *HyperdriveTestSuite) TestBodyParamsGetValues() {
suite.Equal(url.Values{}, BodyParams(suite.TestGetRequest), "returns populated url.Values")
}

func (suite *HyperdriveTestSuite) TestPathParamsGet() {
suite.TestAPI.Router.Handle("/test/{id}", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
suite.IsType(url.Values{}, PathParams(suite.TestGetRequest), "expects an instance of url.Values")
}))
suite.TestAPI.Router.ServeHTTP(httptest.NewRecorder(), suite.TestGetRequest)
}

func (suite *HyperdriveTestSuite) TestPathParamsGetValues() {
suite.TestAPI.Router.Handle("/test/{id}", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
suite.Equal(url.Values{"id": []string{"2"}}, PathParams(r), "returns populated url.Values")
}))
suite.TestAPI.Router.ServeHTTP(httptest.NewRecorder(), suite.TestGetRequest)
}

func (suite *HyperdriveTestSuite) TestParamsGet() {
suite.TestAPI.Router.Handle("/test/{id}", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
suite.IsType(url.Values{}, Params(suite.TestGetRequest), "expects an instance of url.Values")
}))
suite.TestAPI.Router.ServeHTTP(httptest.NewRecorder(), suite.TestGetRequest)

}

func (suite *HyperdriveTestSuite) TestParamsGetValues() {
suite.TestAPI.Router.Handle("/test/{id}", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
suite.Equal(url.Values{"id": []string{"1"}, "a": []string{"b"}}, Params(suite.TestGetRequest), "returns populated url.Values")
}))
suite.TestAPI.Router.ServeHTTP(httptest.NewRecorder(), suite.TestGetRequest)
}