mirror of
https://github.com/bluenviron/mediamtx
synced 2024-12-13 18:24:58 +00:00
146 lines
3.8 KiB
Go
146 lines
3.8 KiB
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)
|
|
}
|