mirror of https://github.com/gofiber/fiber.git
102 lines
2.0 KiB
Go
102 lines
2.0 KiB
Go
package requestid
|
|
|
|
import (
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gofiber/fiber/v3"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// go test -run Test_RequestID
|
|
func Test_RequestID(t *testing.T) {
|
|
t.Parallel()
|
|
app := fiber.New()
|
|
|
|
app.Use(New())
|
|
|
|
app.Get("/", func(c fiber.Ctx) error {
|
|
return c.SendString("Hello, World 👋!")
|
|
})
|
|
|
|
resp, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/", nil))
|
|
require.NoError(t, err)
|
|
require.Equal(t, fiber.StatusOK, resp.StatusCode)
|
|
|
|
reqid := resp.Header.Get(fiber.HeaderXRequestID)
|
|
require.Len(t, reqid, 36)
|
|
|
|
req := httptest.NewRequest(fiber.MethodGet, "/", nil)
|
|
req.Header.Add(fiber.HeaderXRequestID, reqid)
|
|
|
|
resp, err = app.Test(req)
|
|
require.NoError(t, err)
|
|
require.Equal(t, fiber.StatusOK, resp.StatusCode)
|
|
require.Equal(t, reqid, resp.Header.Get(fiber.HeaderXRequestID))
|
|
}
|
|
|
|
// go test -run Test_RequestID_Next
|
|
func Test_RequestID_Next(t *testing.T) {
|
|
t.Parallel()
|
|
app := fiber.New()
|
|
app.Use(New(Config{
|
|
Next: func(_ fiber.Ctx) bool {
|
|
return true
|
|
},
|
|
}))
|
|
|
|
resp, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/", nil))
|
|
require.NoError(t, err)
|
|
require.Empty(t, resp.Header.Get(fiber.HeaderXRequestID))
|
|
require.Equal(t, fiber.StatusNotFound, resp.StatusCode)
|
|
}
|
|
|
|
// go test -run Test_RequestID_FromContext
|
|
func Test_RequestID_FromContext(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
reqID := "ThisIsARequestId"
|
|
|
|
type args struct {
|
|
inputFunc func(c fiber.Ctx) any
|
|
}
|
|
|
|
tests := []struct {
|
|
args args
|
|
name string
|
|
}{
|
|
{
|
|
name: "From fiber.Ctx",
|
|
args: args{
|
|
inputFunc: func(c fiber.Ctx) any {
|
|
return c
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
app := fiber.New()
|
|
app.Use(New(Config{
|
|
Generator: func() string {
|
|
return reqID
|
|
},
|
|
}))
|
|
|
|
var ctxVal string
|
|
|
|
app.Use(func(c fiber.Ctx) error {
|
|
ctxVal = FromContext(tt.args.inputFunc(c))
|
|
return c.Next()
|
|
})
|
|
|
|
_, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/", nil))
|
|
require.NoError(t, err)
|
|
require.Equal(t, reqID, ctxVal)
|
|
})
|
|
}
|
|
}
|