mirror of
https://github.com/labstack/echo.git
synced 2024-11-24 08:22:21 +02:00
c2d29f3895
Signed-off-by: Vishal Rana <vr@labstack.com>
234 lines
5.0 KiB
Go
234 lines
5.0 KiB
Go
package echo
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
type (
|
|
user struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
}
|
|
)
|
|
|
|
var u1 = user{
|
|
ID: "1",
|
|
Name: "Joe",
|
|
}
|
|
|
|
// TODO: Fix me
|
|
func TestEchoMaxParam(t *testing.T) {
|
|
e := New()
|
|
e.MaxParam(8)
|
|
if e.maxParam != 8 {
|
|
t.Errorf("max param should be 8, found %d", e.maxParam)
|
|
}
|
|
}
|
|
|
|
func TestEchoIndex(t *testing.T) {
|
|
e := New()
|
|
e.Index("example/public/index.html")
|
|
w := httptest.NewRecorder()
|
|
r, _ := http.NewRequest(MethodGET, "/", nil)
|
|
e.ServeHTTP(w, r)
|
|
if w.Code != 200 {
|
|
t.Errorf("status code should be 200, found %d", w.Code)
|
|
}
|
|
}
|
|
|
|
func TestEchoStatic(t *testing.T) {
|
|
e := New()
|
|
e.Static("/js", "example/public/js")
|
|
w := httptest.NewRecorder()
|
|
r, _ := http.NewRequest(MethodGET, "/js/main.js", nil)
|
|
e.ServeHTTP(w, r)
|
|
if w.Code != 200 {
|
|
t.Errorf("status code should be 200, found %d", w.Code)
|
|
}
|
|
}
|
|
|
|
func TestEchoMiddleware(t *testing.T) {
|
|
e := New()
|
|
b := new(bytes.Buffer)
|
|
|
|
// func(*echo.Context)
|
|
e.Use(func(c *Context) {
|
|
b.WriteString("a")
|
|
})
|
|
|
|
// func(echo.HandlerFunc) echo.HandlerFunc
|
|
e.Use(func(h HandlerFunc) HandlerFunc {
|
|
return HandlerFunc(func(c *Context) {
|
|
b.WriteString("b")
|
|
h(c)
|
|
})
|
|
})
|
|
|
|
// http.HandlerFunc
|
|
e.Use(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
b.WriteString("c")
|
|
}))
|
|
|
|
// http.Handler
|
|
e.Use(http.Handler(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
b.WriteString("d")
|
|
})))
|
|
|
|
// func(http.Handler) http.Handler
|
|
e.Use(func(h http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
b.WriteString("e")
|
|
h.ServeHTTP(w, r)
|
|
})
|
|
})
|
|
|
|
// func(http.ResponseWriter, *http.Request)
|
|
e.Use(func(w http.ResponseWriter, r *http.Request) {
|
|
b.WriteString("f")
|
|
})
|
|
|
|
// Route
|
|
e.Get("/hello", func(c *Context) {
|
|
c.Text(200, "world")
|
|
})
|
|
|
|
w := httptest.NewRecorder()
|
|
r, _ := http.NewRequest(MethodGET, "/hello", nil)
|
|
e.ServeHTTP(w, r)
|
|
if b.String() != "abcdef" {
|
|
t.Errorf("buffer should be abcdef, found %s", b.String())
|
|
}
|
|
if w.Body.String() != "world" {
|
|
t.Error("body should be world")
|
|
}
|
|
}
|
|
|
|
func TestEchoHandler(t *testing.T) {
|
|
e := New()
|
|
|
|
// func(*echo.Context)
|
|
e.Get("/1", func(c *Context) {
|
|
c.Text(http.StatusOK, "1")
|
|
})
|
|
w := httptest.NewRecorder()
|
|
r, _ := http.NewRequest(MethodGET, "/1", nil)
|
|
e.ServeHTTP(w, r)
|
|
if w.Body.String() != "1" {
|
|
t.Error("body should be 1")
|
|
}
|
|
|
|
// http.Handler/http.HandlerFunc
|
|
e.Get("/2", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte("2"))
|
|
}))
|
|
w = httptest.NewRecorder()
|
|
r, _ = http.NewRequest(MethodGET, "/2", nil)
|
|
e.ServeHTTP(w, r)
|
|
if w.Body.String() != "2" {
|
|
t.Error("body should be 2")
|
|
}
|
|
|
|
// func(http.ResponseWriter, *http.Request)
|
|
e.Get("/3", func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte("3"))
|
|
})
|
|
w = httptest.NewRecorder()
|
|
r, _ = http.NewRequest(MethodGET, "/3", nil)
|
|
e.ServeHTTP(w, r)
|
|
if w.Body.String() != "3" {
|
|
t.Error("body should be 3")
|
|
}
|
|
}
|
|
|
|
func TestEchoSubGroup(t *testing.T) {
|
|
b := new(bytes.Buffer)
|
|
|
|
e := New()
|
|
e.Use(func(*Context) {
|
|
b.WriteString("1")
|
|
})
|
|
e.Get("/users", func(*Context) {})
|
|
|
|
s := e.Sub("/sub")
|
|
s.Use(func(*Context) {
|
|
b.WriteString("2")
|
|
})
|
|
s.Get("/home", func(*Context) {})
|
|
|
|
g := e.Group("/group")
|
|
g.Use(func(*Context) {
|
|
b.WriteString("3")
|
|
})
|
|
g.Get("/home", func(*Context) {})
|
|
|
|
w := httptest.NewRecorder()
|
|
r, _ := http.NewRequest(MethodGET, "/users", nil)
|
|
e.ServeHTTP(w, r)
|
|
if b.String() != "1" {
|
|
t.Errorf("should only execute middleware 1, executed %s", b.String())
|
|
}
|
|
|
|
b.Reset()
|
|
w = httptest.NewRecorder()
|
|
r, _ = http.NewRequest(MethodGET, "/sub/home", nil)
|
|
e.ServeHTTP(w, r)
|
|
if b.String() != "12" {
|
|
t.Errorf("should execute middleware 1 & 2, executed %s", b.String())
|
|
}
|
|
|
|
b.Reset()
|
|
w = httptest.NewRecorder()
|
|
r, _ = http.NewRequest(MethodGET, "/group/home", nil)
|
|
e.ServeHTTP(w, r)
|
|
if b.String() != "3" {
|
|
t.Errorf("should execute middleware 3, executed %s", b.String())
|
|
}
|
|
}
|
|
|
|
func TestEchoMethod(t *testing.T) {
|
|
e := New()
|
|
e.Connect("/", func(*Context) {})
|
|
e.Delete("/", func(*Context) {})
|
|
e.Get("/", func(*Context) {})
|
|
e.Head("/", func(*Context) {})
|
|
e.Options("/", func(*Context) {})
|
|
e.Patch("/", func(*Context) {})
|
|
e.Post("/", func(*Context) {})
|
|
e.Put("/", func(*Context) {})
|
|
e.Trace("/", func(*Context) {})
|
|
}
|
|
|
|
func TestEchoServeHTTP(t *testing.T) {
|
|
e := New()
|
|
|
|
// OK
|
|
e.Get("/users", func(*Context) {
|
|
})
|
|
w := httptest.NewRecorder()
|
|
r, _ := http.NewRequest(MethodGET, "/users", nil)
|
|
e.ServeHTTP(w, r)
|
|
if w.Code != http.StatusOK {
|
|
t.Errorf("status code should be 200, found %d", w.Code)
|
|
}
|
|
|
|
// NotFound
|
|
r, _ = http.NewRequest(MethodGET, "/user", nil)
|
|
w = httptest.NewRecorder()
|
|
e.ServeHTTP(w, r)
|
|
if w.Code != http.StatusNotFound {
|
|
t.Errorf("status code should be 404, found %d", w.Code)
|
|
}
|
|
}
|
|
|
|
func verifyUser(u2 *user, t *testing.T) {
|
|
if u2.ID != u1.ID {
|
|
t.Errorf("user id should be %s, found %s", u1.ID, u2.ID)
|
|
}
|
|
if u2.Name != u1.Name {
|
|
t.Errorf("user name should be %s, found %s", u1.Name, u2.Name)
|
|
}
|
|
}
|