@@ -108,7 +108,7 @@ func testGracefulShutdown(t *testing.T, conn *Connection, inst *test_helpers.Tar
108
108
// Wait until server go down.
109
109
// Server will go down only when it process all requests from our connection
110
110
// (or on timeout).
111
- _ , err = inst . Cmd . Process .Wait ()
111
+ err = inst .Wait ()
112
112
require .Nil (t , err )
113
113
shutdownFinish := time .Now ()
114
114
shutdownTime := shutdownFinish .Sub (shutdownStart )
@@ -256,7 +256,7 @@ func TestNoGracefulShutdown(t *testing.T) {
256
256
require .NotNilf (t , err , "sleep request error" )
257
257
258
258
// Wait until server go down.
259
- _ , err = inst . Cmd . Process .Wait ()
259
+ err = inst .Wait ()
260
260
require .Nil (t , err )
261
261
shutdownFinish := time .Now ()
262
262
shutdownTime := shutdownFinish .Sub (shutdownStart )
@@ -327,7 +327,7 @@ func TestGracefulShutdownRespectsClose(t *testing.T) {
327
327
require .NotNilf (t , err , "sleep request error" )
328
328
329
329
// Wait until server go down.
330
- _ , err = inst . Cmd . Process .Wait ()
330
+ err = inst .Wait ()
331
331
require .Nil (t , err )
332
332
shutdownFinish := time .Now ()
333
333
shutdownTime := shutdownFinish .Sub (shutdownStart )
@@ -401,7 +401,7 @@ func TestGracefulShutdownNotRacesWithRequestReconnect(t *testing.T) {
401
401
402
402
// Wait until server go down.
403
403
// Server is expected to go down on timeout.
404
- _ , err = inst . Cmd . Process .Wait ()
404
+ err = inst .Wait ()
405
405
require .Nil (t , err )
406
406
407
407
// Help test helpers to properly clean up.
@@ -497,7 +497,7 @@ func TestGracefulShutdownCloseConcurrent(t *testing.T) {
497
497
srvStop .Wait ()
498
498
require .Nil (t , sret , "No errors on server SIGTERM" )
499
499
500
- _ , err = inst . Cmd . Process .Wait ()
500
+ err = inst .Wait ()
501
501
require .Nil (t , err )
502
502
503
503
// Help test helpers to properly clean up.
@@ -590,7 +590,7 @@ func TestGracefulShutdownConcurrent(t *testing.T) {
590
590
caseWg .Wait ()
591
591
require .Nil (t , ret , "No errors on concurrent wait" )
592
592
593
- _ , err = inst . Cmd . Process .Wait ()
593
+ err = inst .Wait ()
594
594
require .Nil (t , err )
595
595
596
596
// Help test helpers to properly clean up.
0 commit comments