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() {