mirror of
https://github.com/bluenviron/mediamtx
synced 2024-12-14 02:34:52 +00:00
205 lines
5.6 KiB
Go
205 lines
5.6 KiB
Go
package env
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func stringPtr(v string) *string {
|
|
return &v
|
|
}
|
|
|
|
func intPtr(v int) *int {
|
|
return &v
|
|
}
|
|
|
|
func uint64Ptr(v uint64) *uint64 {
|
|
return &v
|
|
}
|
|
|
|
func boolPtr(v bool) *bool {
|
|
return &v
|
|
}
|
|
|
|
func float64Ptr(v float64) *float64 {
|
|
return &v
|
|
}
|
|
|
|
func durationPtr(v time.Duration) *time.Duration {
|
|
return &v
|
|
}
|
|
|
|
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 subStruct struct {
|
|
MyParam int `json:"myParam"`
|
|
}
|
|
|
|
type mapEntry struct {
|
|
MyValue string `json:"myValue"`
|
|
MyStruct subStruct `json:"myStruct"`
|
|
}
|
|
|
|
type mySubStruct struct {
|
|
URL string `json:"url"`
|
|
Username string `json:"username"`
|
|
Password string `json:"password"`
|
|
MyInt2 int `json:"myInt2"`
|
|
}
|
|
|
|
type testStruct struct {
|
|
MyString string `json:"myString"`
|
|
MyStringOpt *string `json:"myStringOpt"`
|
|
MyInt int `json:"myInt"`
|
|
MyIntOpt *int `json:"myIntOpt"`
|
|
MyUint uint64 `json:"myUint"`
|
|
MyUintOpt *uint64 `json:"myUintOpt"`
|
|
MyFloat float64 `json:"myFloat"`
|
|
MyFloatOpt *float64 `json:"myFloatOpt"`
|
|
MyBool bool `json:"myBool"`
|
|
MyBoolOpt *bool `json:"myBoolOpt"`
|
|
MyDuration myDuration `json:"myDuration"`
|
|
MyDurationOpt *myDuration `json:"myDurationOpt"`
|
|
MyDurationOptUnset *myDuration `json:"myDurationOptUnset"`
|
|
MyMap map[string]*mapEntry `json:"myMap"`
|
|
MySliceString []string `json:"mySliceString"`
|
|
MySliceStringEmpty []string `json:"mySliceStringEmpty"`
|
|
MySliceStringOpt *[]string `json:"mySliceStringOpt"`
|
|
MySliceStringOptUnset *[]string `json:"mySliceStringOptUnset"`
|
|
MySliceSubStruct []mySubStruct `json:"mySliceSubStruct"`
|
|
MySliceSubStructEmpty []mySubStruct `json:"mySliceSubStructEmpty"`
|
|
MySliceSubStructOpt *[]mySubStruct `json:"mySliceSubStructOpt"`
|
|
MySliceSubStructOptUnset *[]mySubStruct `json:"mySliceSubStructOptUnset"`
|
|
Unset *bool `json:"unset"`
|
|
}
|
|
|
|
func TestLoad(t *testing.T) {
|
|
env := map[string]string{
|
|
"MYPREFIX_MYSTRING": "testcontent",
|
|
"MYPREFIX_MYSTRINGOPT": "testcontent2",
|
|
"MYPREFIX_MYINT": "123",
|
|
"MYPREFIX_MYINTOPT": "456",
|
|
"MYPREFIX_MYUINT": "8910",
|
|
"MYPREFIX_MYUINTOPT": "112313",
|
|
"MYPREFIX_MYFLOAT": "15.2",
|
|
"MYPREFIX_MYFLOATOPT": "16.2",
|
|
"MYPREFIX_MYBOOL": "yes",
|
|
"MYPREFIX_MYBOOLOPT": "false",
|
|
"MYPREFIX_MYDURATION": "22s",
|
|
"MYPREFIX_MYDURATIONOPT": "30s",
|
|
"MYPREFIX_MYMAP_MYKEY": "",
|
|
"MYPREFIX_MYMAP_MYKEY2_MYVALUE": "asd",
|
|
"MYPREFIX_MYMAP_MYKEY2_MYSTRUCT_MYPARAM": "456",
|
|
"MYPREFIX_MYSLICESTRING": "val1,val2",
|
|
"MYPREFIX_MYSLICESTRINGEMPTY": "",
|
|
"MYPREFIX_MYSLICESTRINGOPT": "aa",
|
|
"MYPREFIX_MYSLICESUBSTRUCT_0_URL": "url1",
|
|
"MYPREFIX_MYSLICESUBSTRUCT_0_USERNAME": "user1",
|
|
"MYPREFIX_MYSLICESUBSTRUCT_0_PASSWORD": "pass1",
|
|
"MYPREFIX_MYSLICESUBSTRUCT_1_URL": "url2",
|
|
"MYPREFIX_MYSLICESUBSTRUCT_1_PASSWORD": "pass2",
|
|
"MYPREFIX_MYSLICESUBSTRUCTEMPTY": "",
|
|
"MYPREFIX_MYSLICESUBSTRUCTOPT_1_PASSWORD": "pwd",
|
|
}
|
|
|
|
for key, val := range env {
|
|
os.Setenv(key, val)
|
|
defer os.Unsetenv(key)
|
|
}
|
|
|
|
var s testStruct
|
|
err := Load("MYPREFIX", &s)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, testStruct{
|
|
MyString: "testcontent",
|
|
MyStringOpt: stringPtr("testcontent2"),
|
|
MyInt: 123,
|
|
MyIntOpt: intPtr(456),
|
|
MyUint: 8910,
|
|
MyUintOpt: uint64Ptr(112313),
|
|
MyFloat: 15.2,
|
|
MyFloatOpt: float64Ptr(16.2),
|
|
MyBool: true,
|
|
MyBoolOpt: boolPtr(false),
|
|
MyDuration: 22000000000,
|
|
MyDurationOpt: (*myDuration)(durationPtr(30000000000)),
|
|
MyMap: map[string]*mapEntry{
|
|
"mykey": {
|
|
MyValue: "",
|
|
MyStruct: subStruct{
|
|
MyParam: 0,
|
|
},
|
|
},
|
|
"mykey2": {
|
|
MyValue: "asd",
|
|
MyStruct: subStruct{
|
|
MyParam: 456,
|
|
},
|
|
},
|
|
},
|
|
MySliceString: []string{
|
|
"val1",
|
|
"val2",
|
|
},
|
|
MySliceStringEmpty: []string{},
|
|
MySliceStringOpt: &[]string{"aa"},
|
|
MySliceSubStruct: []mySubStruct{
|
|
{
|
|
URL: "url1",
|
|
Username: "user1",
|
|
Password: "pass1",
|
|
},
|
|
{
|
|
URL: "url2",
|
|
Username: "",
|
|
Password: "pass2",
|
|
},
|
|
},
|
|
MySliceSubStructEmpty: []mySubStruct{},
|
|
}, s)
|
|
}
|
|
|
|
func FuzzLoad(f *testing.F) {
|
|
f.Add("MYPREFIX_MYINT", "a")
|
|
f.Add("MYPREFIX_MYUINT", "a")
|
|
f.Add("MYPREFIX_MYFLOAT", "a")
|
|
f.Add("MYPREFIX_MYBOOL", "a")
|
|
f.Add("MYPREFIX_MYSLICESUBSTRUCT_0_MYINT2", "a")
|
|
f.Add("MYPREFIX_MYDURATION", "a")
|
|
f.Add("MYPREFIX_MYDURATION_A", "a")
|
|
|
|
f.Fuzz(func(t *testing.T, key string, val string) {
|
|
env := map[string]string{
|
|
key: val,
|
|
}
|
|
|
|
var s testStruct
|
|
loadWithEnv(env, "MYPREFIX", &s) //nolint:errcheck
|
|
})
|
|
}
|