blackteay/gin
3
0
Fork 0
mirror of https://github.com/gin-gonic/gin.git synced 2025-10-17 05:42:09 +08:00
Code Issues Projects Releases Wiki Activity

feat: add unit test case for engine Shutdown

Browse Source
This commit is contained in:
Yaofu 2022-06-24 17:33:28 +08:00 committed by GitHub
parent 970402c4c9
commit b60c029e25
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
gin_test.go
View File

@ -696,3 +696,26 @@ func assertRoutePresent(t *testing.T, gotRoutes RoutesInfo, wantRoute RouteInfo)
func handlerTest1(c *Context) {} func handlerTest1(c *Context) {}
func handlerTest2(c *Context) {} func handlerTest2(c *Context) {}
// TestEngine_Shutdown to test engine.Shutdown.
func TestEngine_Shutdown(t *testing.T) {
ln, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
fmt.Println(err)
}
r := Default()
r.serverList = append(r.serverList, nil)
go func() {
err = r.RunListener(ln)
assert.Equal(t, err, http.ErrServerClosed)
}()
time.Sleep(time.Millisecond)
assert.Equal(t, len(r.serverList), 2)
r.serverList = append(r.serverList, nil)
ctx, _ := context.WithTimeout(context.Background(), time.Nanosecond)
time.Sleep(time.Nanosecond)
err = r.Shutdown(ctx)
assert.Error(t, err, context.DeadlineExceeded)
err = r.Shutdown(context.TODO())
assert.Nil(t, err)
}