@@ -118,9 +118,7 @@ func SimulateAccessToContextOutsideHandler(
118118 retries int ,
119119) func (ctx context.Context ) {
120120 return func (ctx context.Context ) {
121- wait .Add (1 )
122- go func () {
123- defer wait .Done ()
121+ wait .Go (func () {
124122 global , cancel := context .WithTimeout (global , timeout * time .Duration (retries ))
125123 defer cancel ()
126124 for {
@@ -136,7 +134,7 @@ func SimulateAccessToContextOutsideHandler(
136134 case <- time .After (timeout / 10 ):
137135 }
138136 }
139- }( )
137+ })
140138 }
141139}
142140
@@ -250,12 +248,10 @@ func TestHumaFiber(t *testing.T) {
250248 api .UseMiddleware (HumaMiddleware )
251249 huma .Register (api , HelloOperation (), HelloHandler (simulator ))
252250
253- wait .Add (1 )
254- go func () {
255- defer wait .Done ()
251+ wait .Go (func () {
256252 err := app .Listener (ln )
257253 assert .NoError (t , err )
258- }( )
254+ })
259255 defer wait .Wait ()
260256
261257 err = WaitPing (ctx , server , timeout )
@@ -297,19 +293,17 @@ func TestHumaFiber(t *testing.T) {
297293
298294 // check that delay works
299295 doneFirst := make (chan bool )
300- wait .Add (1 )
301- go func () {
302- defer wait .Done ()
296+ wait .Go (func () {
303297 defer close (doneFirst )
304298 response , err := http .DefaultClient .Do (request )
305299 if response != nil && response .Body != nil {
306300 defer func () {
307301 _ = response .Body .Close ()
308302 }()
309303 }
310- assert .NoError (t , err )
304+ require .NoError (t , err )
311305 HelloResponseValidate (t , expected , response )
312- }( )
306+ })
313307 select {
314308 case <- ctx .Done ():
315309 return
@@ -329,31 +323,27 @@ func TestHumaFiber(t *testing.T) {
329323
330324 // check graceful shutdown
331325 doneSecond := make (chan bool )
332- wait .Add (1 )
333- go func () {
334- defer wait .Done ()
326+ wait .Go (func () {
335327 defer close (doneSecond )
336328 response , err := http .DefaultClient .Do (request )
337329 if response != nil && response .Body != nil {
338330 defer func () {
339331 _ = response .Body .Close ()
340332 }()
341333 }
342- assert .NoError (t , err )
334+ require .NoError (t , err )
343335 HelloResponseValidate (t , expected , response )
344- }( )
336+ })
345337
346338 // perform shutdown
347339 doneShutdown := make (chan bool )
348- wait .Add (1 )
349- go func () {
350- defer wait .Done ()
340+ wait .Go (func () {
351341 defer close (doneShutdown )
352342 time .Sleep (timeout ) // delay before shutdown to start request processing
353343 err := app .ShutdownWithContext (ctx )
354- assert .NoError (t , err )
344+ require .NoError (t , err )
355345 time .Sleep (timeout ) // delay after shutdown to catch request processing
356- }( )
346+ })
357347
358348 // request should be handled
359349 select {
0 commit comments