mediamtx/internal/conf/env_test.go

package conf

import (
	"os"
	"testing"
	"time"

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

type mapEntry struct {
	MyValue string
}

type testStruct struct {
	// string
	MyString string

	// int
	MyInt int

	// bool
	MyBool bool

	// duration
	MyDuration StringDuration

	// slice
	MySlice []string

	// map
	MyMap map[string]*mapEntry
}

func TestEnvironment(t *testing.T) {
	os.Setenv("MYPREFIX_MYSTRING", "testcontent")
	defer os.Unsetenv("MYPREFIX_MYSTRING")

	os.Setenv("MYPREFIX_MYINT", "123")
	defer os.Unsetenv("MYPREFIX_MYINT")

	os.Setenv("MYPREFIX_MYBOOL", "yes")
	defer os.Unsetenv("MYPREFIX_MYBOOL")

	os.Setenv("MYPREFIX_MYDURATION", "22s")
	defer os.Unsetenv("MYPREFIX_MYDURATION")

	os.Setenv("MYPREFIX_MYSLICE", "el1,el2")
	defer os.Unsetenv("MYPREFIX_MYSLICE")

	os.Setenv("MYPREFIX_MYMAP_MYKEY", "")
	defer os.Unsetenv("MYPREFIX_MYMAP_MYKEY")

	os.Setenv("MYPREFIX_MYMAP_MYKEY2_MYVALUE", "asd")
	defer os.Unsetenv("MYPREFIX_MYMAP_MYKEY2_MYVALUE")

	var s testStruct
	err := loadFromEnvironment("MYPREFIX", &s)
	require.NoError(t, err)

	require.Equal(t, "testcontent", s.MyString)
	require.Equal(t, 123, s.MyInt)
	require.Equal(t, true, s.MyBool)
	require.Equal(t, 22*StringDuration(time.Second), s.MyDuration)
	require.Equal(t, []string{"el1", "el2"}, s.MySlice)

	_, ok := s.MyMap["mykey"]
	require.Equal(t, true, ok)

	v, ok := s.MyMap["mykey2"]
	require.Equal(t, true, ok)
	require.Equal(t, "asd", v.MyValue)
}