28 lines
614 B
Go
28 lines
614 B
Go
package middleware
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/mikestefanello/pagoda/pkg/tests"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
echomw "github.com/labstack/echo/v4/middleware"
|
|
)
|
|
|
|
func TestLogRequestID(t *testing.T) {
|
|
ctx, _ := tests.NewContext(c.Web, "/")
|
|
_ = tests.ExecuteMiddleware(ctx, echomw.RequestID())
|
|
_ = tests.ExecuteMiddleware(ctx, LogRequestID())
|
|
|
|
var buf bytes.Buffer
|
|
ctx.Logger().SetOutput(&buf)
|
|
ctx.Logger().Info("test")
|
|
rID := ctx.Response().Header().Get(echo.HeaderXRequestID)
|
|
assert.Contains(t, buf.String(), fmt.Sprintf(`id":"%s"`, rID))
|
|
}
|