package codecprocessor

import (
	"testing"

	"github.com/bluenviron/gortsplib/v5/pkg/format"
	mcav1 "github.com/bluenviron/mediacommon/v2/pkg/codecs/av1"
	"github.com/bluenviron/mediamtx/internal/unit"
	"github.com/stretchr/testify/require"
)

func TestAV1RemoveTUD(t *testing.T) {
	forma := &format.AV1{}

	p, err := New(1450, forma, true, nil)
	require.NoError(t, err)

	u := &unit.Unit{
		PTS: 30000,
		Payload: unit.PayloadAV1{
			{byte(mcav1.OBUTypeTemporalDelimiter) << 3},
			{5},
		},
	}

	err = p.ProcessUnit(u)
	require.NoError(t, err)

	require.Equal(t, unit.PayloadAV1{
		{5},
	}, u.Payload)
}