Added an timeout test with an error handler

Author: ilijamt

middleware/timeout_test.go

@@ -89,11 +89,11 @@ func TestTimeoutErrorOutInHandler(t *testing.T) {
 	assert.Error(t, err)
 }
 
-func TestTimeoutTimesOutAfterPredefinedTimeout(t *testing.T) {
+func TestTimeoutTimesOutAfterPredefinedTimeoutWithErrorHandler(t *testing.T) {
 	m := TimeoutWithConfig(TimeoutConfig{
 		Timeout: time.Second,
 		ErrorHandler: func(err error, e echo.Context) error {
-			return err
+			return errors.New("err")
 		},
 	})
 
@@ -108,5 +108,24 @@ func TestTimeoutTimesOutAfterPredefinedTimeout(t *testing.T) {
 		return nil
 	})(c)
 
+	assert.EqualError(t, err, errors.New("err").Error())
+}
+
+func TestTimeoutTimesOutAfterPredefinedTimeout(t *testing.T) {
+	m := TimeoutWithConfig(TimeoutConfig{
+		Timeout: time.Second,
+	})
+
+	req := httptest.NewRequest(http.MethodGet, "/", nil)
+	rec := httptest.NewRecorder()
+
+	e := echo.New()
+	c := e.NewContext(req, rec)
+
+	err := m(func(c echo.Context) error {
+		time.Sleep(time.Minute)
+		return nil
+	})(c)
+
 	assert.EqualError(t, err, context.DeadlineExceeded.Error())
 }