Function TestMultiNodeDataBlob

test/multinode_test.go:430–554

(t *testing.T)

}

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

	_, _, finish := setupMultiNodeTestWithConfig("TestMultiNodeDataBlob", func(c *config.Config) {
		c.EnableParticipantDataBlob = true
		c.Limit.MaxDataBlobSize = 1024
	})
	defer finish()

	for _, testRTCServicePath := range testRTCServicePaths {
		t.Run(fmt.Sprintf("testRTCServicePath=%s", testRTCServicePath.String()), func(t *testing.T) {
			pubCapture := &dataBlobCapture{}
			subCapture := &dataBlobCapture{}

			// publisher on node 1, subscriber on node 2
			pub := createRTCClient("pub", defaultServerPort, testRTCServicePath, &client.Options{
				AutoSubscribe:             true,
				SignalResponseInterceptor: pubCapture.interceptor(),
			})
			sub := createRTCClient("sub", secondServerPort, testRTCServicePath, &client.Options{
				AutoSubscribe:             true,
				SignalResponseInterceptor: subCapture.interceptor(),
			})
			waitUntilConnected(t, pub, sub)
			defer stopClients(pub, sub)

			// wait for both nodes to see each other so the get request routes correctly
			testutils.WithTimeout(t, func() string {
				if sub.GetRemoteParticipant(pub.ID()) == nil {
					return "sub does not see pub yet"
				}
				return ""
			})

			key := &livekit.DataBlobKey{
				Key: &livekit.DataBlobKey_Generic{
					Generic: "blob-multinode",
				},
			}
			contents := []byte("multinode-content")

			require.NoError(t, pub.SendRequest(&livekit.SignalRequest{
				Message: &livekit.SignalRequest_StoreDataBlobRequest{
					StoreDataBlobRequest: &livekit.StoreDataBlobRequest{
						RequestId: 1,
						Blob: &livekit.DataBlob{
							Key:      key,
							Contents: contents,
						},
					},
				},
			}))

			testutils.WithTimeout(t, func() string {
				resp := pubCapture.takeStoreResponse()