package udp import ( "bufio" "net" "testing" "time" "github.com/bluenviron/mediacommon/pkg/formats/mpegts" "github.com/stretchr/testify/require" "github.com/bluenviron/mediamtx/internal/conf" "github.com/bluenviron/mediamtx/internal/defs" "github.com/bluenviron/mediamtx/internal/staticsources/tester" ) func TestSource(t *testing.T) { te := tester.New( func(p defs.StaticSourceParent) defs.StaticSource { return &Source{ ReadTimeout: conf.StringDuration(10 * time.Second), Parent: p, } }, &conf.Path{ Source: "udp://localhost:9001", }, ) defer te.Close() time.Sleep(50 * time.Millisecond) conn, err := net.Dial("udp", "localhost:9001") require.NoError(t, err) defer conn.Close() track := &mpegts.Track{ Codec: &mpegts.CodecH264{}, } bw := bufio.NewWriter(conn) w := mpegts.NewWriter(bw, []*mpegts.Track{track}) require.NoError(t, err) err = w.WriteH26x(track, 0, 0, true, [][]byte{{ // IDR 5, 1, }}) require.NoError(t, err) err = w.WriteH26x(track, 0, 0, true, [][]byte{{ // non-IDR 5, 2, }}) require.NoError(t, err) err = bw.Flush() require.NoError(t, err) <-te.Unit }