Function TestRecoverWithConfig

middleware/recover_test.go:94–150

(t *testing.T)

}

func TestRecoverWithConfig(t *testing.T) {
	var testCases = []struct {
		name             string
		givenNoPanic     bool
		whenConfig       RecoverConfig
		expectErrContain string
		expectErr        string
	}{
		{
			name:             "ok, default config",
			whenConfig:       DefaultRecoverConfig,
			expectErrContain: "[PANIC RECOVER] testPANIC goroutine",
		},
		{
			name:             "ok, no panic",
			givenNoPanic:     true,
			whenConfig:       DefaultRecoverConfig,
			expectErrContain: "",
		},
		{
			name: "ok, DisablePrintStack",
			whenConfig: RecoverConfig{
				DisablePrintStack: true,
			},
			expectErr: "testPANIC",
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			e := echo.New()

			req := httptest.NewRequest(http.MethodGet, "/", nil)
			rec := httptest.NewRecorder()
			c := e.NewContext(req, rec)

			config := tc.whenConfig
			h := RecoverWithConfig(config)(func(c *echo.Context) error {
				if tc.givenNoPanic {
					return nil
				}
				panic("testPANIC")
			})

			err := h(c)

			if tc.expectErrContain != "" {
				assert.Contains(t, err.Error(), tc.expectErrContain)
			} else if tc.expectErr != "" {
				assert.Contains(t, err.Error(), tc.expectErr)
			} else {
				assert.NoError(t, err)
			}
			assert.Equal(t, http.StatusOK, rec.Code) // status is still untouched. err is returned from middleware chain
		})
	}
}