glenux-archive/spf13--viper
1
0
Fork 0
mirror of https://github.com/spf13/viper synced 2024-12-22 19:47:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add support for yml extension

Browse source
This commit is contained in:
Victor Castell 2014-05-08 16:40:14 +02:00
parent 1a2e68e1bc
commit d2b17a6b2e
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) buf.ReadFrom(in)
switch getConfigType() { switch getConfigType() {
case "yaml": case "yaml", "yml":
if err := yaml.Unmarshal(buf.Bytes(), &config); err != nil { if err := yaml.Unmarshal(buf.Bytes(), &config); err != nil {
jww.ERROR.Fatalf("Error parsing config: %s", err) jww.ERROR.Fatalf("Error parsing config: %s", err)
} }
@ -399,7 +399,7 @@ func Reset() {
configName = "config" configName = "config"
// extensions Supported // extensions Supported
SupportedExts = []string{"json", "toml", "yaml"} SupportedExts = []string{"json", "toml", "yaml", "yml"}
configFile = "" configFile = ""
configType = "" configType = ""

9
viper_test.go
View file

@ -83,6 +83,15 @@ func TestAliases(t *testing.T) {
assert.Equal(t, 45, Get("age")) 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) { func TestJSON(t *testing.T) {
SetConfigType("json") SetConfigType("json")
r := bytes.NewReader(jsonExample) r := bytes.NewReader(jsonExample)