diff --git a/internal/semaphore/sem_test.go b/internal/semaphore/sem_test.go index 1fd5bf601..c1a328963 100644 --- a/internal/semaphore/sem_test.go +++ b/internal/semaphore/sem_test.go @@ -4,17 +4,21 @@ import ( "testing" "github.com/stretchr/testify/require" + "golang.org/x/sync/errgroup" ) func TestSemaphore(t *testing.T) { var sem = New(1) var counter = 0 + var g errgroup.Group for i := 0; i < 10; i++ { sem.Acquire() - go func() { + g.Go(func() error { counter++ sem.Release() - }() + return nil + }) } - require.Equal(t, counter, 9) + require.NoError(t, g.Wait()) + require.Equal(t, counter, 10) }