Add support for yml extension

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

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

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)