From 9ce12ffa4aee1539d2a9bcf41691a33ff87f325a Mon Sep 17 00:00:00 2001
From: Alessandro Ros <aler9.dev@gmail.com>
Date: Sun, 7 Apr 2024 22:46:24 +0200
Subject: [PATCH] fix race condition in WebRTC tests (#3204)

---
 internal/servers/webrtc/server_test.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/internal/servers/webrtc/server_test.go b/internal/servers/webrtc/server_test.go
index fce4bf78..ff80559c 100644
--- a/internal/servers/webrtc/server_test.go
+++ b/internal/servers/webrtc/server_test.go
@@ -221,10 +221,17 @@ func TestServerPublish(t *testing.T) {
 		path.stream.Desc().Medias[0],
 		path.stream.Desc().Medias[0].Formats[0],
 		func(u unit.Unit) error {
+			select {
+			case <-recv:
+				return nil
+			default:
+			}
+
 			require.Equal(t, [][]byte{
 				{1},
 			}, u.(*unit.H264).AU)
 			close(recv)
+
 			return nil
 		})