mirror of https://github.com/gofiber/fiber.git
130 lines
2.9 KiB
Go
130 lines
2.9 KiB
Go
// 🚀 Fiber is an Express inspired web framework written in Go with 💖
|
|
// 📌 API Documentation: https://fiber.wiki
|
|
// 📝 Github Repository: https://github.com/gofiber/fiber
|
|
|
|
//nolint:bodyclose // Much easier to just ignore memory leaks in tests
|
|
package redirect
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/gofiber/fiber/v3"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_Redirect(t *testing.T) {
|
|
app := *fiber.New()
|
|
|
|
app.Use(New(Config{
|
|
Rules: map[string]string{
|
|
"/default": "google.com",
|
|
},
|
|
StatusCode: 301,
|
|
}))
|
|
app.Use(New(Config{
|
|
Rules: map[string]string{
|
|
"/default/*": "fiber.wiki",
|
|
},
|
|
StatusCode: 307,
|
|
}))
|
|
app.Use(New(Config{
|
|
Rules: map[string]string{
|
|
"/redirect/*": "$1",
|
|
},
|
|
StatusCode: 303,
|
|
}))
|
|
app.Use(New(Config{
|
|
Rules: map[string]string{
|
|
"/pattern/*": "golang.org",
|
|
},
|
|
StatusCode: 302,
|
|
}))
|
|
|
|
app.Use(New(Config{
|
|
Rules: map[string]string{
|
|
"/": "/swagger",
|
|
},
|
|
StatusCode: 301,
|
|
}))
|
|
|
|
app.Get("/api/*", func(c fiber.Ctx) error {
|
|
return c.SendString("API")
|
|
})
|
|
|
|
app.Get("/new", func(c fiber.Ctx) error {
|
|
return c.SendString("Hello, World!")
|
|
})
|
|
|
|
tests := []struct {
|
|
name string
|
|
url string
|
|
redirectTo string
|
|
statusCode int
|
|
}{
|
|
{
|
|
name: "should be returns status 302 without a wildcard",
|
|
url: "/default",
|
|
redirectTo: "google.com",
|
|
statusCode: 301,
|
|
},
|
|
{
|
|
name: "should be returns status 307 using wildcard",
|
|
url: "/default/xyz",
|
|
redirectTo: "fiber.wiki",
|
|
statusCode: 307,
|
|
},
|
|
{
|
|
name: "should be returns status 303 without set redirectTo to use the default",
|
|
url: "/redirect/github.com/gofiber/redirect",
|
|
redirectTo: "github.com/gofiber/redirect",
|
|
statusCode: 303,
|
|
},
|
|
{
|
|
name: "should return the status code default",
|
|
url: "/pattern/xyz",
|
|
redirectTo: "golang.org",
|
|
statusCode: 302,
|
|
},
|
|
{
|
|
name: "access URL without rule",
|
|
url: "/new",
|
|
statusCode: 200,
|
|
},
|
|
{
|
|
name: "redirect to swagger route",
|
|
url: "/",
|
|
redirectTo: "/swagger",
|
|
statusCode: 301,
|
|
},
|
|
{
|
|
name: "no redirect to swagger route",
|
|
url: "/api/",
|
|
statusCode: 200,
|
|
},
|
|
{
|
|
name: "no redirect to swagger route #2",
|
|
url: "/api/test",
|
|
statusCode: 200,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
req, err := http.NewRequestWithContext(context.Background(), fiber.MethodGet, tt.url, nil)
|
|
require.NoError(t, err)
|
|
req.Header.Set("Location", "github.com/gofiber/redirect")
|
|
resp, err := app.Test(req)
|
|
if err != nil {
|
|
t.Fatalf(`%s: %s`, t.Name(), err)
|
|
}
|
|
if resp.StatusCode != tt.statusCode {
|
|
t.Fatalf(`%s: StatusCode: got %v - expected %v`, t.Name(), resp.StatusCode, tt.statusCode)
|
|
}
|
|
if resp.Header.Get("Location") != tt.redirectTo {
|
|
t.Fatalf(`%s: Expecting Location: %s`, t.Name(), tt.redirectTo)
|
|
}
|
|
})
|
|
}
|
|
}
|