mediamtx/internal/conf/env/env_test.go

package env

import (
	"encoding/json"
	"os"
	"testing"
	"time"

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

type subStruct struct {
	MyParam int `json:"myParam"`
}

type mapEntry struct {
	MyValue  string    `json:"myValue"`
	MyStruct subStruct `json:"myStruct"`
}

type myDuration time.Duration

func (d *myDuration) UnmarshalJSON(b []byte) error {
	var in string
	if err := json.Unmarshal(b, &in); err != nil {
		return err
	}

	du, err := time.ParseDuration(in)
	if err != nil {
		return err
	}
	*d = myDuration(du)

	return nil
}

// UnmarshalEnv implements env.Unmarshaler.
func (d *myDuration) UnmarshalEnv(_ string, v string) error {
	return d.UnmarshalJSON([]byte(`"` + v + `"`))
}

type mySubStruct struct {
	URL      string `json:"url"`
	Username string `json:"username"`
	Password string `json:"password"`
}

type testStruct struct {
	MyString              string               `json:"myString"`
	MyInt                 int                  `json:"myInt"`
	MyFloat               float64              `json:"myFloat"`
	MyBool                bool                 `json:"myBool"`
	MyDuration            myDuration           `json:"myDuration"`
	MyMap                 map[string]*mapEntry `json:"myMap"`
	MySlice               []string             `json:"mySlice"`
	MySliceEmpty          []string             `json:"mySliceEmpty"`
	MySliceSubStruct      []mySubStruct        `json:"mySliceSubStruct"`
	MySliceSubStructEmpty []mySubStruct        `json:"mySliceSubStructEmpty"`
}

func TestLoad(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_MYFLOAT", "15.2")
	defer os.Unsetenv("MYPREFIX_MYFLOAT")

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

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

	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")

	os.Setenv("MYPREFIX_MYMAP_MYKEY2_MYSTRUCT_MYPARAM", "456")
	defer os.Unsetenv("MYPREFIX_MYMAP_MYKEY2_MYSTRUCT_MYPARAM")

	os.Setenv("MYPREFIX_MYSLICE", "val1,val2")
	defer os.Unsetenv("MYPREFIX_MYSLICE")

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

	os.Setenv("MYPREFIX_MYSLICESUBSTRUCT_0_URL", "url1")
	defer os.Unsetenv("MYPREFIX_MYSLICESUBSTRUCT_0_URL")

	os.Setenv("MYPREFIX_MYSLICESUBSTRUCT_0_USERNAME", "user1")
	defer os.Unsetenv("MYPREFIX_MYSLICESUBSTRUCT_0_USERNAME")

	os.Setenv("MYPREFIX_MYSLICESUBSTRUCT_0_PASSWORD", "pass1")
	defer os.Unsetenv("MYPREFIX_MYSLICESUBSTRUCT_0_PASSWORD")

	os.Setenv("MYPREFIX_MYSLICESUBSTRUCT_1_URL", "url2")
	defer os.Unsetenv("MYPREFIX_MYSLICESUBSTRUCT_1_URL")

	os.Setenv("MYPREFIX_MYSLICESUBSTRUCT_1_PASSWORD", "pass2")
	defer os.Unsetenv("MYPREFIX_MYSLICESUBSTRUCT_1_PASSWORD")

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

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

	require.Equal(t, "testcontent", s.MyString)
	require.Equal(t, 123, s.MyInt)
	require.Equal(t, 15.2, s.MyFloat)
	require.Equal(t, true, s.MyBool)
	require.Equal(t, 22*myDuration(time.Second), s.MyDuration)

	_, 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)
	require.Equal(t, 456, v.MyStruct.MyParam)

	require.Equal(t, []string{"val1", "val2"}, s.MySlice)
	require.Equal(t, []string{}, s.MySliceEmpty)

	require.Equal(t, []mySubStruct{
		{
			URL:      "url1",
			Username: "user1",
			Password: "pass1",
		},
		{
			URL:      "url2",
			Password: "pass2",
		},
	}, s.MySliceSubStruct)

	require.Equal(t, []mySubStruct{}, s.MySliceSubStructEmpty)
}