mirror of
https://github.com/livekit/livekit.git
synced 2026-03-30 19:55:41 +00:00
139 lines
3.1 KiB
Go
139 lines
3.1 KiB
Go
package test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/livekit/livekit-server/pkg/rtc"
|
|
livekit "github.com/livekit/protocol/proto"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/livekit/livekit-server/pkg/testutils"
|
|
)
|
|
|
|
func TestMultiNodeRouting(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
return
|
|
}
|
|
_, _, finish := setupMultiNodeTest("TestMultiNodeRouting")
|
|
defer finish()
|
|
|
|
// creating room on node 1
|
|
_, err := roomClient.CreateRoom(contextWithCreateRoomToken(), &livekit.CreateRoomRequest{
|
|
Name: testRoom,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
// one node connecting to node 1, and another connecting to node 2
|
|
c1 := createRTCClient("c1", defaultServerPort, nil)
|
|
c2 := createRTCClient("c2", secondServerPort, nil)
|
|
waitUntilConnected(t, c1, c2)
|
|
defer stopClients(c1, c2)
|
|
|
|
// c1 publishing, and c2 receiving
|
|
t1, err := c1.AddStaticTrack("audio/opus", "audio", "webcam")
|
|
require.NoError(t, err)
|
|
if t1 != nil {
|
|
defer t1.Stop()
|
|
}
|
|
|
|
testutils.WithTimeout(t, "c2 should receive one track", func() bool {
|
|
if len(c2.SubscribedTracks()) == 0 {
|
|
return false
|
|
}
|
|
// should have received two tracks
|
|
if len(c2.SubscribedTracks()[c1.ID()]) != 1 {
|
|
return false
|
|
}
|
|
|
|
tr1 := c2.SubscribedTracks()[c1.ID()][0]
|
|
streamId, _ := rtc.UnpackStreamID(tr1.StreamID())
|
|
require.Equal(t, c1.ID(), streamId)
|
|
return true
|
|
})
|
|
}
|
|
|
|
func TestConnectWithoutCreation(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
return
|
|
}
|
|
|
|
_, _, finish := setupMultiNodeTest("TestConnectWithoutCreation")
|
|
defer finish()
|
|
|
|
c1 := createRTCClient("c1", defaultServerPort, nil)
|
|
waitUntilConnected(t, c1)
|
|
|
|
c1.Stop()
|
|
}
|
|
|
|
// testing multiple scenarios rooms
|
|
func TestMultinodePublishingUponJoining(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
return
|
|
}
|
|
_, _, finish := setupMultiNodeTest("TestMultinodePublishingUponJoining")
|
|
defer finish()
|
|
|
|
scenarioPublishingUponJoining(t, defaultServerPort, secondServerPort)
|
|
}
|
|
|
|
func TestMultinodeReceiveBeforePublish(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
return
|
|
}
|
|
_, _, finish := setupMultiNodeTest("TestMultinodeReceiveBeforePublish")
|
|
defer finish()
|
|
|
|
scenarioReceiveBeforePublish(t)
|
|
}
|
|
|
|
// reconnecting to the same room, after one of the servers has gone away
|
|
func TestMultinodeReconnectAfterNodeShutdown(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
return
|
|
}
|
|
|
|
_, s2, finish := setupMultiNodeTest("TestMultinodeReconnectAfterNodeShutdown")
|
|
defer finish()
|
|
|
|
// creating room on node 1
|
|
_, err := roomClient.CreateRoom(contextWithCreateRoomToken(), &livekit.CreateRoomRequest{
|
|
Name: testRoom,
|
|
NodeId: s2.Node().Id,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
// one node connecting to node 1, and another connecting to node 2
|
|
c1 := createRTCClient("c1", defaultServerPort, nil)
|
|
c2 := createRTCClient("c2", secondServerPort, nil)
|
|
|
|
waitUntilConnected(t, c1, c2)
|
|
stopClients(c1, c2)
|
|
|
|
// stop s2, and connect to room again
|
|
s2.Stop()
|
|
|
|
time.Sleep(syncDelay)
|
|
|
|
c3 := createRTCClient("c3", defaultServerPort, nil)
|
|
waitUntilConnected(t, c3)
|
|
}
|
|
|
|
func TestMultinodeDataPublishing(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
return
|
|
}
|
|
|
|
_, _, finish := setupMultiNodeTest("TestMultinodeDataPublishing")
|
|
defer finish()
|
|
|
|
scenarioDataPublish(t)
|
|
}
|