瀏覽代碼

Improves unit tests

Manu Mtz.-Almeida 10 年之前
父節點
當前提交
43967fed89
共有 2 個文件被更改,包括 22 次插入0 次删除
  1. 2 0
      README.md
  2. 20 0
      cors_test.go

+ 2 - 0
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()
+}
 ```
 
 

+ 20 - 0
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() {