gortsplib/pkg/codecs/mpeg4audio/config_test.go

package mpeg4audio

import (
	"testing"

	"github.com/stretchr/testify/require"
)

var configCases = []struct {
	name string
	enc  []byte
	dec  Config
}{
	{
		"aac-lc 16khz mono",
		[]byte{0x14, 0x08},
		Config{
			Type:         ObjectTypeAACLC,
			SampleRate:   16000,
			ChannelCount: 1,
		},
	},
	{
		"aac-lc 44.1khz mono",
		[]byte{0x12, 0x08},
		Config{
			Type:         ObjectTypeAACLC,
			SampleRate:   44100,
			ChannelCount: 1,
		},
	},
	{
		"aac-lc 44.1khz 5.1",
		[]byte{0x12, 0x30},
		Config{
			Type:         ObjectTypeAACLC,
			SampleRate:   44100,
			ChannelCount: 6,
		},
	},
	{
		"aac-lc 48khz stereo",
		[]byte{17, 144},
		Config{
			Type:         ObjectTypeAACLC,
			SampleRate:   48000,
			ChannelCount: 2,
		},
	},
	{
		"aac-lc 53khz stereo",
		[]byte{0x17, 0x80, 0x67, 0x84, 0x10},
		Config{
			Type:         ObjectTypeAACLC,
			SampleRate:   53000,
			ChannelCount: 2,
		},
	},
	{
		"aac-lc 96khz stereo delay",
		[]byte{0x10, 0x12, 0x0c, 0x08},
		Config{
			Type:               ObjectTypeAACLC,
			SampleRate:         96000,
			ChannelCount:       2,
			DependsOnCoreCoder: true,
			CoreCoderDelay:     385,
		},
	},
	{
		"aac-lc 44.1khz 8 chans",
		[]byte{0x12, 0x38},
		Config{
			Type:         ObjectTypeAACLC,
			SampleRate:   44100,
			ChannelCount: 8,
		},
	},
	{
		"sbr (he-aac v1) 44.1khz mono",
		[]byte{0x2b, 0x8a, 0x08, 0x00},
		Config{
			Type:                ObjectTypeAACLC,
			SampleRate:          22050,
			ChannelCount:        1,
			ExtensionSampleRate: 44100,
			ExtensionType:       ObjectTypeSBR,
		},
	},
	{
		"sbr (he-aac v1) 44.1khz stereo",
		[]byte{0x2b, 0x92, 0x08, 0x00}, // the data from fdk_aac
		Config{
			Type:                ObjectTypeAACLC,
			SampleRate:          22050,
			ChannelCount:        2,
			ExtensionSampleRate: 44100,
			ExtensionType:       ObjectTypeSBR,
		},
	},
	{
		"ps (he-aac v2) 48khz stereo",
		[]byte{0xeb, 0x09, 0x88, 0x00}, // the data from fdk_aac
		Config{
			Type:                ObjectTypeAACLC,
			SampleRate:          24000,
			ChannelCount:        1, // for he_aac v2, ChannelCount only set to 1 ?
			ExtensionSampleRate: 48000,
			ExtensionType:       ObjectTypePS,
		},
	},
}

func TestConfigUnmarshal(t *testing.T) {
	for _, ca := range configCases {
		t.Run(ca.name, func(t *testing.T) {
			var dec Config
			err := dec.Unmarshal(ca.enc)
			require.NoError(t, err)
			require.Equal(t, ca.dec, dec)
		})
	}
}

func TestConfigUnmarshalErrors(t *testing.T) {
	for _, ca := range []struct {
		name string
		enc  []byte
		err  string
	}{
		{
			"empty",
			[]byte{},
			"not enough bits",
		},
		{
			"unsupported object type",
			[]byte{0xF1},
			"unsupported object type: 30",
		},
		{
			"no sample rate index",
			[]byte{0b00010000},
			"not enough bits",
		},
		{
			"invalid sample rate index",
			[]byte{0b00010110, 0b10000000},
			"invalid sample rate index (13)",
		},
		{
			"channel config 0",
			[]byte{0b00010100, 0b00000000},
			"not yet supported",
		},
	} {
		t.Run(ca.name, func(t *testing.T) {
			var dec Config
			err := dec.Unmarshal(ca.enc)
			require.EqualError(t, err, ca.err)
		})
	}
}

func TestConfigMarshal(t *testing.T) {
	for _, ca := range configCases {
		t.Run(ca.name, func(t *testing.T) {
			enc, err := ca.dec.Marshal()
			require.NoError(t, err)
			require.Equal(t, ca.enc, enc)
		})
	}
}

func TestConfigMarshalErrors(t *testing.T) {
	for _, ca := range []struct {
		name string
		conf Config
		err  string
	}{
		{
			"invalid channel config",
			Config{
				Type:         ObjectTypeAACLC,
				SampleRate:   44100,
				ChannelCount: 0,
			},
			"invalid channel count (0)",
		},
	} {
		t.Run(ca.name, func(t *testing.T) {
			_, err := ca.conf.Marshal()
			require.EqualError(t, err, ca.err)
		})
	}
}