Skip to content

Commit 6da1fe9

Browse files
Improve middleware/requestid test suite (#2191)
Add a test scenario for client sent request id header.
1 parent 6a44987 commit 6da1fe9

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

middleware/requestid/requestid_test.go

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,9 +55,17 @@ func TestRequestID(t *testing.T) {
5555
changeID.Get("/", h)
5656
}
5757

58+
const expectedClientSentID = "client sent id"
59+
clientSentID := app.Party("/client_id")
60+
{
61+
clientSentID.Use(requestid.New())
62+
clientSentID.Get("/", h)
63+
}
64+
5865
e := httptest.New(t, app)
5966
e.GET("/default").Expect().Status(httptest.StatusOK).Body().NotEmpty()
6067
e.GET("/custom").Expect().Status(httptest.StatusOK).Body().IsEqual(expectedCustomID)
6168
e.GET("/custom_err").Expect().Status(httptest.StatusUnauthorized).Body().IsEqual(expectedErrMsg)
6269
e.GET("/custom_change_id").Expect().Status(httptest.StatusOK).Body().IsEqual(expectedCustomIDFromOtherMiddleware)
70+
e.GET("/client_id").WithHeader("X-Request-Id", expectedClientSentID).Expect().Header("X-Request-Id").IsEqual(expectedClientSentID)
6371
}

0 commit comments

Comments
 (0)