func (s *TraceSuite) TestTraceSimple(c *C) {
handler := http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
w.Header().Set("Content-Length", "5")
w.Write([]byte("hello"))
})
buf := &bytes.Buffer{}
l := utils.NewFileLogger(buf, utils.INFO)
trace := &bytes.Buffer{}
t, err := New(handler, trace, Logger(l))
c.Assert(err, IsNil)
srv := httptest.NewServer(t)
defer srv.Close()
re, _, err := testutils.MakeRequest(srv.URL+"/hello", testutils.Method("POST"), testutils.Body("123456"))
c.Assert(err, IsNil)
c.Assert(re.StatusCode, Equals, http.StatusOK)
var r *Record
c.Assert(json.Unmarshal(trace.Bytes(), &r), IsNil)
c.Assert(r.Request.Method, Equals, "POST")
c.Assert(r.Request.URL, Equals, "/hello")
c.Assert(r.Response.Code, Equals, http.StatusOK)
c.Assert(r.Request.BodyBytes, Equals, int64(6))
c.Assert(r.Response.Roundtrip, Not(Equals), float64(0))
c.Assert(r.Response.BodyBytes, Equals, int64(5))
}
func (s *ServerSuite) TestFrontendOptionsCRUD(c *C) {
e := testutils.NewResponder("Hi, I'm endpoint 1")
defer e.Close()
c.Assert(s.mux.Start(), IsNil)
b := MakeBatch(Batch{
Addr: "localhost:31000",
Route: `Path("/")`,
URL: e.URL,
})
c.Assert(s.mux.UpsertServer(b.BK, b.S), IsNil)
c.Assert(s.mux.UpsertFrontend(b.F), IsNil)
c.Assert(s.mux.UpsertListener(b.L), IsNil)
body := "Hello, this request is longer than 8 bytes"
response, bodyBytes, err := testutils.MakeRequest(MakeURL(b.L, "/"), testutils.Body(body))
c.Assert(err, IsNil)
c.Assert(string(bodyBytes), Equals, "Hi, I'm endpoint 1")
c.Assert(response.StatusCode, Equals, http.StatusOK)
settings := engine.HTTPFrontendSettings{
Limits: engine.HTTPFrontendLimits{
MaxBodyBytes: 8,
},
FailoverPredicate: "IsNetworkError()",
}
b.F.Settings = settings
c.Assert(s.mux.UpsertFrontend(b.F), IsNil)
response, _, err = testutils.MakeRequest(MakeURL(b.L, "/"), testutils.Body(body))
c.Assert(err, IsNil)
c.Assert(response.StatusCode, Equals, http.StatusRequestEntityTooLarge)
}