diff --git a/viper.go b/viper.go index 01471d9..cfadfc6 100644 --- a/viper.go +++ b/viper.go @@ -223,7 +223,7 @@ func MarshallReader(in io.Reader) { buf.ReadFrom(in) switch getConfigType() { - case "yaml": + case "yaml", "yml": if err := yaml.Unmarshal(buf.Bytes(), &config); err != nil { jww.ERROR.Fatalf("Error parsing config: %s", err) } @@ -399,7 +399,7 @@ func Reset() { configName = "config" // extensions Supported - SupportedExts = []string{"json", "toml", "yaml"} + SupportedExts = []string{"json", "toml", "yaml", "yml"} configFile = "" configType = "" diff --git a/viper_test.go b/viper_test.go index 01beac9..c5fe738 100644 --- a/viper_test.go +++ b/viper_test.go @@ -83,6 +83,15 @@ func TestAliases(t *testing.T) { assert.Equal(t, 45, Get("age")) } +func TestYML(t *testing.T) { + Reset() + SetConfigType("yml") + r := bytes.NewReader(yamlExample) + + MarshallReader(r) + assert.Equal(t, "steve", Get("name")) +} + func TestJSON(t *testing.T) { SetConfigType("json") r := bytes.NewReader(jsonExample)