From f7d16df78e09e78d4bd9cb38884a299251c5c0f9 Mon Sep 17 00:00:00 2001
From: Mohammed Al Sahaf <msaa1990@gmail.com>
Date: Sun, 3 Aug 2025 04:30:47 +0300
Subject: [PATCH] fix another race

Signed-off-by: Mohammed Al Sahaf <msaa1990@gmail.com>
---
 modules/logging/netwriter_test.go | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/modules/logging/netwriter_test.go b/modules/logging/netwriter_test.go
index 1c1765863..5d3639528 100644
--- a/modules/logging/netwriter_test.go
+++ b/modules/logging/netwriter_test.go
@@ -95,10 +95,15 @@ func (ms *mockServer) handleConnection(conn net.Conn) {
 
 	scanner := bufio.NewScanner(conn)
 	for scanner.Scan() {
-		line := scanner.Text()
-		ms.mu.Lock()
-		ms.messages = append(ms.messages, line)
-		ms.mu.Unlock()
+		select {
+		case <-ms.ctx.Done():
+			return
+		default:
+			line := scanner.Text()
+			ms.mu.Lock()
+			ms.messages = append(ms.messages, line)
+			ms.mu.Unlock()
+		}
 	}
 }