mirror of https://github.com/gofiber/fiber.git
55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
package skip_test
|
|
|
|
import (
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gofiber/fiber/v3"
|
|
"github.com/gofiber/fiber/v3/middleware/skip"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// go test -run Test_Skip
|
|
func Test_Skip(t *testing.T) {
|
|
app := fiber.New()
|
|
|
|
app.Use(skip.New(errTeapotHandler, func(fiber.Ctx) bool { return true }))
|
|
app.Get("/", helloWorldHandler)
|
|
|
|
resp, err := app.Test(httptest.NewRequest("GET", "/", nil))
|
|
require.NoError(t, err)
|
|
require.Equal(t, fiber.StatusOK, resp.StatusCode)
|
|
}
|
|
|
|
// go test -run Test_SkipFalse
|
|
func Test_SkipFalse(t *testing.T) {
|
|
app := fiber.New()
|
|
|
|
app.Use(skip.New(errTeapotHandler, func(fiber.Ctx) bool { return false }))
|
|
app.Get("/", helloWorldHandler)
|
|
|
|
resp, err := app.Test(httptest.NewRequest("GET", "/", nil))
|
|
require.NoError(t, err)
|
|
require.Equal(t, fiber.StatusTeapot, resp.StatusCode)
|
|
}
|
|
|
|
// go test -run Test_SkipNilFunc
|
|
func Test_SkipNilFunc(t *testing.T) {
|
|
app := fiber.New()
|
|
|
|
app.Use(skip.New(errTeapotHandler, nil))
|
|
app.Get("/", helloWorldHandler)
|
|
|
|
resp, err := app.Test(httptest.NewRequest("GET", "/", nil))
|
|
require.NoError(t, err)
|
|
require.Equal(t, fiber.StatusTeapot, resp.StatusCode)
|
|
}
|
|
|
|
func helloWorldHandler(c fiber.Ctx) error {
|
|
return c.SendString("Hello, World 👋!")
|
|
}
|
|
|
|
func errTeapotHandler(fiber.Ctx) error {
|
|
return fiber.ErrTeapot
|
|
}
|