usm: kafka: tests: Improve cleanup to solve flakiness (#26225)

DataDog · Jun 3, 2024 · d3371d7 · d3371d7
1 parent d31dc90
commit d3371d7
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/pkg/network/usm/kafka_monitor_test.go b/pkg/network/usm/kafka_monitor_test.go
@@ -854,8 +854,11 @@ func (can *CannedClientServer) runServer() {
 	})
 
 	go func() {
-		defer f.Close()
-		defer listener.Close()
+		defer func() {
+			listener.Close()
+			f.Close()
+			can.done <- true
+		}()
 
 		conn, err := listener.Accept()
 		require.NoError(can.t, err)
@@ -891,8 +894,6 @@ func (can *CannedClientServer) runServer() {
 		if prevconn != nil {
 			prevconn.Close()
 		}
-
-		can.done <- true
 	}()
 }