It("does not fail when the ResponseWriter is not a ProxyResponseWriter", func() {
handler.ServeHTTP(resp, req, nextHandler)
Expect(resp.Code).To(Equal(503))
Expect(alr.StatusCode).To(Equal(0))
})
}
BeforeEach(func() {
logger = lagertest.NewTestLogger("zipkin")
req = test_util.NewRequest("GET", "example.com", "/", nil)
resp = httptest.NewRecorder()
proxyWriter = utils.NewProxyResponseWriter(resp)
alr = &schema.AccessLogRecord{
Request: req,
}
proxyWriter.AddToContext("AccessLogRecord", alr)
nextCalled = false
heartbeatOK = 1
})
Context("with User-Agent checking", func() {
BeforeEach(func() {
handler = handlers.NewHealthcheck("HTTP-Monitor/1.1", &heartbeatOK, logger)
})
Context("when User-Agent is set to the healthcheck User-Agent", func() {
BeforeEach(func() {
req.Header.Set("User-Agent", "HTTP-Monitor/1.1")
})
TestHealthcheckOK()
