From d2b17a6b2e2a7822467ecc4615f701a57a39c769 Mon Sep 17 00:00:00 2001
From: Victor Castell <victor@victorcastell.com>
Date: Thu, 8 May 2014 16:40:14 +0200
Subject: [PATCH] Add support for yml extension

---
 viper.go      | 4 ++--
 viper_test.go | 9 +++++++++
 2 files changed, 11 insertions(+), 2 deletions(-)

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)