mirror of
https://github.com/labstack/echo.git
synced 2024-11-24 08:22:21 +02:00
aab17b9f69
Signed-off-by: Vishal Rana <vr@labstack.com>
395 lines
8.3 KiB
Go
395 lines
8.3 KiB
Go
package echo
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"reflect"
|
|
"strings"
|
|
|
|
"errors"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"golang.org/x/net/websocket"
|
|
)
|
|
|
|
type (
|
|
user struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
}
|
|
)
|
|
|
|
func TestEcho(t *testing.T) {
|
|
e := New()
|
|
req, _ := http.NewRequest(GET, "/", nil)
|
|
rec := httptest.NewRecorder()
|
|
c := NewContext(req, NewResponse(rec), e)
|
|
|
|
// Router
|
|
assert.NotNil(t, e.Router())
|
|
|
|
// Debug
|
|
e.SetDebug(true)
|
|
assert.True(t, e.Debug())
|
|
|
|
// DefaultHTTPErrorHandler
|
|
e.DefaultHTTPErrorHandler(errors.New("error"), c)
|
|
assert.Equal(t, http.StatusInternalServerError, rec.Code)
|
|
}
|
|
|
|
// TODO: Improve me!
|
|
func TestEchoMaxParam(t *testing.T) {
|
|
e := New()
|
|
e.SetMaxParam(8)
|
|
assert.EqualValues(t, 8, e.maxParam)
|
|
}
|
|
|
|
func TestEchoIndex(t *testing.T) {
|
|
e := New()
|
|
e.Index("examples/website/public/index.html")
|
|
c, b := request(GET, "/", e)
|
|
assert.Equal(t, http.StatusOK, c)
|
|
assert.NotEmpty(t, b)
|
|
}
|
|
|
|
func TestEchoFavicon(t *testing.T) {
|
|
e := New()
|
|
e.Favicon("examples/website/public/favicon.ico")
|
|
c, b := request(GET, "/favicon.ico", e)
|
|
assert.Equal(t, http.StatusOK, c)
|
|
assert.NotEmpty(t, b)
|
|
}
|
|
|
|
func TestEchoStatic(t *testing.T) {
|
|
e := New()
|
|
|
|
// OK
|
|
e.Static("/scripts", "examples/website/public/scripts")
|
|
c, b := request(GET, "/scripts/main.js", e)
|
|
assert.Equal(t, http.StatusOK, c)
|
|
assert.NotEmpty(t, b)
|
|
|
|
// No file
|
|
e.Static("/scripts", "examples/website/public/scripts")
|
|
c, _ = request(GET, "/scripts/index.js", e)
|
|
assert.Equal(t, http.StatusNotFound, c)
|
|
|
|
// Directory
|
|
e.Static("/scripts", "examples/website/public/scripts")
|
|
c, _ = request(GET, "/scripts", e)
|
|
assert.Equal(t, http.StatusForbidden, c)
|
|
}
|
|
|
|
func TestEchoMiddleware(t *testing.T) {
|
|
e := New()
|
|
buf := new(bytes.Buffer)
|
|
|
|
// echo.MiddlewareFunc
|
|
e.Use(MiddlewareFunc(func(h HandlerFunc) HandlerFunc {
|
|
return func(c *Context) error {
|
|
buf.WriteString("a")
|
|
return h(c)
|
|
}
|
|
}))
|
|
|
|
// func(echo.HandlerFunc) echo.HandlerFunc
|
|
e.Use(func(h HandlerFunc) HandlerFunc {
|
|
return func(c *Context) error {
|
|
buf.WriteString("b")
|
|
return h(c)
|
|
}
|
|
})
|
|
|
|
// echo.HandlerFunc
|
|
e.Use(HandlerFunc(func(c *Context) error {
|
|
buf.WriteString("c")
|
|
return nil
|
|
}))
|
|
|
|
// func(*echo.Context) error
|
|
e.Use(func(c *Context) error {
|
|
buf.WriteString("d")
|
|
return nil
|
|
})
|
|
|
|
// func(http.Handler) http.Handler
|
|
e.Use(func(h http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
buf.WriteString("e")
|
|
h.ServeHTTP(w, r)
|
|
})
|
|
})
|
|
|
|
// http.Handler
|
|
e.Use(http.Handler(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
buf.WriteString("f")
|
|
})))
|
|
|
|
// http.HandlerFunc
|
|
e.Use(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
buf.WriteString("g")
|
|
}))
|
|
|
|
// func(http.ResponseWriter, *http.Request)
|
|
e.Use(func(w http.ResponseWriter, r *http.Request) {
|
|
buf.WriteString("h")
|
|
})
|
|
|
|
// Unknown
|
|
assert.Panics(t, func() {
|
|
e.Use(nil)
|
|
})
|
|
|
|
// Route
|
|
e.Get("/", func(c *Context) error {
|
|
return c.String(http.StatusOK, "Hello!")
|
|
})
|
|
|
|
c, b := request(GET, "/", e)
|
|
assert.Equal(t, "abcdefgh", buf.String())
|
|
assert.Equal(t, http.StatusOK, c)
|
|
assert.Equal(t, "Hello!", b)
|
|
|
|
// Error
|
|
e.Use(func(*Context) error {
|
|
return errors.New("error")
|
|
})
|
|
c, b = request(GET, "/", e)
|
|
assert.Equal(t, http.StatusInternalServerError, c)
|
|
}
|
|
|
|
func TestEchoHandler(t *testing.T) {
|
|
e := New()
|
|
|
|
// HandlerFunc
|
|
e.Get("/1", HandlerFunc(func(c *Context) error {
|
|
return c.String(http.StatusOK, "1")
|
|
}))
|
|
|
|
// func(*echo.Context) error
|
|
e.Get("/2", func(c *Context) error {
|
|
return c.String(http.StatusOK, "2")
|
|
})
|
|
|
|
// http.Handler/http.HandlerFunc
|
|
e.Get("/3", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte("3"))
|
|
}))
|
|
|
|
// func(http.ResponseWriter, *http.Request)
|
|
e.Get("/4", func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte("4"))
|
|
})
|
|
|
|
for _, p := range []string{"1", "2", "3", "4"} {
|
|
c, b := request(GET, "/"+p, e)
|
|
assert.Equal(t, http.StatusOK, c)
|
|
assert.Equal(t, p, b)
|
|
}
|
|
|
|
// Unknown
|
|
assert.Panics(t, func() {
|
|
e.Get("/5", nil)
|
|
})
|
|
}
|
|
|
|
func TestEchoConnect(t *testing.T) {
|
|
e := New()
|
|
testMethod(t, CONNECT, "/", e)
|
|
}
|
|
|
|
func TestEchoDelete(t *testing.T) {
|
|
e := New()
|
|
testMethod(t, DELETE, "/", e)
|
|
}
|
|
|
|
func TestEchoGet(t *testing.T) {
|
|
e := New()
|
|
testMethod(t, GET, "/", e)
|
|
}
|
|
|
|
func TestEchoHead(t *testing.T) {
|
|
e := New()
|
|
testMethod(t, HEAD, "/", e)
|
|
}
|
|
|
|
func TestEchoOptions(t *testing.T) {
|
|
e := New()
|
|
testMethod(t, OPTIONS, "/", e)
|
|
}
|
|
|
|
func TestEchoPatch(t *testing.T) {
|
|
e := New()
|
|
testMethod(t, PATCH, "/", e)
|
|
}
|
|
|
|
func TestEchoPost(t *testing.T) {
|
|
e := New()
|
|
testMethod(t, POST, "/", e)
|
|
}
|
|
|
|
func TestEchoPut(t *testing.T) {
|
|
e := New()
|
|
testMethod(t, PUT, "/", e)
|
|
}
|
|
|
|
func TestEchoTrace(t *testing.T) {
|
|
e := New()
|
|
testMethod(t, TRACE, "/", e)
|
|
}
|
|
|
|
func TestEchoWebSocket(t *testing.T) {
|
|
e := New()
|
|
e.WebSocket("/ws", func(c *Context) error {
|
|
c.socket.Write([]byte("test"))
|
|
return nil
|
|
})
|
|
srv := httptest.NewServer(e)
|
|
defer srv.Close()
|
|
addr := srv.Listener.Addr().String()
|
|
origin := "http://localhost"
|
|
url := fmt.Sprintf("ws://%s/ws", addr)
|
|
ws, err := websocket.Dial(url, "", origin)
|
|
if assert.NoError(t, err) {
|
|
ws.Write([]byte("test"))
|
|
defer ws.Close()
|
|
buf := new(bytes.Buffer)
|
|
buf.ReadFrom(ws)
|
|
assert.Equal(t, "test", buf.String())
|
|
}
|
|
}
|
|
|
|
func TestEchoURL(t *testing.T) {
|
|
e := New()
|
|
|
|
static := func(*Context) error { return nil }
|
|
getUser := func(*Context) error { return nil }
|
|
getFile := func(*Context) error { return nil }
|
|
|
|
e.Get("/static/file", static)
|
|
e.Get("/users/:id", getUser)
|
|
g := e.Group("/group")
|
|
g.Get("/users/:uid/files/:fid", getFile)
|
|
|
|
assert.Equal(t, "/static/file", e.URL(static))
|
|
assert.Equal(t, "/users/:id", e.URL(getUser))
|
|
assert.Equal(t, "/users/1", e.URL(getUser, "1"))
|
|
assert.Equal(t, "/group/users/1/files/:fid", e.URL(getFile, "1"))
|
|
assert.Equal(t, "/group/users/1/files/1", e.URL(getFile, "1", "1"))
|
|
}
|
|
|
|
func TestEchoRoutes(t *testing.T) {
|
|
e := New()
|
|
h := func(*Context) error { return nil }
|
|
routes := []Route{
|
|
{GET, "/users/:user/events", h},
|
|
{GET, "/users/:user/events/public", h},
|
|
{POST, "/repos/:owner/:repo/git/refs", h},
|
|
{POST, "/repos/:owner/:repo/git/tags", h},
|
|
}
|
|
for _, r := range routes {
|
|
e.add(r.Method, r.Path, h)
|
|
}
|
|
|
|
for i, r := range e.Routes() {
|
|
assert.Equal(t, routes[i].Method, r.Method)
|
|
assert.Equal(t, routes[i].Path, r.Path)
|
|
}
|
|
}
|
|
|
|
func TestEchoGroup(t *testing.T) {
|
|
e := New()
|
|
buf := new(bytes.Buffer)
|
|
e.Use(func(*Context) error {
|
|
buf.WriteString("0")
|
|
return nil
|
|
})
|
|
h := func(*Context) error { return nil }
|
|
|
|
//--------
|
|
// Routes
|
|
//--------
|
|
|
|
e.Get("/users", h)
|
|
|
|
// Group
|
|
g1 := e.Group("/group1")
|
|
g1.Use(func(*Context) error {
|
|
buf.WriteString("1")
|
|
return nil
|
|
})
|
|
g1.Get("/", h)
|
|
|
|
// Group with no parent middleware
|
|
g2 := e.Group("/group2", func(*Context) error {
|
|
buf.WriteString("2")
|
|
return nil
|
|
})
|
|
g2.Get("/", h)
|
|
|
|
// Nested groups
|
|
g3 := e.Group("/group3")
|
|
g4 := g3.Group("/group4")
|
|
g4.Get("/", func(c *Context) error {
|
|
return c.NoContent(http.StatusOK)
|
|
})
|
|
|
|
request(GET, "/users", e)
|
|
// println(len(e.middleware))
|
|
assert.Equal(t, "0", buf.String())
|
|
|
|
buf.Reset()
|
|
request(GET, "/group1/", e)
|
|
// println(len(g1.echo.middleware))
|
|
assert.Equal(t, "01", buf.String())
|
|
|
|
buf.Reset()
|
|
request(GET, "/group2/", e)
|
|
assert.Equal(t, "2", buf.String())
|
|
|
|
buf.Reset()
|
|
c, _ := request(GET, "/group3/group4/", e)
|
|
assert.Equal(t, http.StatusOK, c)
|
|
}
|
|
|
|
func TestEchoNotFound(t *testing.T) {
|
|
e := New()
|
|
r, _ := http.NewRequest(GET, "/files", nil)
|
|
w := httptest.NewRecorder()
|
|
e.ServeHTTP(w, r)
|
|
assert.Equal(t, http.StatusNotFound, w.Code)
|
|
}
|
|
|
|
func TestEchoHTTPError(t *testing.T) {
|
|
m := http.StatusText(http.StatusBadRequest)
|
|
he := NewHTTPError(http.StatusBadRequest, m)
|
|
assert.Equal(t, http.StatusBadRequest, he.Code())
|
|
assert.Equal(t, m, he.Error())
|
|
}
|
|
|
|
func testMethod(t *testing.T, method, path string, e *Echo) {
|
|
m := fmt.Sprintf("%c%s", method[0], strings.ToLower(method[1:]))
|
|
p := reflect.ValueOf(path)
|
|
h := reflect.ValueOf(func(c *Context) error {
|
|
c.String(http.StatusOK, method)
|
|
return nil
|
|
})
|
|
i := interface{}(e)
|
|
reflect.ValueOf(i).MethodByName(m).Call([]reflect.Value{p, h})
|
|
_, body := request(method, path, e)
|
|
if body != method {
|
|
t.Errorf("expected body `%s`, got %s.", method, body)
|
|
}
|
|
}
|
|
|
|
func request(method, path string, e *Echo) (int, string) {
|
|
r, _ := http.NewRequest(method, path, nil)
|
|
w := httptest.NewRecorder()
|
|
e.ServeHTTP(w, r)
|
|
return w.Code, w.Body.String()
|
|
}
|