func TestRemoveOfMiddleNode(t *testing.T) {
	handler := interfaces.HandlerFuncNext(func(rw http.ResponseWriter, r *http.Request, ctx context.Context, n interfaces.HandlerFunc) {})
	tree := NewRadixTree()
	tree.Add("/web", handler)
	tree.Add("/web/news", nil)
	tree.Add("/web/article", nil)
	tree.Remove("/web")
	web, _, _, err := tree.Get("/web")
	if err != nil || web != nil {
		t.Fatalf("Deletion of middle node failed: %v, %v", err, web)
	}
}
Exemple #2
0
// Recursively converts a slice of Handlers to a single linked list of
// HandlerContainers and returns the first one. At the end of the list, there
// is another HandlerContainer whose ServeHTTP method is empty and does not
// invoke any further Handlers.
func ContainHandler(a []interfaces.Handler) *HandlerContainer {
	var next *HandlerContainer
	length := len(a)

	if length > 1 {
		next = ContainHandler(a[1:])
	} else if length == 1 {
		next = &HandlerContainer{
			interfaces.HandlerFuncNext(func(rw http.ResponseWriter, r *http.Request, ctx context.Context, n interfaces.HandlerFunc) {}),
			nil,
		}
	} else {
		return &HandlerContainer{
			interfaces.HandlerFuncNext(func(rw http.ResponseWriter, r *http.Request, ctx context.Context, n interfaces.HandlerFunc) {}),
			nil,
		}
	}

	return &HandlerContainer{
		a[0],
		next,
	}
}
func TestWrapper(t *testing.T) {
	mockRouter := &MockRouter{}
	requestHandler := NewRequestHandler(mockRouter)
	requestHandler.SetRouter(mockRouter)
	testFunc := func(rw http.ResponseWriter, r *http.Request, ctx context.Context, n interfaces.HandlerFunc) {
		rw.Write([]byte("TestFunc"))
	}
	requestHandler.AppendGlobalPostHander(interfaces.HandlerFuncNext(testFunc))
	request := &http.Request{}
	writer := httptest.NewRecorder()
	requestHandler.ServeHTTP(writer, request)
	if writer.Body.String() != "OKTestFunc" {
		t.Fatalf("Handler was not executed. Body: %s", writer.Body.String())
	}
}