瀏覽代碼

Improve serveError code coverage (#980)

Eason Lin 9 年之前
父節點
當前提交
436d8e2af9
共有 1 個文件被更改,包括 12 次插入0 次删除
  1. 12 0
      routes_test.go

+ 12 - 0
routes_test.go

@@ -439,3 +439,15 @@ func TestRouteRawPathNoUnescape(t *testing.T) {
 	w := performRequest(route, "POST", "/project/Some%2FOther%2FProject/build/333")
 	assert.Equal(t, w.Code, 200)
 }
+
+func TestRouteServeErrorWithWriteHeader(t *testing.T) {
+	route := New()
+	route.Use(func(c *Context) {
+		c.Status(421)
+		c.Next()
+	})
+
+	w := performRequest(route, "GET", "/NotFound")
+	assert.Equal(t, 421, w.Code)
+	assert.Equal(t, 0, w.Body.Len())
+}