You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
112 lines
2.8 KiB
112 lines
2.8 KiB
package middleware
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/zenazn/goji/web"
|
|
)
|
|
|
|
func testOptions(r *http.Request, f func(*web.C, http.ResponseWriter, *http.Request)) *httptest.ResponseRecorder {
|
|
var c web.C
|
|
|
|
h := func(w http.ResponseWriter, r *http.Request) {
|
|
f(&c, w, r)
|
|
}
|
|
m := AutomaticOptions(&c, http.HandlerFunc(h))
|
|
w := httptest.NewRecorder()
|
|
m.ServeHTTP(w, r)
|
|
|
|
return w
|
|
}
|
|
|
|
var optionsTestEnv = map[interface{}]interface{}{
|
|
web.ValidMethodsKey: []string{
|
|
"hello",
|
|
"world",
|
|
},
|
|
}
|
|
|
|
func TestAutomaticOptions(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
// Shouldn't interfere with normal requests
|
|
r, _ := http.NewRequest("GET", "/", nil)
|
|
rr := testOptions(r,
|
|
func(c *web.C, w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte{'h', 'i'})
|
|
},
|
|
)
|
|
if rr.Code != http.StatusOK {
|
|
t.Errorf("status is %d, not 200", rr.Code)
|
|
}
|
|
if rr.Body.String() != "hi" {
|
|
t.Errorf("body was %q, should be %q", rr.Body.String(), "hi")
|
|
}
|
|
allow := rr.HeaderMap.Get("Allow")
|
|
if allow != "" {
|
|
t.Errorf("Allow header was set to %q, should be empty", allow)
|
|
}
|
|
|
|
// If we respond non-404 to an OPTIONS request, also don't interfere
|
|
r, _ = http.NewRequest("OPTIONS", "/", nil)
|
|
rr = testOptions(r,
|
|
func(c *web.C, w http.ResponseWriter, r *http.Request) {
|
|
c.Env = optionsTestEnv
|
|
w.Write([]byte{'h', 'i'})
|
|
},
|
|
)
|
|
if rr.Code != http.StatusOK {
|
|
t.Errorf("status is %d, not 200", rr.Code)
|
|
}
|
|
if rr.Body.String() != "hi" {
|
|
t.Errorf("body was %q, should be %q", rr.Body.String(), "hi")
|
|
}
|
|
allow = rr.HeaderMap.Get("Allow")
|
|
if allow != "" {
|
|
t.Errorf("Allow header was set to %q, should be empty", allow)
|
|
}
|
|
|
|
// Provide options if we 404. Make sure we nom the output bytes
|
|
r, _ = http.NewRequest("OPTIONS", "/", nil)
|
|
rr = testOptions(r,
|
|
func(c *web.C, w http.ResponseWriter, r *http.Request) {
|
|
c.Env = optionsTestEnv
|
|
w.WriteHeader(http.StatusNotFound)
|
|
w.Write([]byte{'h', 'i'})
|
|
},
|
|
)
|
|
if rr.Code != http.StatusOK {
|
|
t.Errorf("status is %d, not 200", rr.Code)
|
|
}
|
|
if rr.Body.Len() != 0 {
|
|
t.Errorf("body was %q, should be empty", rr.Body.String())
|
|
}
|
|
allow = rr.HeaderMap.Get("Allow")
|
|
correctHeaders := "hello, world, OPTIONS"
|
|
if allow != "hello, world, OPTIONS" {
|
|
t.Errorf("Allow header should be %q, was %q", correctHeaders,
|
|
allow)
|
|
}
|
|
|
|
// If we somehow 404 without giving a list of valid options, don't do
|
|
// anything
|
|
r, _ = http.NewRequest("OPTIONS", "/", nil)
|
|
rr = testOptions(r,
|
|
func(c *web.C, w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
w.Write([]byte{'h', 'i'})
|
|
},
|
|
)
|
|
if rr.Code != http.StatusNotFound {
|
|
t.Errorf("status is %d, not 404", rr.Code)
|
|
}
|
|
if rr.Body.String() != "hi" {
|
|
t.Errorf("body was %q, should be %q", rr.Body.String(), "hi")
|
|
}
|
|
allow = rr.HeaderMap.Get("Allow")
|
|
if allow != "" {
|
|
t.Errorf("Allow header was set to %q, should be empty", allow)
|
|
}
|
|
}
|