mirror of
https://github.com/gofiber/fiber.git
synced 2025-02-23 00:03:43 +00:00
81 lines
2.2 KiB
Go
81 lines
2.2 KiB
Go
package middleware
|
|
|
|
import (
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gofiber/fiber"
|
|
"github.com/gofiber/utils"
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
// go test -run Test_Middleware_Favicon
|
|
func Test_Middleware_Favicon(t *testing.T) {
|
|
app := fiber.New()
|
|
|
|
app.Use(Favicon())
|
|
|
|
app.Get("/", func(c *fiber.Ctx) {})
|
|
|
|
// Skip Favicon middleware
|
|
resp, err := app.Test(httptest.NewRequest("GET", "/", nil))
|
|
utils.AssertEqual(t, nil, err, "app.Test(req)")
|
|
utils.AssertEqual(t, fiber.StatusOK, resp.StatusCode, "Status code")
|
|
|
|
resp, err = app.Test(httptest.NewRequest("GET", "/favicon.ico", nil))
|
|
utils.AssertEqual(t, nil, err, "app.Test(req)")
|
|
utils.AssertEqual(t, fiber.StatusNoContent, resp.StatusCode, "Status code")
|
|
|
|
resp, err = app.Test(httptest.NewRequest("OPTIONS", "/favicon.ico", nil))
|
|
utils.AssertEqual(t, nil, err, "app.Test(req)")
|
|
utils.AssertEqual(t, fiber.StatusOK, resp.StatusCode, "Status code")
|
|
|
|
resp, err = app.Test(httptest.NewRequest("PUT", "/favicon.ico", nil))
|
|
utils.AssertEqual(t, nil, err, "app.Test(req)")
|
|
utils.AssertEqual(t, fiber.StatusMethodNotAllowed, resp.StatusCode, "Status code")
|
|
utils.AssertEqual(t, "GET, HEAD, OPTIONS", resp.Header.Get(fiber.HeaderAllow))
|
|
}
|
|
|
|
// go test -run Test_Middleware_Favicon_Not_Found
|
|
func Test_Middleware_Favicon_Not_Found(t *testing.T) {
|
|
defer func() {
|
|
if err := recover(); err == nil {
|
|
t.Fatal("should cache panic")
|
|
}
|
|
}()
|
|
|
|
fiber.New().Use(Favicon("non-exist.ico"))
|
|
}
|
|
|
|
// go test -run Test_Middleware_Favicon_Found
|
|
func Test_Middleware_Favicon_Found(t *testing.T) {
|
|
app := fiber.New()
|
|
|
|
app.Use(Favicon("./testdata/favicon.ico"))
|
|
|
|
app.Get("/", func(c *fiber.Ctx) {})
|
|
|
|
resp, err := app.Test(httptest.NewRequest("GET", "/favicon.ico", nil))
|
|
|
|
utils.AssertEqual(t, nil, err, "app.Test(req)")
|
|
utils.AssertEqual(t, fiber.StatusOK, resp.StatusCode, "Status code")
|
|
utils.AssertEqual(t, "image/x-icon", resp.Header.Get(fiber.HeaderContentType))
|
|
}
|
|
|
|
// go test -v -run=^$ -bench=Benchmark_Middleware_Favicon -benchmem -count=4
|
|
func Benchmark_Middleware_Favicon(b *testing.B) {
|
|
app := fiber.New()
|
|
app.Use(Favicon())
|
|
app.Get("/", func(c *fiber.Ctx) {})
|
|
handler := app.Handler()
|
|
|
|
c := &fasthttp.RequestCtx{}
|
|
c.Request.SetRequestURI("/")
|
|
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
for n := 0; n < b.N; n++ {
|
|
handler(c)
|
|
}
|
|
}
|