go-packages/iris
1
0
Fork 0
mirror of https://github.com/kataras/iris.git synced 2025-01-23 02:31:04 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Improve middleware/requestid test suite (#2191)

Browse Source 
Add a test scenario for client sent request id header.
This commit is contained in:
Lemuel Roberto Bonifácio 2023-08-21 14:12:54 -03:00 committed by GitHub
parent 6a449876e9
commit 6da1fe9cbc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
middleware/requestid/requestid_test.go
View File

@ -55,9 +55,17 @@ func TestRequestID(t *testing.T) {
changeID.Get("/", h) changeID.Get("/", h)
} }
const expectedClientSentID = "client sent id"
clientSentID := app.Party("/client_id")
{
clientSentID.Use(requestid.New())
clientSentID.Get("/", h)
}
e := httptest.New(t, app) e := httptest.New(t, app)
e.GET("/default").Expect().Status(httptest.StatusOK).Body().NotEmpty() e.GET("/default").Expect().Status(httptest.StatusOK).Body().NotEmpty()
e.GET("/custom").Expect().Status(httptest.StatusOK).Body().IsEqual(expectedCustomID) e.GET("/custom").Expect().Status(httptest.StatusOK).Body().IsEqual(expectedCustomID)
e.GET("/custom_err").Expect().Status(httptest.StatusUnauthorized).Body().IsEqual(expectedErrMsg) e.GET("/custom_err").Expect().Status(httptest.StatusUnauthorized).Body().IsEqual(expectedErrMsg)
e.GET("/custom_change_id").Expect().Status(httptest.StatusOK).Body().IsEqual(expectedCustomIDFromOtherMiddleware) e.GET("/custom_change_id").Expect().Status(httptest.StatusOK).Body().IsEqual(expectedCustomIDFromOtherMiddleware)
e.GET("/client_id").WithHeader("X-Request-Id", expectedClientSentID).Expect().Header("X-Request-Id").IsEqual(expectedClientSentID)
} }