Add support for yml extension

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

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)