fix race condition in WebRTC tests

bluenviron · Apr 7, 2024 · 12d9be2 · 12d9be2
1 parent d5a18bf
commit 12d9be2
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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
 		})