title: 测试 url: guide/testing menu: side: parent: guide

  1. weight: 12

测试

测试处理程序 (Testing handler)

GET /users/:id

下面的处理程序是根据用户的 id 从数据库取到该用户数据，如果用户不存在则返回 404 和提示语句。

创建 User

POST /users

  • 接受 JSON 格式的关键信息
  • 创建成功返回 201 - Created
  • 发生错误返回 500 - Internal Server Error

获取 User

GET /users/:email

  • 获取成功返回 200 - OK
  • 未获取 User 返回 404 - Not Found
  • 发生其它错误返回 500 - Internal Server Error

handler.go

  1. package handler
  3. import (
  4.     "net/http"
  6.     "github.com/labstack/echo"
  7. )
  9. type (
  10.     User struct {
  11.         Name  string `json:"name" form:"name"`
  12.         Email string `json:"email" form:"email"`
  13.     }
  14.     handler struct {
  15.         db map[string]*User
  16.     }
  17. )
  19. func (h *handler) createUser(c echo.Context) error {
  20.     u := new(User)
  21.     if err := c.Bind(u); err != nil {
  22.         return err
  23.     }
  24.     return c.JSON(http.StatusCreated, u)
  25. }
  27. func (h *handler) getUser(c echo.Context) error {
  28.     email := c.Param("email")
  29.     user := h.db[email]
  30.     if user == nil {
  31.         return echo.NewHTTPError(http.StatusNotFound, "user not found")
  32.     }
  33.     return c.JSON(http.StatusOK, user)
  34. }

handler_test.go

  1. package handler
  3. import (
  4.     "net/http"
  5.     "net/http/httptest"
  6.     "strings"
  7.     "testing"
  9.     "github.com/labstack/echo"
  10.     "github.com/stretchr/testify/assert"
  11. )
  13. var (
  14.     mockDB = map[string]*User{
  15.         "jon@labstack.com": &User{"Jon Snow", "jon@labstack.com"},
  16.     }
  17.     userJSON = `{"name":"Jon Snow","email":"jon@labstack.com"}`
  18. )
  20. func TestCreateUser(t *testing.T) {
  21.     // 设置
  22.     e := echo.New()
  23.     req := httptest.NewRequest(echo.POST, "/", strings.NewReader(userJSON))
  24.     req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
  25.     rec := httptest.NewRecorder()
  26.     c := e.NewContext(req, rec)
  27.     h := &handler{mockDB}
  29.     // 断言
  30.     if assert.NoError(t, h.createUser(c)) {
  31.         assert.Equal(t, http.StatusCreated, rec.Code)
  32.         assert.Equal(t, userJSON, rec.Body.String())
  33.     }
  34. }
  36. func TestGetUser(t *testing.T) {
  37.     // 设置
  38.     e := echo.New()
  39.     req := httptest.NewRequest(echo.GET, "/", nil)
  40.     rec := httptest.NewRecorder()
  41.     c := e.NewContext(req, rec)
  42.     c.SetPath("/users/:email")
  43.     c.SetParamNames("email")
  44.     c.SetParamValues("jon@labstack.com")
  45.     h := &handler{mockDB}
  47.     // 断言
  48.     if assert.NoError(t, h.getUser(c)) {
  49.         assert.Equal(t, http.StatusOK, rec.Code)
  50.         assert.Equal(t, userJSON, rec.Body.String())
  51.     }
  52. }

使用 Form 表单作为关键信息

  1. f := make(url.Values)
  2. f.Set("name", "Jon Snow")
  3. f.Set("email", "jon@labstack.com")
  4. req := httptest.NewRequest(echo.POST, "/", strings.NewReader(f.Encode()))
  5. req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationForm)

设置路径 (Path) 参数

  1. c.SetParamNames("id", "email")
  2. c.SetParamValues("1", "jon@labstack.com")

设置查询 (Query) 参数

  1. q := make(url.Values)
  2. q.Set("email", "jon@labstack.com")
  3. req := http.NewRequest(echo.POST, "/?"+q.Encode(), nil)

测试中间件

待定 你可以在这里查看框架自带中间件的测试代码。