mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
main: remove query_params*.go
This commit is contained in:
parent
3653287c97
commit
08b370ebe4
@ -1,36 +0,0 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func NewLowerQueryParamsHandler(hdlr http.Handler) *LowerQueryParamsHandler {
|
||||
return &LowerQueryParamsHandler{hdlr}
|
||||
}
|
||||
|
||||
type LowerQueryParamsHandler struct {
|
||||
Handler http.Handler
|
||||
}
|
||||
|
||||
func (h *LowerQueryParamsHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
|
||||
err := req.ParseForm()
|
||||
if err == nil {
|
||||
lowerBoolQueryParams(req)
|
||||
}
|
||||
h.Handler.ServeHTTP(w, req)
|
||||
}
|
||||
|
||||
func lowerBoolQueryParams(req *http.Request) {
|
||||
form := req.Form
|
||||
for key, vals := range form {
|
||||
for i, val := range vals {
|
||||
lowered := strings.ToLower(val)
|
||||
if lowered == "true" || lowered == "false" {
|
||||
req.Form[key][i] = lowered
|
||||
} else {
|
||||
req.Form[key][i] = val
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
@ -1,46 +0,0 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"testing"
|
||||
)
|
||||
|
||||
type NilResponseWriter struct{}
|
||||
|
||||
func (w NilResponseWriter) Header() http.Header {
|
||||
return http.Header{}
|
||||
}
|
||||
|
||||
func (w NilResponseWriter) Write(data []byte) (int, error) {
|
||||
return 0, nil
|
||||
}
|
||||
|
||||
func (w NilResponseWriter) WriteHeader(code int) {
|
||||
return
|
||||
}
|
||||
|
||||
type FunctionHandler struct {
|
||||
f func(*http.Request)
|
||||
}
|
||||
|
||||
func (h FunctionHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||
h.f(r)
|
||||
}
|
||||
|
||||
func TestQueryParamsLowered(t *testing.T) {
|
||||
assertFunc := func(req *http.Request) {
|
||||
if len(req.Form["One"]) != 1 || req.Form["One"][0] != "true" {
|
||||
t.Errorf("Unexpected value for One: %s", req.Form["One"])
|
||||
} else if len(req.Form["TWO"]) != 1 || req.Form["TWO"][0] != "false" {
|
||||
t.Errorf("Unexpected value for TWO")
|
||||
} else if len(req.Form["three"]) != 2 || req.Form["three"][0] != "true" || req.Form["three"][1] != "false" {
|
||||
t.Errorf("Unexpected value for three")
|
||||
}
|
||||
}
|
||||
assertHdlr := FunctionHandler{assertFunc}
|
||||
hdlr := NewLowerQueryParamsHandler(assertHdlr)
|
||||
respWriter := NilResponseWriter{}
|
||||
|
||||
req, _ := http.NewRequest("GET", "http://example.com?One=TRUE&TWO=False&three=true&three=FALSE", nil)
|
||||
hdlr.ServeHTTP(respWriter, req)
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user