fiber/middleware/skip/skip_test.go
leonklingele 167a8b5e94
🚀 Feature: Add and apply more stricter golangci-lint linting rules (#2286)
* golangci-lint: add and apply more stricter linting rules

* github: drop security workflow now that we use gosec linter inside golangci-lint

* github: use official golangci-lint CI linter

* Add editorconfig and gitattributes file
2023-01-27 09:01:37 +01:00

58 lines
1.4 KiB
Go

package skip_test
import (
"net/http/httptest"
"testing"
"github.com/gofiber/fiber/v2"
"github.com/gofiber/fiber/v2/middleware/skip"
"github.com/gofiber/fiber/v2/utils"
)
// go test -run Test_Skip
func Test_Skip(t *testing.T) {
t.Parallel()
app := fiber.New()
app.Use(skip.New(errTeapotHandler, func(*fiber.Ctx) bool { return true }))
app.Get("/", helloWorldHandler)
resp, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/", nil))
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, fiber.StatusOK, resp.StatusCode)
}
// go test -run Test_SkipFalse
func Test_SkipFalse(t *testing.T) {
t.Parallel()
app := fiber.New()
app.Use(skip.New(errTeapotHandler, func(*fiber.Ctx) bool { return false }))
app.Get("/", helloWorldHandler)
resp, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/", nil))
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, fiber.StatusTeapot, resp.StatusCode)
}
// go test -run Test_SkipNilFunc
func Test_SkipNilFunc(t *testing.T) {
t.Parallel()
app := fiber.New()
app.Use(skip.New(errTeapotHandler, nil))
app.Get("/", helloWorldHandler)
resp, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/", nil))
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, fiber.StatusTeapot, resp.StatusCode)
}
func helloWorldHandler(c *fiber.Ctx) error {
return c.SendString("Hello, World 👋!")
}
func errTeapotHandler(*fiber.Ctx) error {
return fiber.ErrTeapot
}