85 lines
2.0 KiB
Go
85 lines
2.0 KiB
Go
package rtmp
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/bluenviron/gortsplib/v4/pkg/description"
|
|
"github.com/bluenviron/gortsplib/v4/pkg/format"
|
|
"github.com/bluenviron/mediamtx/internal/logger"
|
|
"github.com/bluenviron/mediamtx/internal/protocols/rtmp/bytecounter"
|
|
"github.com/bluenviron/mediamtx/internal/protocols/rtmp/message"
|
|
"github.com/bluenviron/mediamtx/internal/stream"
|
|
"github.com/bluenviron/mediamtx/internal/test"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestFromStreamNoSupportedCodecs(t *testing.T) {
|
|
stream, err := stream.New(
|
|
512,
|
|
1460,
|
|
&description.Session{Medias: []*description.Media{{
|
|
Type: description.MediaTypeVideo,
|
|
Formats: []format.Format{&format.VP8{}},
|
|
}}},
|
|
true,
|
|
test.NilLogger,
|
|
)
|
|
require.NoError(t, err)
|
|
|
|
l := test.Logger(func(logger.Level, string, ...interface{}) {
|
|
t.Error("should not happen")
|
|
})
|
|
|
|
err = FromStream(stream, l, nil, nil, 0)
|
|
require.Equal(t, errNoSupportedCodecsFrom, err)
|
|
}
|
|
|
|
func TestFromStreamSkipUnsupportedTracks(t *testing.T) {
|
|
stream, err := stream.New(
|
|
512,
|
|
1460,
|
|
&description.Session{Medias: []*description.Media{
|
|
{
|
|
Type: description.MediaTypeVideo,
|
|
Formats: []format.Format{&format.VP8{}},
|
|
},
|
|
{
|
|
Type: description.MediaTypeVideo,
|
|
Formats: []format.Format{&format.H264{}},
|
|
},
|
|
{
|
|
Type: description.MediaTypeVideo,
|
|
Formats: []format.Format{&format.H264{}},
|
|
},
|
|
}},
|
|
true,
|
|
test.NilLogger,
|
|
)
|
|
require.NoError(t, err)
|
|
|
|
n := 0
|
|
|
|
l := test.Logger(func(l logger.Level, format string, args ...interface{}) {
|
|
require.Equal(t, logger.Warn, l)
|
|
switch n {
|
|
case 0:
|
|
require.Equal(t, "skipping track 1 (VP8)", fmt.Sprintf(format, args...))
|
|
case 1:
|
|
require.Equal(t, "skipping track 3 (H264)", fmt.Sprintf(format, args...))
|
|
}
|
|
n++
|
|
})
|
|
|
|
var buf bytes.Buffer
|
|
bc := bytecounter.NewReadWriter(&buf)
|
|
conn := &Conn{mrw: message.NewReadWriter(&buf, bc, false)}
|
|
|
|
err = FromStream(stream, l, conn, nil, 0)
|
|
require.NoError(t, err)
|
|
defer stream.RemoveReader(l)
|
|
|
|
require.Equal(t, 2, n)
|
|
}
|