mirror of
https://github.com/labstack/echo.git
synced 2024-11-24 08:22:21 +02:00
27 lines
568 B
Go
27 lines
568 B
Go
package middleware
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/labstack/echo"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestRecover(t *testing.T) {
|
|
e := echo.New()
|
|
buf := new(bytes.Buffer)
|
|
e.Logger.SetOutput(buf)
|
|
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
|
rec := httptest.NewRecorder()
|
|
c := e.NewContext(req, rec)
|
|
h := Recover()(echo.HandlerFunc(func(c echo.Context) error {
|
|
panic("test")
|
|
}))
|
|
h(c)
|
|
assert.Equal(t, http.StatusInternalServerError, rec.Code)
|
|
assert.Contains(t, buf.String(), "PANIC RECOVER")
|
|
}
|