mirror of
https://github.com/Luzifer/rconfig.git
synced 2024-11-08 07:50:07 +00:00
86 lines
2.6 KiB
Go
86 lines
2.6 KiB
Go
package rconfig
|
|
|
|
import (
|
|
"os"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestVardefaultParsing(t *testing.T) {
|
|
type test struct {
|
|
MySecretValue string `default:"secret" env:"foo" vardefault:"my_secret_value"`
|
|
MyUsername string `default:"luzifer" vardefault:"username"`
|
|
SomeVar string `flag:"var" description:"some variable"`
|
|
IntVar int64 `vardefault:"int_var" default:"23"`
|
|
}
|
|
|
|
var (
|
|
cfg test
|
|
args = []string{}
|
|
err error
|
|
vardefaults = map[string]string{
|
|
"my_secret_value": "veryverysecretkey",
|
|
"unkownkey": "hi there",
|
|
"int_var": "42",
|
|
}
|
|
)
|
|
|
|
exec := func(desc string, tests [][2]interface{}) {
|
|
require.NoError(t, parse(&cfg, args))
|
|
|
|
for _, test := range tests {
|
|
assert.Equal(t, test[1], reflect.ValueOf(test[0]).Elem().Interface(), desc)
|
|
}
|
|
}
|
|
|
|
SetVariableDefaults(vardefaults)
|
|
exec("manually provided variables", [][2]interface{}{
|
|
{&cfg.IntVar, int64(42)},
|
|
{&cfg.MySecretValue, "veryverysecretkey"},
|
|
{&cfg.MyUsername, "luzifer"},
|
|
{&cfg.SomeVar, ""},
|
|
})
|
|
|
|
SetVariableDefaults(VarDefaultsFromYAML([]byte("---\nmy_secret_value: veryverysecretkey\nunknownkey: hi there\nint_var: 42\n")))
|
|
exec("defaults from YAML data", [][2]interface{}{
|
|
{&cfg.IntVar, int64(42)},
|
|
{&cfg.MySecretValue, "veryverysecretkey"},
|
|
{&cfg.MyUsername, "luzifer"},
|
|
{&cfg.SomeVar, ""},
|
|
})
|
|
|
|
tmp, _ := os.CreateTemp("", "")
|
|
t.Cleanup(func() {
|
|
tmp.Close() //nolint:errcheck,gosec,revive // Just cleanup, will be closed automatically
|
|
os.Remove(tmp.Name()) //nolint:errcheck,gosec,revive // Just cleanup of tmp-file
|
|
})
|
|
yamlData := "---\nmy_secret_value: veryverysecretkey\nunknownkey: hi there\nint_var: 42\n"
|
|
_, err = tmp.WriteString(yamlData)
|
|
require.NoError(t, err)
|
|
SetVariableDefaults(VarDefaultsFromYAMLFile(tmp.Name()))
|
|
exec("defaults from YAML file", [][2]interface{}{
|
|
{&cfg.IntVar, int64(42)},
|
|
{&cfg.MySecretValue, "veryverysecretkey"},
|
|
{&cfg.MyUsername, "luzifer"},
|
|
{&cfg.SomeVar, ""},
|
|
})
|
|
|
|
SetVariableDefaults(VarDefaultsFromYAML([]byte("---\nmy_secret_value = veryverysecretkey\nunknownkey = hi there\nint_var = 42\n")))
|
|
exec("defaults from invalid YAML data", [][2]interface{}{
|
|
{&cfg.IntVar, int64(23)},
|
|
{&cfg.MySecretValue, "secret"},
|
|
{&cfg.MyUsername, "luzifer"},
|
|
{&cfg.SomeVar, ""},
|
|
})
|
|
|
|
SetVariableDefaults(VarDefaultsFromYAMLFile("/tmp/this_file_should_not_exist_146e26723r"))
|
|
exec("defaults from non-existing YAML file", [][2]interface{}{
|
|
{&cfg.IntVar, int64(23)},
|
|
{&cfg.MySecretValue, "secret"},
|
|
{&cfg.MyUsername, "luzifer"},
|
|
{&cfg.SomeVar, ""},
|
|
})
|
|
}
|