From 43967fed892dfa5d1f46d418412ca258723d3ab8 Mon Sep 17 00:00:00 2001
From: "Manu Mtz.-Almeida" <manu.valladolid@gmail.com>
Date: Thu, 12 Nov 2015 15:19:58 +0100
Subject: [PATCH] Improves unit tests

---
 README.md    |  2 ++
 cors_test.go | 20 ++++++++++++++++++++
 2 files changed, 22 insertions(+)

diff --git a/README.md b/README.md
index 24a2212..8c9eb4f 100644
--- a/README.md
+++ b/README.md
@@ -59,6 +59,7 @@ func main() {
 ###Default() allows all origins
 
 ```go
+func main() {
     router := gin.Default()
     // same as
     // config := cors.DefaultConfig()
@@ -66,6 +67,7 @@ func main() {
     // router.Use(cors.Default())
     router.Use(cors.Default())
     router.Run()
+}
 ```
 
 
diff --git a/cors_test.go b/cors_test.go
index 8e2e826..1f50425 100644
--- a/cors_test.go
+++ b/cors_test.go
@@ -39,6 +39,26 @@ func performRequest(r http.Handler, method, origin string) *httptest.ResponseRec
 	return w
 }
 
+func TestConfigAddAllow(t *testing.T) {
+	config := Config{}
+	config.AddAllowMethods("POST")
+	config.AddAllowMethods("GET", "PUT")
+	config.AddExposeHeaders()
+
+	config.AddAllowHeaders("Some", " cool")
+	config.AddAllowHeaders("header")
+	config.AddExposeHeaders()
+
+	config.AddExposeHeaders()
+	config.AddExposeHeaders("exposed", "header")
+	config.AddExposeHeaders("hey")
+
+	assert.Equal(t, config.AllowMethods, []string{"POST", "GET", "PUT"})
+	assert.Equal(t, config.AllowHeaders, []string{"Some", " cool", "header"})
+	assert.Equal(t, config.ExposeHeaders, []string{"exposed", "header", "hey"})
+
+}
+
 func TestBadConfig(t *testing.T) {
 	assert.Panics(t, func() { New(Config{}) })
 	assert.Panics(t, func() {