From e6d44851b1930bd11dfd078d75a4d18b6908dcfb Mon Sep 17 00:00:00 2001
From: Mohammed Al Sahaf <msaa1990@gmail.com>
Date: Mon, 4 Aug 2025 00:26:21 +0300
Subject: [PATCH] fix race in `TestNetWriter_ConnectionRetry`

Signed-off-by: Mohammed Al Sahaf <msaa1990@gmail.com>
---
 modules/logging/netwriter_test.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/logging/netwriter_test.go b/modules/logging/netwriter_test.go
index 5d3639528..c1f11e4f8 100644
--- a/modules/logging/netwriter_test.go
+++ b/modules/logging/netwriter_test.go
@@ -139,10 +139,10 @@ func (ms *mockServer) restart(t *testing.T) {
 	if err != nil {
 		t.Fatalf("Failed to restart mock server: %v", err)
 	}
+	ms.mu.Lock()
 	ms.listener = listener
 
 	// Clear existing messages to track only new ones
-	ms.mu.Lock()
 	ms.messages = nil
 	ms.mu.Unlock()