gortsplib/pkg/headers/keyval_test.go

package headers

import (
	"testing"

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

func TestKeyValParse(t *testing.T) {
	for _, ca := range []struct {
		name string
		s    string
		kvs  map[string]string
	}{
		{
			"base",
			`key1=v1,key2=v2`,
			map[string]string{
				"key1": "v1",
				"key2": "v2",
			},
		},
		{
			"with space",
			`key1=v1, key2=v2`,
			map[string]string{
				"key1": "v1",
				"key2": "v2",
			},
		},
		{
			"with apexes",
			`key1="v1", key2=v2`,
			map[string]string{
				"key1": "v1",
				"key2": "v2",
			},
		},
		{
			"with apexes and comma",
			`key1="v,1", key2="v2"`,
			map[string]string{
				"key1": "v,1",
				"key2": "v2",
			},
		},
		{
			"with apexes and equal",
			`key1="v=1", key2="v2"`,
			map[string]string{
				"key1": "v=1",
				"key2": "v2",
			},
		},
	} {
		t.Run(ca.name, func(t *testing.T) {
			kvs, err := keyValParse(ca.s, ',')
			require.NoError(t, err)
			require.Equal(t, ca.kvs, kvs)
		})
	}
}

func TestKeyValParseError(t *testing.T) {
	for _, ca := range []struct {
		name string
		s    string
		err  string
	}{
		{
			"apexes not closed",
			`key1="v,1`,
			"apexes not closed (\"v,1)",
		},
		{
			"no key",
			`value`,
			"unable to find key (value)",
		},
	} {
		t.Run(ca.name, func(t *testing.T) {
			_, err := keyValParse(ca.s, ',')
			require.Equal(t, ca.err, err.Error())
		})
	}
}