70 lines
2.0 KiB
Go
70 lines
2.0 KiB
Go
package webrtc
|
|
|
|
import (
|
|
"math/rand"
|
|
"testing"
|
|
|
|
"github.com/pion/webrtc/v3"
|
|
"gitlab.crans.org/nounous/ghostream/stream"
|
|
)
|
|
|
|
func TestServe(t *testing.T) {
|
|
// Init streams messaging and WebRTC server
|
|
streams := make(map[string]*stream.Stream)
|
|
remoteSdpChan := make(chan struct {
|
|
StreamID string
|
|
RemoteDescription webrtc.SessionDescription
|
|
})
|
|
localSdpChan := make(chan webrtc.SessionDescription)
|
|
cfg := Options{
|
|
Enabled: true,
|
|
MinPortUDP: 10000,
|
|
MaxPortUDP: 10005,
|
|
STUNServers: []string{"stun:stun.l.google.com:19302"},
|
|
}
|
|
go Serve(streams, remoteSdpChan, localSdpChan, &cfg)
|
|
|
|
// New client connection
|
|
mediaEngine := webrtc.MediaEngine{}
|
|
mediaEngine.RegisterDefaultCodecs()
|
|
api := webrtc.NewAPI(webrtc.WithMediaEngine(mediaEngine))
|
|
peerConnection, _ := api.NewPeerConnection(webrtc.Configuration{})
|
|
|
|
// Create video track
|
|
codec, payloadType := getPayloadType(mediaEngine, webrtc.RTPCodecTypeVideo, "VP8")
|
|
videoTrack, err := webrtc.NewTrack(payloadType, rand.Uint32(), "video", "pion", codec)
|
|
if err != nil {
|
|
t.Error("Failed to create new video track", err)
|
|
}
|
|
if _, err = peerConnection.AddTrack(videoTrack); err != nil {
|
|
t.Error("Failed to add video track", err)
|
|
}
|
|
|
|
// Create audio track
|
|
codec, payloadType = getPayloadType(mediaEngine, webrtc.RTPCodecTypeAudio, "opus")
|
|
audioTrack, err := webrtc.NewTrack(payloadType, rand.Uint32(), "audio", "pion", codec)
|
|
if err != nil {
|
|
t.Error("Failed to create new audio track", err)
|
|
}
|
|
if _, err = peerConnection.AddTrack(audioTrack); err != nil {
|
|
t.Error("Failed to add audio track", err)
|
|
}
|
|
|
|
// Create offer
|
|
offer, _ := peerConnection.CreateOffer(nil)
|
|
|
|
// Create channel that is blocked until ICE Gathering is complete
|
|
gatherComplete := webrtc.GatheringCompletePromise(peerConnection)
|
|
peerConnection.SetLocalDescription(offer)
|
|
<-gatherComplete
|
|
|
|
// Send offer to server
|
|
remoteSdpChan <- struct {
|
|
StreamID string
|
|
RemoteDescription webrtc.SessionDescription
|
|
}{"demo", *peerConnection.LocalDescription()}
|
|
_ = <-localSdpChan
|
|
|
|
// FIXME: verify connection did work
|
|
}
|