mirror of
https://github.com/pion/webrtc.git
synced 2025-10-28 01:31:33 +08:00
106 lines
2.8 KiB
Go
106 lines
2.8 KiB
Go
//go:build !js
|
|
// +build !js
|
|
|
|
package webrtc
|
|
|
|
import (
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/pion/transport/test"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestICETransport_OnConnectionStateChange(t *testing.T) {
|
|
report := test.CheckRoutines(t)
|
|
defer report()
|
|
|
|
lim := test.TimeOut(time.Second * 30)
|
|
defer lim.Stop()
|
|
|
|
pcOffer, pcAnswer, err := newPair()
|
|
assert.NoError(t, err)
|
|
|
|
iceOfferComplete := make(chan struct{})
|
|
iceAnswerComplete := make(chan struct{})
|
|
|
|
pcOffer.SCTP().Transport().ICETransport().OnConnectionStateChange(func(s ICETransportState) {
|
|
if s == ICETransportStateConnected {
|
|
close(iceOfferComplete)
|
|
}
|
|
})
|
|
|
|
pcAnswer.SCTP().Transport().ICETransport().OnConnectionStateChange(func(s ICETransportState) {
|
|
if s == ICETransportStateConnected {
|
|
close(iceAnswerComplete)
|
|
}
|
|
})
|
|
|
|
assert.NoError(t, signalPair(pcOffer, pcAnswer))
|
|
<-iceOfferComplete
|
|
|
|
closePairNow(t, pcOffer, pcAnswer)
|
|
}
|
|
|
|
func TestICETransport_OnSelectedCandidatePairChange(t *testing.T) {
|
|
report := test.CheckRoutines(t)
|
|
defer report()
|
|
|
|
lim := test.TimeOut(time.Second * 30)
|
|
defer lim.Stop()
|
|
|
|
pcOffer, pcAnswer, err := newPair()
|
|
assert.NoError(t, err)
|
|
|
|
iceComplete := make(chan bool)
|
|
pcAnswer.OnICEConnectionStateChange(func(iceState ICEConnectionState) {
|
|
if iceState == ICEConnectionStateConnected {
|
|
time.Sleep(3 * time.Second)
|
|
close(iceComplete)
|
|
}
|
|
})
|
|
|
|
senderCalledCandidateChange := int32(0)
|
|
pcOffer.SCTP().Transport().ICETransport().OnSelectedCandidatePairChange(func(pair *ICECandidatePair) {
|
|
atomic.StoreInt32(&senderCalledCandidateChange, 1)
|
|
})
|
|
|
|
assert.NoError(t, signalPair(pcOffer, pcAnswer))
|
|
<-iceComplete
|
|
|
|
if atomic.LoadInt32(&senderCalledCandidateChange) == 0 {
|
|
t.Fatalf("Sender ICETransport OnSelectedCandidateChange was never called")
|
|
}
|
|
|
|
closePairNow(t, pcOffer, pcAnswer)
|
|
}
|
|
|
|
func TestICETransport_GetSelectedCandidatePair(t *testing.T) {
|
|
offerer, answerer, err := newPair()
|
|
assert.NoError(t, err)
|
|
|
|
peerConnectionConnected := untilConnectionState(PeerConnectionStateConnected, offerer, answerer)
|
|
|
|
offererSelectedPair, err := offerer.SCTP().Transport().ICETransport().GetSelectedCandidatePair()
|
|
assert.NoError(t, err)
|
|
assert.Nil(t, offererSelectedPair)
|
|
|
|
answererSelectedPair, err := answerer.SCTP().Transport().ICETransport().GetSelectedCandidatePair()
|
|
assert.NoError(t, err)
|
|
assert.Nil(t, answererSelectedPair)
|
|
|
|
assert.NoError(t, signalPair(offerer, answerer))
|
|
peerConnectionConnected.Wait()
|
|
|
|
offererSelectedPair, err = offerer.SCTP().Transport().ICETransport().GetSelectedCandidatePair()
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, offererSelectedPair)
|
|
|
|
answererSelectedPair, err = answerer.SCTP().Transport().ICETransport().GetSelectedCandidatePair()
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, answererSelectedPair)
|
|
|
|
closePairNow(t, offerer, answerer)
|
|
}
|