func TestRecoveryPanicString(t *testing.T) {
h := &Recovery{
Reporter: reporter.ReporterFunc(func(ctx context.Context, err error) error {
return nil
}),
handler: httpx.HandlerFunc(func(ctx context.Context, w http.ResponseWriter, r *http.Request) error {
panic("boom")
}),
}
ctx := context.Background()
req, _ := http.NewRequest("GET", "/", nil)
resp := httptest.NewRecorder()
defer func() {
if err := recover(); err != nil {
t.Fatal("Expected the panic to be handled.")
}
}()
err := h.ServeHTTPContext(ctx, resp, req)
if got, want := err.Error(), "boom"; got != want {
t.Fatalf("err => %v; want %v", got, want)
}
}
func (b *Builder) reporter() reporter.Reporter {
if b.Reporter == nil {
return reporter.ReporterFunc(func(ctx context.Context, err error) error {
fmt.Fprintf(os.Stderr, "reporting err: %v\n", err)
return nil
})
}
return b.Reporter
}
func TestRecovery(t *testing.T) {
var (
called bool
errBoom = errors.New("boom")
)
h := &Recovery{
Reporter: reporter.ReporterFunc(func(ctx context.Context, err error) error {
called = true
e := err.(*reporter.Error)
if e.Err != errBoom {
t.Fatalf("err => %v; want %v", err, errBoom)
}
if got, want := e.Context["request_id"], "1234"; got != want {
t.Fatalf("RequestID => %s; want %s", got, want)
}
return nil
}),
handler: httpx.HandlerFunc(func(ctx context.Context, w http.ResponseWriter, r *http.Request) error {
panic(errBoom)
}),
}
ctx := context.Background()
req, _ := http.NewRequest("GET", "/", nil)
req.Header.Set("X-Request-ID", "1234")
resp := httptest.NewRecorder()
ctx = httpx.WithRequest(ctx, req)
defer func() {
if err := recover(); err != nil {
t.Fatal("Expected the panic to be handled.")
}
}()
err := h.ServeHTTPContext(ctx, resp, req)
if err != errBoom {
t.Fatalf("err => %v; want %v", err, errBoom)
}
}
// newTestServer returns a new httptest.Server for testing empire's http server.
func newTestServer(t testing.TB, e *empire.Empire, opts server.Options) *Server {
if e == nil {
e = NewEmpire(t)
}
// Log reporter errors to stderr
ctx := reporter.WithReporter(context.Background(), reporter.ReporterFunc(func(ctx context.Context, err error) error {
fmt.Fprintf(os.Stderr, "reported error: %v\n", err)
return nil
}))
s := server.New(e, opts)
svr := httptest.NewUnstartedServer(middleware.Handler(ctx, s))
u, _ := url.Parse(svr.URL)
s.URL = u
return &Server{
Empire: e,
Server: s,
svr: svr,
}
}