mirror of
https://github.com/gin-gonic/gin.git
synced 2025-10-14 12:12:12 +08:00
test: refactor test context initialization and handler logic
- Remove an empty line in `TestContextInitQueryCache` - Add `TestContextNext` function with tests for `Next` method behavior with no handlers, one handler, and multiple handlers Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
This commit is contained in:
parent
7e8e21b21c
commit
151f477408
@ -610,7 +610,6 @@ func TestContextInitQueryCache(t *testing.T) {
|
||||
assert.Equal(t, test.expectedQueryCache, test.testContext.queryCache)
|
||||
})
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestContextDefaultQueryOnEmptyRequest(t *testing.T) {
|
||||
@ -3038,3 +3037,47 @@ func TestInterceptedHeader(t *testing.T) {
|
||||
assert.Equal(t, "", w.Result().Header.Get("X-Test"))
|
||||
assert.Equal(t, "present", w.Result().Header.Get("X-Test-2"))
|
||||
}
|
||||
|
||||
func TestContextNext(t *testing.T) {
|
||||
c, _ := CreateTestContext(httptest.NewRecorder())
|
||||
|
||||
// Test with no handlers
|
||||
c.Next()
|
||||
assert.Equal(t, int8(0), c.index)
|
||||
|
||||
// Test with one handler
|
||||
c.index = -1
|
||||
c.handlers = HandlersChain{func(c *Context) {
|
||||
c.Set("key", "value")
|
||||
}}
|
||||
c.Next()
|
||||
assert.Equal(t, int8(1), c.index)
|
||||
value, exists := c.Get("key")
|
||||
assert.True(t, exists)
|
||||
assert.Equal(t, "value", value)
|
||||
|
||||
// Test with multiple handlers
|
||||
c.handlers = HandlersChain{
|
||||
func(c *Context) {
|
||||
c.Set("key1", "value1")
|
||||
c.Next()
|
||||
c.Set("key2", "value2")
|
||||
},
|
||||
nil,
|
||||
func(c *Context) {
|
||||
c.Set("key3", "value3")
|
||||
},
|
||||
}
|
||||
c.index = -1
|
||||
c.Next()
|
||||
assert.Equal(t, int8(4), c.index)
|
||||
value, exists = c.Get("key1")
|
||||
assert.True(t, exists)
|
||||
assert.Equal(t, "value1", value)
|
||||
value, exists = c.Get("key2")
|
||||
assert.True(t, exists)
|
||||
assert.Equal(t, "value2", value)
|
||||
value, exists = c.Get("key3")
|
||||
assert.True(t, exists)
|
||||
assert.Equal(t, "value3", value)
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user