usm: kafka: test: Add deadline to raw client read (#27762)

DataDog · Jul 20, 2024 · 431f2b1 · 431f2b1
1 parent d699e3b
commit 431f2b1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkg/network/usm/kafka_monitor_test.go b/pkg/network/usm/kafka_monitor_test.go
@@ -782,6 +782,9 @@ func (can *CannedClientServer) runClient(msgs []Message) {
 	require.NoError(can.t, err)
 	can.t.Cleanup(func() { _ = conn.Close() })
 
+	// Safety measure to avoid blocking forever in the case of bugs.
+	conn.SetDeadline(time.Now().Add(30 * time.Second))
+
 	reader := bufio.NewReader(conn)
 	for _, msg := range msgs {
 		buf := make([]byte, 0)