mirrors/tailscale
1
0
Fork 0
Code Issues Pull Requests Projects Wiki Activity

tsweb: JSONHandler: supports HTTPError

Browse Source 
Signed-off-by: Zijie Lu <zijie@tailscale.com>
This commit is contained in:
Zijie Lu
2020-06-09 17:40:45 -04:00
parent 51d176ecff
commit d1a30be275
2 changed files with 29 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tsweb/jsonhandler_test.go
View File

@ -59,6 +59,19 @@ func TestNewJSONHandler(t *testing.T) {
checkStatus(w, "success")
})
t.Run("2 1 HTTPError", func(t *testing.T) {
h := JSONHandler(func(w http.ResponseWriter, r *http.Request) HTTPError {
return Error(http.StatusForbidden, "forbidden", nil)
})
w := httptest.NewRecorder()
r := httptest.NewRequest("GET", "/", nil)
h.ServeHTTP(w, r)
if w.Code != http.StatusForbidden {
t.Fatalf("wrong code: %d %d", w.Code, http.StatusForbidden)
}
})
// 2 2
h22 := JSONHandler(func(w http.ResponseWriter, r *http.Request) (*Data, error) {
return &Data{Name: "tailscale"}, nil