From 4ab498024721782d1a93a149e32f08452a109843 Mon Sep 17 00:00:00 2001
From: Nedyalko Dyakov <nedyalko.dyakov@gmail.com>
Date: Tue, 22 Apr 2025 15:47:15 +0300
Subject: [PATCH] fix race in tests

---
 auth/auth_test.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/auth/auth_test.go b/auth/auth_test.go
index 88835a48..be762a85 100644
--- a/auth/auth_test.go
+++ b/auth/auth_test.go
@@ -2,6 +2,7 @@ package auth
 
 import (
 	"errors"
+	"sync"
 	"testing"
 	"time"
 )
@@ -47,10 +48,13 @@ func TestStreamingCredentialsProvider(t *testing.T) {
 
 		var receivedCreds []Credentials
 		var receivedErrors []error
+		var mu sync.Mutex
 
 		listener := NewReAuthCredentialsListener(
 			func(creds Credentials) error {
+				mu.Lock()
 				receivedCreds = append(receivedCreds, creds)
+				mu.Unlock()
 				return nil
 			},
 			func(err error) {
@@ -84,12 +88,14 @@ func TestStreamingCredentialsProvider(t *testing.T) {
 
 		// Wait for update to be processed
 		time.Sleep(100 * time.Millisecond)
+		mu.Lock()
 		if len(receivedCreds) != 2 {
 			t.Fatalf("expected 2 received credentials, got %d", len(receivedCreds))
 		}
 		if receivedCreds[1] != newCreds {
 			t.Fatalf("expected received credential %v, got %v", newCreds, receivedCreds[1])
 		}
+		mu.Unlock()
 
 		// Cancel subscription
 		if err := cancel(); err != nil {