テスト

Echo のハンドラとミドルウェアは echo.Context を扱う通常の関数なので、標準の net/http/httptest パッケージで簡単にテストできます。echotest パッケージは、 定型コードを減らすヘルパーを提供します。

ハンドラをテストする

2 つのハンドラを考えます。

CreateUser：POST /users

• JSON ペイロードを受け取ります。
• 成功時に 201 Created を返します。
• エラー時に 500 Internal Server Error を返します。

GetUser：GET /users/:email

• 成功時に 200 OK を返します。
• ユーザーが存在しない場合は 404 Not Found、それ以外のエラーでは 500 Internal Server Error を返します。

handler.go：

package handler

import (
  "net/http"

  "github.com/labstack/echo/v5"
)

type (
  User struct {
    Name  string `json:"name" form:"name"`
    Email string `json:"email" form:"email"`
  }
  handler struct {
    db map[string]*User
  }
)

func (h *handler) createUser(c *echo.Context) error {
  u := new(User)
  if err := c.Bind(u); err != nil {
    return err
  }
  return c.JSON(http.StatusCreated, u)
}

func (h *handler) getUser(c *echo.Context) error {
  email := c.Param("email")
  user := h.db[email]
  if user == nil {
    return echo.NewHTTPError(http.StatusNotFound, "user not found")
  }
  return c.JSON(http.StatusOK, user)
}

handler_test.go：

package handler

import (
  "fmt"
  "net/http"
  "net/http/httptest"
  "strings"
  "testing"

  "github.com/labstack/echo/v5"
  "github.com/labstack/echo/v5/echotest"
  "github.com/stretchr/testify/assert"
)

var (
  mockDB = map[string]*User{
    "[email protected]": {Name: "Jon Snow", Email: "[email protected]"},
  }
  userJSON = `{"name":"Jon Snow","email":"[email protected]"}`
)

func TestCreateUser(t *testing.T) {
  // Setup
  e := echo.New()
  req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader(userJSON))
  req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)

  rec := httptest.NewRecorder()
  c := e.NewContext(req, rec)

  h := &handler{mockDB}

  // Assertions
  if assert.NoError(t, h.createUser(c)) {
    assert.Equal(t, http.StatusCreated, rec.Code)
    assert.Equal(t, userJSON+"\n", rec.Body.String())
  }
}

echotest ヘルパーを使う

echotest.ContextConfig は、リクエストの宣言的な記述からコンテキスト（と recorder）を構築します。

// Same test as above, using echotest.
func TestCreateUserWithEchoTest(t *testing.T) {
  c, rec := echotest.ContextConfig{
    Headers: map[string][]string{
      echo.HeaderContentType: {echo.MIMEApplicationJSON},
    },
    JSONBody: []byte(userJSON),
  }.ToContextRecorder(t)

  h := &handler{mockDB}

  // Assertions
  if assert.NoError(t, h.createUser(c)) {
    assert.Equal(t, http.StatusCreated, rec.Code)
    assert.Equal(t, userJSON+"\n", rec.Body.String())
  }
}

// Even shorter, using ServeWithHandler.
func TestCreateUserWithServeHandler(t *testing.T) {
  h := &handler{mockDB}

  rec := echotest.ContextConfig{
    Headers: map[string][]string{
      echo.HeaderContentType: {echo.MIMEApplicationJSON},
    },
    JSONBody: []byte(userJSON),
  }.ServeWithHandler(t, h.createUser)

  assert.Equal(t, http.StatusCreated, rec.Code)
  assert.Equal(t, userJSON+"\n", rec.Body.String())
}

func TestGetUser(t *testing.T) {
  c, rec := echotest.ContextConfig{
    PathValues: echo.PathValues{
      {Name: "email", Value: "[email protected]"},
    },
    Headers: map[string][]string{
      echo.HeaderContentType: {echo.MIMEApplicationJSON},
    },
  }.ToContextRecorder(t)

  h := &handler{mockDB}

  // Assertions
  if assert.NoError(t, h.getUser(c)) {
    assert.Equal(t, http.StatusOK, rec.Code)
    assert.Equal(t, userJSON+"\n", rec.Body.String())
  }
}

フォームペイロードを使う

// import "net/url"
f := make(url.Values)
f.Set("name", "Jon Snow")
f.Set("email", "[email protected]")
req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader(f.Encode()))
req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationForm)

echotest を使った multipart フォームペイロード：

func TestContext_MultipartForm(t *testing.T) {
  testConf := echotest.ContextConfig{
    MultipartForm: &echotest.MultipartForm{
      Fields: map[string]string{
        "key": "value",
      },
      Files: []echotest.MultipartFormFile{
        {
          Fieldname: "file",
          Filename:  "test.json",
          Content:   echotest.LoadBytes(t, "testdata/test.json"),
        },
      },
    },
  }
  c := testConf.ToContext(t)

  assert.Equal(t, "value", c.FormValue("key"))
  assert.Equal(t, http.MethodPost, c.Request().Method)
  assert.Equal(t, true, strings.HasPrefix(c.Request().Header.Get(echo.HeaderContentType), "multipart/form-data; boundary="))

  fv, err := c.FormFile("file")
  if err != nil {
    t.Fatal(err)
  }
  assert.Equal(t, "test.json", fv.Filename)
}

パスパラメーターを設定する

c.SetPathValues(echo.PathValues{
  {Name: "id", Value: "1"},
  {Name: "email", Value: "[email protected]"},
})

クエリパラメーターを設定する

// import "net/url"
q := make(url.Values)
q.Set("email", "[email protected]")
req := httptest.NewRequest(http.MethodGet, "/?"+q.Encode(), nil)

ミドルウェアをテストする

func TestMiddleware(t *testing.T) {
  handler := func(c *echo.Context) error {
    return c.JSON(http.StatusTeapot, fmt.Sprintf("email: %s", c.Param("email")))
  }
  middleware := func(next echo.HandlerFunc) echo.HandlerFunc {
    return func(c *echo.Context) error {
      c.Set("user_id", int64(1234))
      return next(c)
    }
  }

  c, rec := echotest.ContextConfig{
    PathValues: echo.PathValues{{Name: "email", Value: "[email protected]"}},
  }.ToContextRecorder(t)

  if err := middleware(handler)(c); err != nil {
    t.Fatal(err)
  }

  // Check that the middleware set the value.
  userID, err := echo.ContextGet[int64](c, "user_id")
  assert.NoError(t, err)
  assert.Equal(t, int64(1234), userID)

  // Check that the handler returned the correct response.
  assert.Equal(t, http.StatusTeapot, rec.Code)
}

ヒント

その他の例は、Echo ソース内の ミドルウェアテストケースを参照してください。