Fix the metric race condition in tests.

Jigsaw-Code · Jan 21, 2025 · 220d1d7 · 220d1d7
1 parent bcac22b
commit 220d1d7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/internal/integration_test/integration_test.go b/internal/integration_test/integration_test.go
@@ -288,6 +288,8 @@ type fakeUDPAssociationMetrics struct {
 var _ service.UDPAssociationMetrics = (*fakeUDPAssociationMetrics)(nil)
 
 func (m *fakeUDPAssociationMetrics) AddAuthentication(key string) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
 	m.accessKey = key
 }
 

diff --git a/service/udp_test.go b/service/udp_test.go
@@ -161,6 +161,8 @@ type fakeUDPAssociationMetrics struct {
 var _ UDPAssociationMetrics = (*fakeUDPAssociationMetrics)(nil)
 
 func (m *fakeUDPAssociationMetrics) AddAuthentication(key string) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
 	m.accessKey = key
 }