forked from gocraft/web
/
error_test.go
160 lines (128 loc) · 5.39 KB
/
error_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
package web_test
import (
"fmt"
"github.com/gocraft/web"
. "launchpad.net/gocheck"
"net/http"
"strings"
)
type ErrorTestSuite struct{}
var _ = Suite(&ErrorTestSuite{})
func ErrorHandlerWithNoContext(w web.ResponseWriter, r *web.Request, err interface{}) {
w.WriteHeader(http.StatusInternalServerError)
fmt.Fprintf(w, "Contextless Error")
}
func (s *ErrorTestSuite) TestNoErorHandler(c *C) {
router := web.New(Context{})
router.Get("/action", (*Context).ErrorAction)
admin := router.Subrouter(AdminContext{}, "/admin")
admin.Get("/action", (*AdminContext).ErrorAction)
rw, req := newTestRequest("GET", "/action")
router.ServeHTTP(rw, req)
c.Assert(strings.TrimSpace(string(rw.Body.Bytes())), Equals, "Application Error")
c.Assert(rw.Code, Equals, http.StatusInternalServerError)
rw, req = newTestRequest("GET", "/admin/action")
router.ServeHTTP(rw, req)
c.Assert(strings.TrimSpace(string(rw.Body.Bytes())), Equals, "Application Error")
c.Assert(rw.Code, Equals, http.StatusInternalServerError)
}
func (s *ErrorTestSuite) TestHandlerOnRoot(c *C) {
router := web.New(Context{})
router.Error((*Context).ErrorHandler)
router.Get("/action", (*Context).ErrorAction)
admin := router.Subrouter(AdminContext{}, "/admin")
admin.Get("/action", (*AdminContext).ErrorAction)
rw, req := newTestRequest("GET", "/action")
router.ServeHTTP(rw, req)
c.Assert(strings.TrimSpace(string(rw.Body.Bytes())), Equals, "My Error")
c.Assert(rw.Code, Equals, http.StatusInternalServerError)
rw, req = newTestRequest("GET", "/admin/action")
router.ServeHTTP(rw, req)
c.Assert(strings.TrimSpace(string(rw.Body.Bytes())), Equals, "My Error")
c.Assert(rw.Code, Equals, http.StatusInternalServerError)
}
func (s *ErrorTestSuite) TestContextlessError(c *C) {
router := web.New(Context{})
router.Error(ErrorHandlerWithNoContext)
router.Get("/action", (*Context).ErrorAction)
admin := router.Subrouter(AdminContext{}, "/admin")
admin.Get("/action", (*AdminContext).ErrorAction)
rw, req := newTestRequest("GET", "/action")
router.ServeHTTP(rw, req)
c.Assert(strings.TrimSpace(string(rw.Body.Bytes())), Equals, "Contextless Error")
c.Assert(rw.Code, Equals, http.StatusInternalServerError)
rw, req = newTestRequest("GET", "/admin/action")
router.ServeHTTP(rw, req)
c.Assert(strings.TrimSpace(string(rw.Body.Bytes())), Equals, "Contextless Error")
c.Assert(rw.Code, Equals, http.StatusInternalServerError)
}
func (s *ErrorTestSuite) TestMultipleErrorHandlers(c *C) {
router := web.New(Context{})
router.Error((*Context).ErrorHandler)
router.Get("/action", (*Context).ErrorAction)
admin := router.Subrouter(AdminContext{}, "/admin")
admin.Error((*AdminContext).ErrorHandler)
admin.Get("/action", (*AdminContext).ErrorAction)
rw, req := newTestRequest("GET", "/action")
router.ServeHTTP(rw, req)
c.Assert(strings.TrimSpace(string(rw.Body.Bytes())), Equals, "My Error")
c.Assert(rw.Code, Equals, http.StatusInternalServerError)
rw, req = newTestRequest("GET", "/admin/action")
router.ServeHTTP(rw, req)
c.Assert(strings.TrimSpace(string(rw.Body.Bytes())), Equals, "Admin Error")
c.Assert(rw.Code, Equals, http.StatusInternalServerError)
}
func (s *ErrorTestSuite) TestMultipleErrorHandlers2(c *C) {
router := web.New(Context{})
router.Get("/action", (*Context).ErrorAction)
admin := router.Subrouter(AdminContext{}, "/admin")
admin.Error((*AdminContext).ErrorHandler)
admin.Get("/action", (*AdminContext).ErrorAction)
api := router.Subrouter(ApiContext{}, "/api")
api.Error((*ApiContext).ErrorHandler)
api.Get("/action", (*ApiContext).ErrorAction)
rw, req := newTestRequest("GET", "/action")
router.ServeHTTP(rw, req)
c.Assert(strings.TrimSpace(string(rw.Body.Bytes())), Equals, "Application Error")
c.Assert(rw.Code, Equals, http.StatusInternalServerError)
rw, req = newTestRequest("GET", "/admin/action")
router.ServeHTTP(rw, req)
c.Assert(strings.TrimSpace(string(rw.Body.Bytes())), Equals, "Admin Error")
c.Assert(rw.Code, Equals, http.StatusInternalServerError)
rw, req = newTestRequest("GET", "/api/action")
router.ServeHTTP(rw, req)
c.Assert(strings.TrimSpace(string(rw.Body.Bytes())), Equals, "Api Error")
c.Assert(rw.Code, Equals, http.StatusInternalServerError)
}
func (s *ErrorTestSuite) TestRootMiddlewarePanic(c *C) {
router := web.New(Context{})
router.Middleware((*Context).ErrorMiddleware)
router.Error((*Context).ErrorHandler)
admin := router.Subrouter(AdminContext{}, "/admin")
admin.Error((*AdminContext).ErrorHandler)
admin.Get("/action", (*AdminContext).ErrorAction)
rw, req := newTestRequest("GET", "/admin/action")
router.ServeHTTP(rw, req)
assertResponse(c, rw, "My Error", 500)
}
func (s *ErrorTestSuite) TestNonRootMiddlewarePanic(c *C) {
router := web.New(Context{})
router.Error((*Context).ErrorHandler)
admin := router.Subrouter(AdminContext{}, "/admin")
admin.Middleware((*AdminContext).ErrorMiddleware)
admin.Error((*AdminContext).ErrorHandler)
admin.Get("/action", (*AdminContext).ErrorAction)
rw, req := newTestRequest("GET", "/admin/action")
router.ServeHTTP(rw, req)
assertResponse(c, rw, "Admin Error", 500)
}
func (s *ErrorTestSuite) TestConsistentContext(c *C) {
router := web.New(Context{})
router.Error((*Context).ErrorHandler)
admin := router.Subrouter(Context{}, "/admin")
admin.Error((*Context).ErrorHandlerSecondary)
admin.Get("/foo", (*Context).ErrorAction)
rw, req := newTestRequest("GET", "/admin/foo")
router.ServeHTTP(rw, req)
assertResponse(c, rw, "My Secondary Error", 500)
}