mirror of
https://github.com/livekit/livekit.git
synced 2026-03-31 04:35:40 +00:00
87 lines
2.9 KiB
Go
87 lines
2.9 KiB
Go
package rtc
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/livekit/livekit-server/pkg/testutils"
|
|
livekit "github.com/livekit/livekit-server/proto"
|
|
"github.com/pion/webrtc/v3"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMissingAnswerDuringICERestart(t *testing.T) {
|
|
params := TransportParams{
|
|
Target: livekit.SignalTarget_PUBLISHER,
|
|
Config: &WebRTCConfig{},
|
|
Stats: nil,
|
|
}
|
|
transportA, err := NewPCTransport(params)
|
|
require.NoError(t, err)
|
|
_, err = transportA.pc.CreateDataChannel("test", nil)
|
|
require.NoError(t, err)
|
|
transportB, err := NewPCTransport(params)
|
|
require.NoError(t, err)
|
|
|
|
// exchange ICE
|
|
transportA.pc.OnICECandidate(func(candidate *webrtc.ICECandidate) {
|
|
if candidate == nil {
|
|
return
|
|
}
|
|
t.Logf("got ICE candidate from A: %v", candidate)
|
|
require.NoError(t, transportB.AddICECandidate(candidate.ToJSON()))
|
|
})
|
|
transportB.pc.OnICECandidate(func(candidate *webrtc.ICECandidate) {
|
|
if candidate == nil {
|
|
return
|
|
}
|
|
t.Logf("got ICE candidate from B: %v", candidate)
|
|
require.NoError(t, transportA.AddICECandidate(candidate.ToJSON()))
|
|
})
|
|
// set offer/answer
|
|
handleOffer := handleOfferFunc(t, transportA, transportB)
|
|
transportA.OnOffer(handleOffer)
|
|
|
|
// first establish connection
|
|
require.NoError(t, transportA.CreateAndSendOffer(nil))
|
|
|
|
// ensure we are connected the first time
|
|
testutils.WithTimeout(t, "initial ICE connectivity", func() bool {
|
|
return transportA.pc.ICEConnectionState() == webrtc.ICEConnectionStateConnected &&
|
|
transportB.pc.ICEConnectionState() == webrtc.ICEConnectionStateConnected
|
|
})
|
|
require.Equal(t, webrtc.ICEConnectionStateConnected, transportA.pc.ICEConnectionState())
|
|
require.Equal(t, webrtc.ICEConnectionStateConnected, transportB.pc.ICEConnectionState())
|
|
|
|
// offer again, but missed
|
|
transportA.OnOffer(func(sd webrtc.SessionDescription) {})
|
|
require.NoError(t, transportA.CreateAndSendOffer(nil))
|
|
require.Equal(t, webrtc.SignalingStateHaveLocalOffer, transportA.pc.SignalingState())
|
|
require.Equal(t, negotiationStateClient, transportA.negotiationState)
|
|
|
|
// now restart ICE
|
|
t.Logf("creating offer with ICE restart")
|
|
transportA.OnOffer(handleOffer)
|
|
require.NoError(t, transportA.CreateAndSendOffer(&webrtc.OfferOptions{
|
|
ICERestart: true,
|
|
}))
|
|
|
|
testutils.WithTimeout(t, "restarted ICE connectivity", func() bool {
|
|
return transportA.pc.ICEConnectionState() == webrtc.ICEConnectionStateConnected &&
|
|
transportB.pc.ICEConnectionState() == webrtc.ICEConnectionStateConnected
|
|
})
|
|
}
|
|
|
|
func handleOfferFunc(t *testing.T, current, other *PCTransport) func(sd webrtc.SessionDescription) {
|
|
return func(sd webrtc.SessionDescription) {
|
|
t.Logf("handling offer")
|
|
t.Logf("setting other remote description")
|
|
require.NoError(t, other.SetRemoteDescription(sd))
|
|
answer, err := other.pc.CreateAnswer(nil)
|
|
require.NoError(t, err)
|
|
require.NoError(t, other.pc.SetLocalDescription(answer))
|
|
|
|
t.Logf("setting answer on current")
|
|
require.NoError(t, current.SetRemoteDescription(answer))
|
|
}
|
|
}
|