mirror of
https://github.com/pion/webrtc.git
synced 2025-10-05 07:06:51 +08:00
e500917a6e
SampleBuilder provides a simple API to build an RTCSample from RTP packets. This is useful when proxying audio/video data to multiple peers. This is the alternative to allowing users to push RTP Packets directly. This would be confusing as we would throw away some of the information that users give us and could lead to weird edge cases Resolves #112
99 lines
2.8 KiB
Go
99 lines
2.8 KiB
Go
package samplebuilder
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/pions/webrtc/pkg/media"
|
|
"github.com/pions/webrtc/pkg/rtp"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type sampleBuilderTest struct {
|
|
message string
|
|
packets []*rtp.Packet
|
|
samples []*media.RTCSample
|
|
bufferSize uint16
|
|
}
|
|
|
|
var testCases = []sampleBuilderTest{
|
|
{
|
|
message: "SampleBuilder shouldn't emit anything if only one RTP packet has been pushed",
|
|
packets: []*rtp.Packet{
|
|
{SequenceNumber: 5000, Timestamp: 5, Payload: []byte{0x01}},
|
|
},
|
|
samples: []*media.RTCSample{},
|
|
bufferSize: 50,
|
|
},
|
|
{
|
|
message: "SampleBuilder should emit one packet, we had three packets with unique timestamps",
|
|
packets: []*rtp.Packet{
|
|
{SequenceNumber: 5000, Timestamp: 5, Payload: []byte{0x01}},
|
|
{SequenceNumber: 5001, Timestamp: 6, Payload: []byte{0x02}},
|
|
{SequenceNumber: 5002, Timestamp: 7, Payload: []byte{0x03}},
|
|
},
|
|
samples: []*media.RTCSample{
|
|
{Data: []byte{0x02}, Samples: 1},
|
|
},
|
|
bufferSize: 50,
|
|
},
|
|
{
|
|
message: "SampleBuilder should emit one packet, we had two packets but two with duplicate timestamps",
|
|
packets: []*rtp.Packet{
|
|
{SequenceNumber: 5000, Timestamp: 5, Payload: []byte{0x01}},
|
|
{SequenceNumber: 5001, Timestamp: 6, Payload: []byte{0x02}},
|
|
{SequenceNumber: 5002, Timestamp: 6, Payload: []byte{0x03}},
|
|
{SequenceNumber: 5003, Timestamp: 7, Payload: []byte{0x04}},
|
|
},
|
|
samples: []*media.RTCSample{
|
|
{Data: []byte{0x02, 0x03}, Samples: 1},
|
|
},
|
|
bufferSize: 50,
|
|
},
|
|
{
|
|
message: "SampleBuilder shouldn't emit a packet because we have a gap before a valid one",
|
|
packets: []*rtp.Packet{
|
|
{SequenceNumber: 5000, Timestamp: 5, Payload: []byte{0x01}},
|
|
{SequenceNumber: 5007, Timestamp: 6, Payload: []byte{0x02}},
|
|
{SequenceNumber: 5008, Timestamp: 7, Payload: []byte{0x03}},
|
|
},
|
|
samples: []*media.RTCSample{},
|
|
bufferSize: 50,
|
|
},
|
|
{
|
|
message: "SampleBuilder should emit multiple valid packets",
|
|
packets: []*rtp.Packet{
|
|
{SequenceNumber: 5000, Timestamp: 1, Payload: []byte{0x01}},
|
|
{SequenceNumber: 5001, Timestamp: 2, Payload: []byte{0x02}},
|
|
{SequenceNumber: 5002, Timestamp: 3, Payload: []byte{0x03}},
|
|
{SequenceNumber: 5003, Timestamp: 4, Payload: []byte{0x04}},
|
|
{SequenceNumber: 5004, Timestamp: 5, Payload: []byte{0x05}},
|
|
{SequenceNumber: 5005, Timestamp: 6, Payload: []byte{0x06}},
|
|
},
|
|
samples: []*media.RTCSample{
|
|
{Data: []byte{0x02}, Samples: 1},
|
|
{Data: []byte{0x03}, Samples: 1},
|
|
{Data: []byte{0x04}, Samples: 1},
|
|
{Data: []byte{0x05}, Samples: 1},
|
|
},
|
|
bufferSize: 50,
|
|
},
|
|
}
|
|
|
|
func TestSampleBuilder(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
for _, t := range testCases {
|
|
s := New(t.bufferSize)
|
|
samples := []*media.RTCSample{}
|
|
|
|
for _, p := range t.packets {
|
|
s.Push(p)
|
|
}
|
|
for sample := s.Pop(); sample != nil; sample = s.Pop() {
|
|
samples = append(samples, sample)
|
|
}
|
|
|
|
assert.Equal(samples, t.samples, t.message)
|
|
}
|
|
}
|