glenux-archive/spf13--viper
1
0
Fork 0
mirror of https://github.com/spf13/viper synced 2024-06-25 21:29:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support for yml extension

Browse source
This commit is contained in:
Victor Castell 2014-05-08 16:40:14 +02:00 committed by spf13
parent 1a2e68e1bc
commit 16d5c7c27d
2 changed files with 11 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
viper.go
View file

@ -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 = ""

9
viper_test.go
View file

@ -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)