mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
8fc4bdcf97
* WIP: add k8s grouper - This shares a great deal of behaviour with allGrouper. Once it's tested, refactor that to share as much as possible - Glide hell. Checked in the yaml and lock files but a glide i -v will be required to bring vendor/ up-to-date. Will address once this is ready. * Update README. Make the watch tracking work. (To follow: add the junk that was pulled in via the glide update.) * Vendor updates. * go fmt * Use the allGrouper with a k8s-backed DBStore instead. This is much tidier :-) * Fix up go vet
84 lines
2.2 KiB
Go
84 lines
2.2 KiB
Go
package restful
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
// go test -v -test.run TestContainer_computeAllowedMethods ...restful
|
|
func TestContainer_computeAllowedMethods(t *testing.T) {
|
|
wc := NewContainer()
|
|
ws1 := new(WebService).Path("/users")
|
|
ws1.Route(ws1.GET("{i}").To(dummy))
|
|
ws1.Route(ws1.POST("{i}").To(dummy))
|
|
wc.Add(ws1)
|
|
httpRequest, _ := http.NewRequest("GET", "http://api.his.com/users/1", nil)
|
|
rreq := Request{Request: httpRequest}
|
|
m := wc.computeAllowedMethods(&rreq)
|
|
if len(m) != 2 {
|
|
t.Errorf("got %d expected 2 methods, %v", len(m), m)
|
|
}
|
|
}
|
|
|
|
func TestContainer_HandleWithFilter(t *testing.T) {
|
|
prefilterCalled := false
|
|
postfilterCalled := false
|
|
httpHandlerCalled := false
|
|
|
|
wc := NewContainer()
|
|
wc.Filter(func(request *Request, response *Response, chain *FilterChain) {
|
|
prefilterCalled = true
|
|
chain.ProcessFilter(request, response)
|
|
})
|
|
wc.HandleWithFilter("/", http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
|
|
httpHandlerCalled = true
|
|
w.Write([]byte("ok"))
|
|
}))
|
|
wc.Filter(func(request *Request, response *Response, chain *FilterChain) {
|
|
postfilterCalled = true
|
|
chain.ProcessFilter(request, response)
|
|
})
|
|
|
|
recorder := httptest.NewRecorder()
|
|
request, _ := http.NewRequest("GET", "/", nil)
|
|
wc.ServeHTTP(recorder, request)
|
|
if recorder.Code != http.StatusOK {
|
|
t.Errorf("unexpected code %d", recorder.Code)
|
|
}
|
|
if recorder.Body.String() != "ok" {
|
|
t.Errorf("unexpected body %s", recorder.Body.String())
|
|
}
|
|
if !prefilterCalled {
|
|
t.Errorf("filter added before calling HandleWithFilter wasn't called")
|
|
}
|
|
if !postfilterCalled {
|
|
t.Errorf("filter added after calling HandleWithFilter wasn't called")
|
|
}
|
|
if !httpHandlerCalled {
|
|
t.Errorf("handler added by calling HandleWithFilter wasn't called")
|
|
}
|
|
}
|
|
|
|
func TestContainerAddAndRemove(t *testing.T) {
|
|
ws1 := new(WebService).Path("/")
|
|
ws2 := new(WebService).Path("/users")
|
|
wc := NewContainer()
|
|
wc.Add(ws1)
|
|
wc.Add(ws2)
|
|
wc.Remove(ws2)
|
|
if len(wc.webServices) != 1 {
|
|
t.Errorf("expected one webservices")
|
|
}
|
|
if !wc.isRegisteredOnRoot {
|
|
t.Errorf("expected on root registered")
|
|
}
|
|
wc.Remove(ws1)
|
|
if len(wc.webServices) > 0 {
|
|
t.Errorf("expected zero webservices")
|
|
}
|
|
if wc.isRegisteredOnRoot {
|
|
t.Errorf("expected not on root registered")
|
|
}
|
|
}
|