mirror of
https://github.com/bluenviron/mediamtx.git
synced 2026-01-24 20:39:15 -08:00
1b9dfbd367
Some checks are pending
code_lint / golangci_lint (push) Waiting to run
code_lint / mod_tidy (push) Waiting to run
code_lint / api_docs (push) Waiting to run
code_test / test_64 (push) Waiting to run
code_test / test_32 (push) Waiting to run
code_test / test_e2e (push) Waiting to run
106 lines
2.5 KiB
Go
106 lines
2.5 KiB
Go
package rtmp
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/bluenviron/gortsplib/v4/pkg/format"
|
|
"github.com/bluenviron/mediacommon/v2/pkg/codecs/mpeg4audio"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/bluenviron/mediamtx/internal/protocols/rtmp/amf0"
|
|
"github.com/bluenviron/mediamtx/internal/protocols/rtmp/bytecounter"
|
|
"github.com/bluenviron/mediamtx/internal/protocols/rtmp/message"
|
|
)
|
|
|
|
func TestWriteTracks(t *testing.T) {
|
|
videoTrack := &format.H264{
|
|
PayloadTyp: 96,
|
|
SPS: []byte{
|
|
0x67, 0x64, 0x00, 0x0c, 0xac, 0x3b, 0x50, 0xb0,
|
|
0x4b, 0x42, 0x00, 0x00, 0x03, 0x00, 0x02, 0x00,
|
|
0x00, 0x03, 0x00, 0x3d, 0x08,
|
|
},
|
|
PPS: []byte{
|
|
0x68, 0xee, 0x3c, 0x80,
|
|
},
|
|
PacketizationMode: 1,
|
|
}
|
|
|
|
audioTrack := &format.MPEG4Audio{
|
|
PayloadTyp: 96,
|
|
Config: &mpeg4audio.Config{
|
|
Type: 2,
|
|
SampleRate: 44100,
|
|
ChannelCount: 2,
|
|
},
|
|
SizeLength: 13,
|
|
IndexLength: 3,
|
|
IndexDeltaLength: 3,
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
c := &dummyConn{
|
|
rw: &buf,
|
|
}
|
|
c.initialize()
|
|
|
|
w := &Writer{
|
|
Conn: c,
|
|
VideoTrack: videoTrack,
|
|
AudioTrack: audioTrack,
|
|
}
|
|
err := w.Initialize()
|
|
require.NoError(t, err)
|
|
|
|
bc := bytecounter.NewReadWriter(&buf)
|
|
mrw := message.NewReadWriter(bc, bc, true)
|
|
|
|
msg, err := mrw.Read()
|
|
require.NoError(t, err)
|
|
require.Equal(t, &message.DataAMF0{
|
|
ChunkStreamID: 4,
|
|
MessageStreamID: 0x1000000,
|
|
Payload: []interface{}{
|
|
"@setDataFrame",
|
|
"onMetaData",
|
|
amf0.Object{
|
|
{Key: "videodatarate", Value: float64(0)},
|
|
{Key: "videocodecid", Value: float64(7)},
|
|
{Key: "audiodatarate", Value: float64(0)},
|
|
{Key: "audiocodecid", Value: float64(10)},
|
|
},
|
|
},
|
|
}, msg)
|
|
|
|
msg, err = mrw.Read()
|
|
require.NoError(t, err)
|
|
require.Equal(t, &message.Video{
|
|
ChunkStreamID: message.VideoChunkStreamID,
|
|
MessageStreamID: 0x1000000,
|
|
Codec: message.CodecH264,
|
|
IsKeyFrame: true,
|
|
Type: message.VideoTypeConfig,
|
|
Payload: []byte{
|
|
0x1, 0x64, 0x0,
|
|
0xc, 0xff, 0xe1, 0x0, 0x15, 0x67, 0x64, 0x0,
|
|
0xc, 0xac, 0x3b, 0x50, 0xb0, 0x4b, 0x42, 0x0,
|
|
0x0, 0x3, 0x0, 0x2, 0x0, 0x0, 0x3, 0x0,
|
|
0x3d, 0x8, 0x1, 0x0, 0x4, 0x68, 0xee, 0x3c,
|
|
0x80,
|
|
},
|
|
}, msg)
|
|
|
|
msg, err = mrw.Read()
|
|
require.NoError(t, err)
|
|
require.Equal(t, &message.Audio{
|
|
ChunkStreamID: message.AudioChunkStreamID,
|
|
MessageStreamID: 0x1000000,
|
|
Codec: message.CodecMPEG4Audio,
|
|
Rate: message.Rate44100,
|
|
Depth: message.Depth16,
|
|
IsStereo: true,
|
|
AACType: message.AudioAACTypeConfig,
|
|
Payload: []byte{0x12, 0x10},
|
|
}, msg)
|
|
}
|