mirror of
https://github.com/aler9/gortsplib
synced 2025-10-04 23:02:45 +08:00
74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
package gortsplib
|
|
|
|
import (
|
|
"testing"
|
|
|
|
psdp "github.com/pion/sdp/v3"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestTrackH265Attributes(t *testing.T) {
|
|
track := &TrackH265{
|
|
PayloadType: 96,
|
|
VPS: []byte{0x01, 0x02},
|
|
SPS: []byte{0x03, 0x04},
|
|
PPS: []byte{0x05, 0x06},
|
|
}
|
|
require.Equal(t, 90000, track.ClockRate())
|
|
require.Equal(t, "", track.GetControl())
|
|
require.Equal(t, []byte{0x01, 0x02}, track.SafeVPS())
|
|
require.Equal(t, []byte{0x03, 0x04}, track.SafeSPS())
|
|
require.Equal(t, []byte{0x05, 0x06}, track.SafePPS())
|
|
|
|
track.SafeSetVPS([]byte{0x07, 0x08})
|
|
track.SafeSetSPS([]byte{0x09, 0x0A})
|
|
track.SafeSetPPS([]byte{0x0B, 0x0C})
|
|
require.Equal(t, []byte{0x07, 0x08}, track.SafeVPS())
|
|
require.Equal(t, []byte{0x09, 0x0A}, track.SafeSPS())
|
|
require.Equal(t, []byte{0x0B, 0x0C}, track.SafePPS())
|
|
}
|
|
|
|
func TestTrackH265Clone(t *testing.T) {
|
|
track := &TrackH265{
|
|
PayloadType: 96,
|
|
VPS: []byte{0x01, 0x02},
|
|
SPS: []byte{0x03, 0x04},
|
|
PPS: []byte{0x05, 0x06},
|
|
}
|
|
|
|
clone := track.clone()
|
|
require.NotSame(t, track, clone)
|
|
require.Equal(t, track, clone)
|
|
}
|
|
|
|
func TestTrackH265MediaDescription(t *testing.T) {
|
|
track := &TrackH265{
|
|
PayloadType: 96,
|
|
VPS: []byte{0x01, 0x02},
|
|
SPS: []byte{0x03, 0x04},
|
|
PPS: []byte{0x05, 0x06},
|
|
}
|
|
|
|
require.Equal(t, &psdp.MediaDescription{
|
|
MediaName: psdp.MediaName{
|
|
Media: "video",
|
|
Protos: []string{"RTP", "AVP"},
|
|
Formats: []string{"96"},
|
|
},
|
|
Attributes: []psdp.Attribute{
|
|
{
|
|
Key: "rtpmap",
|
|
Value: "96 H265/90000",
|
|
},
|
|
{
|
|
Key: "fmtp",
|
|
Value: "96 sprop-vps=AQI=; sprop-sps=AwQ=; sprop-pps=BQY=",
|
|
},
|
|
{
|
|
Key: "control",
|
|
Value: "",
|
|
},
|
|
},
|
|
}, track.MediaDescription())
|
|
}
|