bingyi/gf
1
0
Fork 0
mirror of https://gitee.com/johng/gf synced 2026-06-07 02:12:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

improve unit testing case for CORS feature for package ghttp

Browse Source
This commit is contained in:
John
2020-02-16 13:32:06 +08:00
parent 7d9bccf912
commit 5169137069
1 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
net/ghttp/ghttp_unit_middleware_test.go
View File

@ -596,7 +596,7 @@ func Test_Middleware_CORSAndAuth(t *testing.T) {
s := g.Server(p)
s.Group("/api.v2", func(group *ghttp.RouterGroup) {
group.Middleware(MiddlewareAuth, MiddlewareCORS)
group.ALL("/user/list", func(r *ghttp.Request) {
group.POST("/user/list", func(r *ghttp.Request) {
r.Response.Write("list")
})
})
@ -608,11 +608,21 @@ func Test_Middleware_CORSAndAuth(t *testing.T) {
gtest.Case(t, func() {
client := ghttp.NewClient()
client.SetPrefix(fmt.Sprintf("http://127.0.0.1:%d", p))
// Common Checks.
gtest.Assert(client.GetContent("/"), "Not Found")
gtest.Assert(client.GetContent("/api.v2"), "Not Found")
gtest.Assert(client.GetContent("/api.v2/user/list"), "Forbidden")
gtest.Assert(client.GetContent("/api.v2/user/list", "token=123456"), "list")
// Auth Checks.
gtest.Assert(client.PostContent("/api.v2/user/list"), "Forbidden")
gtest.Assert(client.PostContent("/api.v2/user/list", "token=123456"), "list")
// CORS Checks.
resp, err := client.Post("/api.v2/user/list", "token=123456")
gtest.Assert(err, nil)
gtest.Assert(len(resp.Header["Access-Control-Allow-Headers"]), 1)
gtest.Assert(resp.Header["Access-Control-Allow-Headers"][0], "Origin,Content-Type,Accept,User-Agent,Cookie,Authorization,X-Auth-Token,X-Requested-With")
gtest.Assert(resp.Header["Access-Control-Allow-Methods"][0], "GET,PUT,POST,DELETE,PATCH,HEAD,CONNECT,OPTIONS,TRACE")
gtest.Assert(resp.Header["Access-Control-Allow-Origin"][0], "*")
gtest.Assert(resp.Header["Access-Control-Max-Age"][0], "3628800")
resp.Close()
})
}