mirror of
https://github.com/kataras/iris.git
synced 2026-01-08 20:41:57 +00:00
add the stale release
This commit is contained in:
@@ -5,7 +5,7 @@ import (
|
||||
"encoding/hex"
|
||||
"net/http/httputil"
|
||||
|
||||
"github.com/kataras/iris/v12/context"
|
||||
"github.com/kataras/iris/context"
|
||||
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
@@ -1,63 +0,0 @@
|
||||
package requestid_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/kataras/iris/v12"
|
||||
"github.com/kataras/iris/v12/context"
|
||||
"github.com/kataras/iris/v12/httptest"
|
||||
"github.com/kataras/iris/v12/middleware/requestid"
|
||||
)
|
||||
|
||||
func TestRequestID(t *testing.T) {
|
||||
app := iris.New()
|
||||
h := func(ctx iris.Context) {
|
||||
ctx.WriteString(requestid.Get(ctx))
|
||||
}
|
||||
|
||||
def := app.Party("/default")
|
||||
{
|
||||
def.Use(requestid.New())
|
||||
def.Get("/", h)
|
||||
}
|
||||
|
||||
const expectedCustomID = "my_id"
|
||||
custom := app.Party("/custom")
|
||||
{
|
||||
customGen := func(ctx *context.Context) string {
|
||||
return expectedCustomID
|
||||
}
|
||||
|
||||
custom.Use(requestid.New(customGen))
|
||||
custom.Get("/", h)
|
||||
}
|
||||
|
||||
const expectedErrMsg = "no id"
|
||||
customWithErr := app.Party("/custom_err")
|
||||
{
|
||||
customGen := func(ctx *context.Context) string {
|
||||
ctx.StopWithText(iris.StatusUnauthorized, expectedErrMsg)
|
||||
return ""
|
||||
}
|
||||
|
||||
customWithErr.Use(requestid.New(customGen))
|
||||
customWithErr.Get("/", h)
|
||||
}
|
||||
|
||||
const expectedCustomIDFromOtherMiddleware = "my custom id"
|
||||
changeID := app.Party("/custom_change_id")
|
||||
{
|
||||
changeID.Use(func(ctx iris.Context) {
|
||||
ctx.SetID(expectedCustomIDFromOtherMiddleware)
|
||||
ctx.Next()
|
||||
})
|
||||
changeID.Use(requestid.New())
|
||||
changeID.Get("/", h)
|
||||
}
|
||||
|
||||
e := httptest.New(t, app)
|
||||
e.GET("/default").Expect().Status(httptest.StatusOK).Body().NotEmpty()
|
||||
e.GET("/custom").Expect().Status(httptest.StatusOK).Body().Equal(expectedCustomID)
|
||||
e.GET("/custom_err").Expect().Status(httptest.StatusUnauthorized).Body().Equal(expectedErrMsg)
|
||||
e.GET("/custom_change_id").Expect().Status(httptest.StatusOK).Body().Equal(expectedCustomIDFromOtherMiddleware)
|
||||
}
|
||||
Reference in New Issue
Block a user