| 729 | } |
| 730 | |
| 731 | func TestEchoWrapMiddleware(t *testing.T) { |
| 732 | e := New() |
| 733 | |
| 734 | var actualID string |
| 735 | var actualPattern string |
| 736 | e.Use(WrapMiddleware(func(h http.Handler) http.Handler { |
| 737 | return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
| 738 | actualID = r.PathValue("id") |
| 739 | actualPattern = r.Pattern |
| 740 | h.ServeHTTP(w, r) |
| 741 | }) |
| 742 | })) |
| 743 | |
| 744 | e.GET("/:id", func(c *Context) error { |
| 745 | return c.String(http.StatusTeapot, "OK") |
| 746 | }) |
| 747 | |
| 748 | req := httptest.NewRequest(http.MethodGet, "/123", nil) |
| 749 | rec := httptest.NewRecorder() |
| 750 | e.ServeHTTP(rec, req) |
| 751 | |
| 752 | assert.Equal(t, http.StatusTeapot, rec.Code) |
| 753 | assert.Equal(t, "OK", rec.Body.String()) |
| 754 | assert.Equal(t, "123", actualID) |
| 755 | assert.Equal(t, "/:id", actualPattern) |
| 756 | } |
| 757 | |
| 758 | func TestEchoConnect(t *testing.T) { |
| 759 | e := New() |