Function TestConnectionStats

test/singlenode_test.go:247–456

(t *testing.T)

}

func TestConnectionStats(t *testing.T) {
	if testing.Short() {
		t.SkipNow()
		return
	}

	s, finish := setupSingleNodeTest("TestConnectionStats")
	defer finish()

	for _, testRTCServicePath := range testRTCServicePaths {
		t.Run(fmt.Sprintf("testRTCServicePath=%s", testRTCServicePath.String()), func(t *testing.T) {
			c1 := createRTCClient("c1", defaultServerPort, testRTCServicePath, nil)
			c2 := createRTCClient("c2", defaultServerPort, testRTCServicePath, nil)
			waitUntilConnected(t, c1, c2)
			defer func() {
				c1.Stop()
				c2.Stop()
			}()

			// both clients publish audio + video
			t1, err := c1.AddStaticTrack("audio/opus", "audio", "c1audio")
			require.NoError(t, err)
			defer t1.Stop()
			t2, err := c1.AddStaticTrack("video/vp8", "video", "c1video")
			require.NoError(t, err)
			defer t2.Stop()

			t3, err := c2.AddStaticTrack("audio/opus", "audio", "c2audio")
			require.NoError(t, err)
			defer t3.Stop()
			t4, err := c2.AddStaticTrack("video/vp8", "video", "c2video")
			require.NoError(t, err)
			defer t4.Stop()

			// wait for cross-subscriptions: each client should receive 2 tracks from the other
			testutils.WithTimeout(t, func() string {
				if len(c1.SubscribedTracks()[c2.ID()]) != 2 {
					return "c1 did not subscribe to both tracks from c2"
				}
				if len(c2.SubscribedTracks()[c1.ID()]) != 2 {
					return "c2 did not subscribe to both tracks from c1"
				}
				return ""
			})

			room := s.RoomManager().GetRoom(context.Background(), testRoom)
			require.NotNil(t, room)

			// hook the upstream WebRTCReceiver.OnStatsUpdate and downstream DownTrack.OnStatsUpdate
			// callbacks so we can verify the AnalyticsStat delivered through each carries valid
			// delta data. MediaTrack.Receivers() returns one entry per potential codec; only those
			// matching the actually published codec are *sfu.WebRTCReceiver, the rest are
			// placeholder *rtc.DummyReceiver instances that we skip.
			type statCapture struct {
				lock sync.Mutex
				stat *livekit.AnalyticsStat
			}
			receiverCaptures := make(map[livekit.TrackID]*statCapture)