mirror of
https://github.com/aler9/gortsplib
synced 2025-10-05 15:16:51 +08:00
67 lines
1.2 KiB
Go
67 lines
1.2 KiB
Go
package rtpsimpleaudio
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/pion/rtp"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func uint16Ptr(v uint16) *uint16 {
|
|
return &v
|
|
}
|
|
|
|
func uint32Ptr(v uint32) *uint32 {
|
|
return &v
|
|
}
|
|
|
|
var cases = []struct {
|
|
name string
|
|
frame []byte
|
|
pkt *rtp.Packet
|
|
}{
|
|
{
|
|
"single",
|
|
[]byte{0x01, 0x02, 0x03, 0x04},
|
|
&rtp.Packet{
|
|
Header: rtp.Header{
|
|
Version: 2,
|
|
Marker: false,
|
|
PayloadType: 0,
|
|
SequenceNumber: 17645,
|
|
SSRC: 0x9dbb7812,
|
|
},
|
|
Payload: []byte{0x01, 0x02, 0x03, 0x04},
|
|
},
|
|
},
|
|
}
|
|
|
|
func TestEncode(t *testing.T) {
|
|
for _, ca := range cases {
|
|
t.Run(ca.name, func(t *testing.T) {
|
|
e := &Encoder{
|
|
PayloadType: 0,
|
|
SSRC: uint32Ptr(0x9dbb7812),
|
|
InitialSequenceNumber: uint16Ptr(0x44ed),
|
|
PayloadMaxSize: 1000,
|
|
}
|
|
err := e.Init()
|
|
require.NoError(t, err)
|
|
|
|
pkt, err := e.Encode(ca.frame)
|
|
require.NoError(t, err)
|
|
require.Equal(t, ca.pkt, pkt)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestEncodeRandomInitialState(t *testing.T) {
|
|
e := &Encoder{
|
|
PayloadType: 0,
|
|
}
|
|
err := e.Init()
|
|
require.NoError(t, err)
|
|
require.NotEqual(t, nil, e.SSRC)
|
|
require.NotEqual(t, nil, e.InitialSequenceNumber)
|
|
}
|