mirror of
https://github.com/gofiber/fiber.git
synced 2025-02-06 17:11:33 +00:00
845a7f8b8e
* Initial improvements * Update test * Improve RemoveEscapeChar performance * Fix lint issues * Re-add comments * Add dedicated request handlers * Fix lint issues * Add test case for app.All with custom method * Add test for custom Ctx and Request Methods * Simplify test logic * Simplify test
806 lines
20 KiB
Go
806 lines
20 KiB
Go
// ⚡️ Fiber is an Express inspired web framework written in Go with ☕️
|
|
// 📃 Github Repository: https://github.com/gofiber/fiber
|
|
// 📌 API Documentation: https://docs.gofiber.io
|
|
|
|
package fiber
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/gofiber/utils/v2"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
var routesFixture routeJSON
|
|
|
|
func init() {
|
|
dat, err := os.ReadFile("./.github/testdata/testRoutes.json")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if err := json.Unmarshal(dat, &routesFixture); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func Test_Route_Match_SameLength(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := New()
|
|
|
|
app.Get("/:param", func(c Ctx) error {
|
|
return c.SendString(c.Params("param"))
|
|
})
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/:param", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, ":param", app.getString(body))
|
|
|
|
// with param
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/test", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
body, err = io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, "test", app.getString(body))
|
|
}
|
|
|
|
func Test_Route_Match_Star(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := New()
|
|
|
|
app.Get("/*", func(c Ctx) error {
|
|
return c.SendString(c.Params("*"))
|
|
})
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/*", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, "*", app.getString(body))
|
|
|
|
// with param
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/test", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
body, err = io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, "test", app.getString(body))
|
|
|
|
// without parameter
|
|
route := Route{
|
|
star: true,
|
|
path: "/*",
|
|
routeParser: routeParser{},
|
|
}
|
|
params := [maxParams]string{}
|
|
match := route.match("", "", ¶ms)
|
|
require.True(t, match)
|
|
require.Equal(t, [maxParams]string{}, params)
|
|
|
|
// with parameter
|
|
match = route.match("/favicon.ico", "/favicon.ico", ¶ms)
|
|
require.True(t, match)
|
|
require.Equal(t, [maxParams]string{"favicon.ico"}, params)
|
|
|
|
// without parameter again
|
|
match = route.match("", "", ¶ms)
|
|
require.True(t, match)
|
|
require.Equal(t, [maxParams]string{}, params)
|
|
}
|
|
|
|
func Test_Route_Match_Root(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := New()
|
|
|
|
app.Get("/", func(c Ctx) error {
|
|
return c.SendString("root")
|
|
})
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, "root", app.getString(body))
|
|
}
|
|
|
|
func Test_Route_Match_Parser(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := New()
|
|
|
|
app.Get("/foo/:ParamName", func(c Ctx) error {
|
|
return c.SendString(c.Params("ParamName"))
|
|
})
|
|
app.Get("/Foobar/*", func(c Ctx) error {
|
|
return c.SendString(c.Params("*"))
|
|
})
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/foo/bar", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, "bar", app.getString(body))
|
|
|
|
// with star
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/Foobar/test", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
body, err = io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, "test", app.getString(body))
|
|
}
|
|
|
|
func Test_Route_Match_Middleware(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := New()
|
|
|
|
app.Use("/foo/*", func(c Ctx) error {
|
|
return c.SendString(c.Params("*"))
|
|
})
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/foo/*", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, "*", app.getString(body))
|
|
|
|
// with param
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/foo/bar/fasel", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
body, err = io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, "bar/fasel", app.getString(body))
|
|
}
|
|
|
|
func Test_Route_Match_UnescapedPath(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := New(Config{UnescapePath: true})
|
|
|
|
app.Use("/créer", func(c Ctx) error {
|
|
return c.SendString("test")
|
|
})
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/cr%C3%A9er", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, StatusOK, resp.StatusCode, "Status code")
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, "test", app.getString(body))
|
|
// without special chars
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/créer", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, StatusOK, resp.StatusCode, "Status code")
|
|
|
|
// check deactivated behavior
|
|
app.config.UnescapePath = false
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/cr%C3%A9er", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, StatusNotFound, resp.StatusCode, "Status code")
|
|
}
|
|
|
|
func Test_Route_Match_WithEscapeChar(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := New()
|
|
// static route and escaped part
|
|
app.Get("/v1/some/resource/name\\:customVerb", func(c Ctx) error {
|
|
return c.SendString("static")
|
|
})
|
|
// group route
|
|
group := app.Group("/v2/\\:firstVerb")
|
|
group.Get("/\\:customVerb", func(c Ctx) error {
|
|
return c.SendString("group")
|
|
})
|
|
// route with resource param and escaped part
|
|
app.Get("/v3/:resource/name\\:customVerb", func(c Ctx) error {
|
|
return c.SendString(c.Params("resource"))
|
|
})
|
|
|
|
// check static route
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/v1/some/resource/name:customVerb", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, StatusOK, resp.StatusCode, "Status code")
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, "static", app.getString(body))
|
|
|
|
// check group route
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/v2/:firstVerb/:customVerb", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, StatusOK, resp.StatusCode, "Status code")
|
|
|
|
body, err = io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, "group", app.getString(body))
|
|
|
|
// check param route
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/v3/awesome/name:customVerb", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, StatusOK, resp.StatusCode, "Status code")
|
|
|
|
body, err = io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, "awesome", app.getString(body))
|
|
}
|
|
|
|
func Test_Route_Match_Middleware_HasPrefix(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := New()
|
|
|
|
app.Use("/foo", func(c Ctx) error {
|
|
return c.SendString("middleware")
|
|
})
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/foo/bar", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, "middleware", app.getString(body))
|
|
}
|
|
|
|
func Test_Route_Match_Middleware_Root(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := New()
|
|
|
|
app.Use("/", func(c Ctx) error {
|
|
return c.SendString("middleware")
|
|
})
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/everything", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, "middleware", app.getString(body))
|
|
}
|
|
|
|
func Test_Router_Register_Missing_Handler(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := New()
|
|
defer func() {
|
|
if err := recover(); err != nil {
|
|
require.Equal(t, "missing handler/middleware in route: /doe\n", fmt.Sprintf("%v", err))
|
|
}
|
|
}()
|
|
app.register([]string{"USE"}, "/doe", nil, nil)
|
|
}
|
|
|
|
func Test_Ensure_Router_Interface_Implementation(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
var app any = (*App)(nil)
|
|
_, ok := app.(Router)
|
|
require.True(t, ok)
|
|
|
|
var group any = (*Group)(nil)
|
|
_, ok = group.(Router)
|
|
require.True(t, ok)
|
|
}
|
|
|
|
func Test_Router_Handler_Catch_Error(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := New()
|
|
app.config.ErrorHandler = func(_ Ctx, _ error) error {
|
|
return errors.New("fake error")
|
|
}
|
|
|
|
app.Get("/", func(_ Ctx) error {
|
|
return ErrForbidden
|
|
})
|
|
|
|
c := &fasthttp.RequestCtx{}
|
|
|
|
app.Handler()(c)
|
|
|
|
require.Equal(t, StatusInternalServerError, c.Response.Header.StatusCode())
|
|
}
|
|
|
|
func Test_Router_NotFound(t *testing.T) {
|
|
t.Parallel()
|
|
app := New()
|
|
app.Use(func(c Ctx) error {
|
|
return c.Next()
|
|
})
|
|
appHandler := app.Handler()
|
|
c := &fasthttp.RequestCtx{}
|
|
|
|
c.Request.Header.SetMethod("DELETE")
|
|
c.URI().SetPath("/this/route/does/not/exist")
|
|
|
|
appHandler(c)
|
|
|
|
require.Equal(t, 404, c.Response.StatusCode())
|
|
require.Equal(t, "Cannot DELETE /this/route/does/not/exist", string(c.Response.Body()))
|
|
}
|
|
|
|
func Test_Router_NotFound_HTML_Inject(t *testing.T) {
|
|
t.Parallel()
|
|
app := New()
|
|
app.Use(func(c Ctx) error {
|
|
return c.Next()
|
|
})
|
|
appHandler := app.Handler()
|
|
c := &fasthttp.RequestCtx{}
|
|
|
|
c.Request.Header.SetMethod("DELETE")
|
|
c.URI().SetPath("/does/not/exist<script>alert('foo');</script>")
|
|
|
|
appHandler(c)
|
|
|
|
require.Equal(t, 404, c.Response.StatusCode())
|
|
require.Equal(t, "Cannot DELETE /does/not/exist<script>alert('foo');</script>", string(c.Response.Body()))
|
|
}
|
|
|
|
func Test_App_Rebuild_Tree(t *testing.T) {
|
|
t.Parallel()
|
|
app := New()
|
|
|
|
app.Get("/test", func(c Ctx) error {
|
|
app.Get("/dynamically-defined", func(c Ctx) error {
|
|
return c.SendStatus(http.StatusOK)
|
|
})
|
|
|
|
app.RebuildTree()
|
|
|
|
return c.SendStatus(http.StatusOK)
|
|
})
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/dynamically-defined", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, http.StatusNotFound, resp.StatusCode, "Status code")
|
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/test", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, http.StatusOK, resp.StatusCode, "Status code")
|
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/dynamically-defined", nil))
|
|
require.NoError(t, err, "app.Test(req)")
|
|
require.Equal(t, http.StatusOK, resp.StatusCode, "Status code")
|
|
}
|
|
|
|
//////////////////////////////////////////////
|
|
///////////////// BENCHMARKS /////////////////
|
|
//////////////////////////////////////////////
|
|
|
|
func registerDummyRoutes(app *App) {
|
|
h := func(_ Ctx) error {
|
|
return nil
|
|
}
|
|
for _, r := range routesFixture.GithubAPI {
|
|
app.Add([]string{r.Method}, r.Path, h)
|
|
}
|
|
}
|
|
|
|
// go test -v -run=^$ -bench=Benchmark_App_MethodNotAllowed -benchmem -count=4
|
|
func Benchmark_App_MethodNotAllowed(b *testing.B) {
|
|
app := New()
|
|
h := func(c Ctx) error {
|
|
return c.SendString("Hello World!")
|
|
}
|
|
app.All("/this/is/a/", h)
|
|
app.Get("/this/is/a/dummy/route/oke", h)
|
|
appHandler := app.Handler()
|
|
c := &fasthttp.RequestCtx{}
|
|
|
|
c.Request.Header.SetMethod("DELETE")
|
|
c.URI().SetPath("/this/is/a/dummy/route/oke")
|
|
|
|
b.ResetTimer()
|
|
for n := 0; n < b.N; n++ {
|
|
appHandler(c)
|
|
}
|
|
b.StopTimer()
|
|
require.Equal(b, 405, c.Response.StatusCode())
|
|
require.Equal(b, MethodGet, string(c.Response.Header.Peek("Allow")))
|
|
require.Equal(b, utils.StatusMessage(StatusMethodNotAllowed), string(c.Response.Body()))
|
|
}
|
|
|
|
// go test -v ./... -run=^$ -bench=Benchmark_Router_NotFound -benchmem -count=4
|
|
func Benchmark_Router_NotFound(b *testing.B) {
|
|
app := New()
|
|
app.Use(func(c Ctx) error {
|
|
return c.Next()
|
|
})
|
|
registerDummyRoutes(app)
|
|
appHandler := app.Handler()
|
|
c := &fasthttp.RequestCtx{}
|
|
|
|
c.Request.Header.SetMethod("DELETE")
|
|
c.URI().SetPath("/this/route/does/not/exist")
|
|
|
|
b.ResetTimer()
|
|
for n := 0; n < b.N; n++ {
|
|
appHandler(c)
|
|
}
|
|
require.Equal(b, 404, c.Response.StatusCode())
|
|
require.Equal(b, "Cannot DELETE /this/route/does/not/exist", string(c.Response.Body()))
|
|
}
|
|
|
|
// go test -v ./... -run=^$ -bench=Benchmark_Router_Handler -benchmem -count=4
|
|
func Benchmark_Router_Handler(b *testing.B) {
|
|
app := New()
|
|
registerDummyRoutes(app)
|
|
appHandler := app.Handler()
|
|
|
|
c := &fasthttp.RequestCtx{}
|
|
|
|
c.Request.Header.SetMethod("DELETE")
|
|
c.URI().SetPath("/user/keys/1337")
|
|
|
|
b.ResetTimer()
|
|
|
|
for n := 0; n < b.N; n++ {
|
|
appHandler(c)
|
|
}
|
|
}
|
|
|
|
func Benchmark_Router_Handler_Strict_Case(b *testing.B) {
|
|
app := New(Config{
|
|
StrictRouting: true,
|
|
CaseSensitive: true,
|
|
})
|
|
registerDummyRoutes(app)
|
|
appHandler := app.Handler()
|
|
|
|
c := &fasthttp.RequestCtx{}
|
|
|
|
c.Request.Header.SetMethod("DELETE")
|
|
c.URI().SetPath("/user/keys/1337")
|
|
|
|
b.ResetTimer()
|
|
|
|
for n := 0; n < b.N; n++ {
|
|
appHandler(c)
|
|
}
|
|
}
|
|
|
|
// go test -v ./... -run=^$ -bench=Benchmark_Router_Chain -benchmem -count=4
|
|
func Benchmark_Router_Chain(b *testing.B) {
|
|
app := New()
|
|
handler := func(c Ctx) error {
|
|
return c.Next()
|
|
}
|
|
app.Get("/", handler, handler, handler, handler, handler, handler)
|
|
|
|
appHandler := app.Handler()
|
|
|
|
c := &fasthttp.RequestCtx{}
|
|
|
|
c.Request.Header.SetMethod(MethodGet)
|
|
c.URI().SetPath("/")
|
|
b.ResetTimer()
|
|
for n := 0; n < b.N; n++ {
|
|
appHandler(c)
|
|
}
|
|
}
|
|
|
|
// go test -v ./... -run=^$ -bench=Benchmark_Router_WithCompression -benchmem -count=4
|
|
func Benchmark_Router_WithCompression(b *testing.B) {
|
|
app := New()
|
|
handler := func(c Ctx) error {
|
|
return c.Next()
|
|
}
|
|
app.Get("/", handler)
|
|
app.Get("/", handler)
|
|
app.Get("/", handler)
|
|
app.Get("/", handler)
|
|
app.Get("/", handler)
|
|
app.Get("/", handler)
|
|
|
|
appHandler := app.Handler()
|
|
c := &fasthttp.RequestCtx{}
|
|
|
|
c.Request.Header.SetMethod(MethodGet)
|
|
c.URI().SetPath("/")
|
|
b.ResetTimer()
|
|
for n := 0; n < b.N; n++ {
|
|
appHandler(c)
|
|
}
|
|
}
|
|
|
|
// go test -run=^$ -bench=Benchmark_Startup_Process -benchmem -count=9
|
|
func Benchmark_Startup_Process(b *testing.B) {
|
|
for n := 0; n < b.N; n++ {
|
|
app := New()
|
|
registerDummyRoutes(app)
|
|
app.startupProcess()
|
|
}
|
|
}
|
|
|
|
// go test -v ./... -run=^$ -bench=Benchmark_Router_Next -benchmem -count=4
|
|
func Benchmark_Router_Next(b *testing.B) {
|
|
app := New()
|
|
registerDummyRoutes(app)
|
|
app.startupProcess()
|
|
|
|
request := &fasthttp.RequestCtx{}
|
|
|
|
request.Request.Header.SetMethod("DELETE")
|
|
request.URI().SetPath("/user/keys/1337")
|
|
var res bool
|
|
var err error
|
|
|
|
c := app.AcquireCtx(request).(*DefaultCtx) //nolint:errcheck, forcetypeassert // not needed
|
|
|
|
b.ResetTimer()
|
|
for n := 0; n < b.N; n++ {
|
|
c.indexRoute = -1
|
|
res, err = app.next(c)
|
|
}
|
|
require.NoError(b, err)
|
|
require.True(b, res)
|
|
require.Equal(b, 4, c.indexRoute)
|
|
}
|
|
|
|
// go test -v ./... -run=^$ -bench=Benchmark_Router_Next_Default -benchmem -count=4
|
|
func Benchmark_Router_Next_Default(b *testing.B) {
|
|
app := New()
|
|
app.Get("/", func(_ Ctx) error {
|
|
return nil
|
|
})
|
|
|
|
h := app.Handler()
|
|
|
|
fctx := &fasthttp.RequestCtx{}
|
|
fctx.Request.Header.SetMethod(MethodGet)
|
|
fctx.Request.SetRequestURI("/")
|
|
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
|
|
for n := 0; n < b.N; n++ {
|
|
h(fctx)
|
|
}
|
|
}
|
|
|
|
// go test -benchmem -run=^$ -bench ^Benchmark_Router_Next_Default_Parallel$ github.com/gofiber/fiber/v3 -count=1
|
|
func Benchmark_Router_Next_Default_Parallel(b *testing.B) {
|
|
app := New()
|
|
app.Get("/", func(_ Ctx) error {
|
|
return nil
|
|
})
|
|
|
|
h := app.Handler()
|
|
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
fctx := &fasthttp.RequestCtx{}
|
|
fctx.Request.Header.SetMethod(MethodGet)
|
|
fctx.Request.SetRequestURI("/")
|
|
|
|
for pb.Next() {
|
|
h(fctx)
|
|
}
|
|
})
|
|
}
|
|
|
|
// go test -v ./... -run=^$ -bench=Benchmark_Route_Match -benchmem -count=4
|
|
func Benchmark_Route_Match(b *testing.B) {
|
|
var match bool
|
|
var params [maxParams]string
|
|
|
|
parsed := parseRoute("/user/keys/:id")
|
|
route := &Route{
|
|
use: false,
|
|
root: false,
|
|
star: false,
|
|
routeParser: parsed,
|
|
Params: parsed.params,
|
|
path: "/user/keys/:id",
|
|
|
|
Path: "/user/keys/:id",
|
|
Method: "DELETE",
|
|
}
|
|
route.Handlers = append(route.Handlers, func(_ Ctx) error {
|
|
return nil
|
|
})
|
|
b.ResetTimer()
|
|
for n := 0; n < b.N; n++ {
|
|
match = route.match("/user/keys/1337", "/user/keys/1337", ¶ms)
|
|
}
|
|
|
|
require.True(b, match)
|
|
require.Equal(b, []string{"1337"}, params[0:len(parsed.params)])
|
|
}
|
|
|
|
// go test -v ./... -run=^$ -bench=Benchmark_Route_Match_Star -benchmem -count=4
|
|
func Benchmark_Route_Match_Star(b *testing.B) {
|
|
var match bool
|
|
var params [maxParams]string
|
|
|
|
parsed := parseRoute("/*")
|
|
route := &Route{
|
|
use: false,
|
|
root: false,
|
|
star: true,
|
|
routeParser: parsed,
|
|
Params: parsed.params,
|
|
path: "/user/keys/bla",
|
|
|
|
Path: "/user/keys/bla",
|
|
Method: "DELETE",
|
|
}
|
|
route.Handlers = append(route.Handlers, func(_ Ctx) error {
|
|
return nil
|
|
})
|
|
b.ResetTimer()
|
|
|
|
for n := 0; n < b.N; n++ {
|
|
match = route.match("/user/keys/bla", "/user/keys/bla", ¶ms)
|
|
}
|
|
|
|
require.True(b, match)
|
|
require.Equal(b, []string{"user/keys/bla"}, params[0:len(parsed.params)])
|
|
}
|
|
|
|
// go test -v ./... -run=^$ -bench=Benchmark_Route_Match_Root -benchmem -count=4
|
|
func Benchmark_Route_Match_Root(b *testing.B) {
|
|
var match bool
|
|
var params [maxParams]string
|
|
|
|
parsed := parseRoute("/")
|
|
route := &Route{
|
|
use: false,
|
|
root: true,
|
|
star: false,
|
|
path: "/",
|
|
routeParser: parsed,
|
|
Params: parsed.params,
|
|
|
|
Path: "/",
|
|
Method: "DELETE",
|
|
}
|
|
route.Handlers = append(route.Handlers, func(_ Ctx) error {
|
|
return nil
|
|
})
|
|
|
|
b.ResetTimer()
|
|
|
|
for n := 0; n < b.N; n++ {
|
|
match = route.match("/", "/", ¶ms)
|
|
}
|
|
|
|
require.True(b, match)
|
|
require.Equal(b, []string{}, params[0:len(parsed.params)])
|
|
}
|
|
|
|
// go test -v ./... -run=^$ -bench=Benchmark_Router_Handler_CaseSensitive -benchmem -count=4
|
|
func Benchmark_Router_Handler_CaseSensitive(b *testing.B) {
|
|
app := New()
|
|
app.config.CaseSensitive = true
|
|
registerDummyRoutes(app)
|
|
appHandler := app.Handler()
|
|
|
|
c := &fasthttp.RequestCtx{}
|
|
|
|
c.Request.Header.SetMethod("DELETE")
|
|
c.URI().SetPath("/user/keys/1337")
|
|
|
|
b.ResetTimer()
|
|
|
|
for n := 0; n < b.N; n++ {
|
|
appHandler(c)
|
|
}
|
|
}
|
|
|
|
// go test -v ./... -run=^$ -bench=Benchmark_Router_Handler_Unescape -benchmem -count=4
|
|
func Benchmark_Router_Handler_Unescape(b *testing.B) {
|
|
app := New()
|
|
app.config.UnescapePath = true
|
|
registerDummyRoutes(app)
|
|
app.Delete("/créer", func(_ Ctx) error {
|
|
return nil
|
|
})
|
|
|
|
appHandler := app.Handler()
|
|
|
|
c := &fasthttp.RequestCtx{}
|
|
|
|
c.Request.Header.SetMethod(MethodDelete)
|
|
c.URI().SetPath("/cr%C3%A9er")
|
|
|
|
b.ResetTimer()
|
|
|
|
for n := 0; n < b.N; n++ {
|
|
c.URI().SetPath("/cr%C3%A9er")
|
|
appHandler(c)
|
|
}
|
|
}
|
|
|
|
// go test -run=^$ -bench=Benchmark_Router_Handler_StrictRouting -benchmem -count=4
|
|
func Benchmark_Router_Handler_StrictRouting(b *testing.B) {
|
|
app := New()
|
|
app.config.CaseSensitive = true
|
|
registerDummyRoutes(app)
|
|
appHandler := app.Handler()
|
|
|
|
c := &fasthttp.RequestCtx{}
|
|
|
|
c.Request.Header.SetMethod("DELETE")
|
|
c.URI().SetPath("/user/keys/1337")
|
|
|
|
b.ResetTimer()
|
|
|
|
for n := 0; n < b.N; n++ {
|
|
appHandler(c)
|
|
}
|
|
}
|
|
|
|
// go test -run=^$ -bench=Benchmark_Router_Github_API -benchmem -count=16
|
|
func Benchmark_Router_Github_API(b *testing.B) {
|
|
app := New()
|
|
registerDummyRoutes(app)
|
|
app.startupProcess()
|
|
|
|
c := &fasthttp.RequestCtx{}
|
|
var match bool
|
|
var err error
|
|
|
|
b.ResetTimer()
|
|
|
|
for i := range routesFixture.TestRoutes {
|
|
c.Request.Header.SetMethod(routesFixture.TestRoutes[i].Method)
|
|
for n := 0; n < b.N; n++ {
|
|
c.URI().SetPath(routesFixture.TestRoutes[i].Path)
|
|
|
|
ctx := app.AcquireCtx(c).(*DefaultCtx) //nolint:errcheck, forcetypeassert // not needed
|
|
|
|
match, err = app.next(ctx)
|
|
app.ReleaseCtx(ctx)
|
|
}
|
|
|
|
require.NoError(b, err)
|
|
require.True(b, match)
|
|
}
|
|
}
|
|
|
|
type testRoute struct {
|
|
Method string `json:"method"`
|
|
Path string `json:"path"`
|
|
}
|
|
|
|
type routeJSON struct {
|
|
TestRoutes []testRoute `json:"test_routes"`
|
|
GithubAPI []testRoute `json:"github_api"`
|
|
}
|