131 lines
2.4 KiB
Go
131 lines
2.4 KiB
Go
package flags
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"path"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/mitchellh/mapstructure"
|
|
)
|
|
|
|
func TestConfigUtil_Values(t *testing.T) {
|
|
t.Parallel()
|
|
type config struct {
|
|
B BoolValue `mapstructure:"bool"`
|
|
D DurationValue `mapstructure:"duration"`
|
|
S StringValue `mapstructure:"string"`
|
|
U UintValue `mapstructure:"uint"`
|
|
}
|
|
|
|
cases := []struct {
|
|
in string
|
|
success string
|
|
failure string
|
|
}{
|
|
{
|
|
`{ }`,
|
|
`"false" "0s" "" "0"`,
|
|
"",
|
|
},
|
|
{
|
|
`{ "bool": true, "duration": "2h", "string": "hello", "uint": 23 }`,
|
|
`"true" "2h0m0s" "hello" "23"`,
|
|
"",
|
|
},
|
|
{
|
|
`{ "bool": "nope" }`,
|
|
"",
|
|
"got 'string'",
|
|
},
|
|
{
|
|
`{ "duration": "nope" }`,
|
|
"",
|
|
`invalid duration "nope"`,
|
|
},
|
|
{
|
|
`{ "string": 123 }`,
|
|
"",
|
|
"got 'float64'",
|
|
},
|
|
{
|
|
`{ "uint": -1 }`,
|
|
"",
|
|
"value cannot be negative",
|
|
},
|
|
{
|
|
`{ "uint": 4294967296 }`,
|
|
"",
|
|
"value is too large",
|
|
},
|
|
}
|
|
for i, c := range cases {
|
|
var raw interface{}
|
|
dec := json.NewDecoder(bytes.NewBufferString(c.in))
|
|
if err := dec.Decode(&raw); err != nil {
|
|
t.Fatalf("(case %d) err: %v", i, err)
|
|
}
|
|
|
|
var r config
|
|
msdec, err := mapstructure.NewDecoder(&mapstructure.DecoderConfig{
|
|
DecodeHook: ConfigDecodeHook,
|
|
Result: &r,
|
|
ErrorUnused: true,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("(case %d) err: %v", i, err)
|
|
}
|
|
|
|
err = msdec.Decode(raw)
|
|
if c.failure != "" {
|
|
if err == nil || !strings.Contains(err.Error(), c.failure) {
|
|
t.Fatalf("(case %d) err: %v", i, err)
|
|
}
|
|
continue
|
|
}
|
|
if err != nil {
|
|
t.Fatalf("(case %d) err: %v", i, err)
|
|
}
|
|
|
|
actual := fmt.Sprintf("%q %q %q %q",
|
|
r.B.String(),
|
|
r.D.String(),
|
|
r.S.String(),
|
|
r.U.String())
|
|
if actual != c.success {
|
|
t.Fatalf("(case %d) bad: %s", i, actual)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestConfigUtil_Visit(t *testing.T) {
|
|
t.Parallel()
|
|
var trail []string
|
|
visitor := func(path string) error {
|
|
trail = append(trail, path)
|
|
return nil
|
|
}
|
|
|
|
basePath := "../../test/command/merge"
|
|
if err := Visit(basePath, visitor); err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
if err := Visit(path.Join(basePath, "subdir", "c.json"), visitor); err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
expected := []string{
|
|
path.Join(basePath, "a.json"),
|
|
path.Join(basePath, "b.json"),
|
|
path.Join(basePath, "nope"),
|
|
path.Join(basePath, "zero.json"),
|
|
path.Join(basePath, "subdir", "c.json"),
|
|
}
|
|
if !reflect.DeepEqual(trail, expected) {
|
|
t.Fatalf("bad: %#v", trail)
|
|
}
|
|
}
|