Fixed issue in tests

evg4b · Apr 21, 2024 · b15cf31 · b15cf31
1 parent bb16f15
commit b15cf31
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/internal/helpers/graceful_shutdown_internal_test.go b/internal/helpers/graceful_shutdown_internal_test.go
@@ -5,6 +5,7 @@ import (
 	"os"
 	"os/signal"
 	"sync"
+	"sync/atomic"
 	"syscall"
 	"testing"
 	"time"
@@ -54,18 +55,19 @@ func TestGracefulShutdown(t *testing.T) {
 	t.Run("shutdown when context is done", WithGoroutines(func(t *testing.T, env *Env) {
 		ctx, cancel := context.WithCancel(context.Background())
 
-		called := false
+		called := &atomic.Bool{}
+
 		env.Go(func() {
 			err := GracefulShutdown(ctx, func(_ context.Context) error {
-				called = true
+				called.Store(true)
 
 				return nil
 			})
 			require.NoError(t, err)
 		})
 
 		env.CheckAfterAll(func() {
-			assert.True(t, called)
+			assert.True(t, called.Load())
 		})
 
 		cancel()