概要

Golangのユニットテストをtestify/suitを用いて書きました。
https://github.com/stretchr/testify/tree/master/suite
Golangのテスト時のモックに関しても一緒に書きたいと思います。

実装

実際のテストの中身は今回の記事の対象外の為、一番簡易なMVCで言う所のController層のテストで紹介します。

controller.go

package controller

import (
	"article/api/service"
	"net/http"

	"github.com/gin-gonic/gin"
)

type Controller struct {
	service service.ServiceInterface
}

func NewController(service service.ServiceInterface) *Controller {
	return &Controller{service: service}
}

type ControllerInterface interface {
	GetArticleController(c *gin.Context)
}

func (controller Controller) GetArticleController(c *gin.Context) {
	articles := controller.service.GetArticleService()
	c.JSON(http.StatusOK, articles)
}

controller.goから参照されており、テストではモックにするservice.goは以下になります。

service.go

package service

import (
	"fmt"

	"github.com/gin-gonic/gin"

	"article/api/dao"
	"article/api/util"
)

type Service struct {
	dao dao.DaoInterface
}

func NewService(dao dao.DaoInterface) *Service {
	return &Service{dao: dao}
}

type ServiceInterface interface {
	GetArticleService() []util.Article
}

func (s Service) GetArticleService() []util.Article {
	//some code
}

実際のcontroller.goのテストファイルは以下のようになります。

controller_test.go

package controller

import (
	"article/api/service"
	"article/api/util"
	"net/http/httptest"
	"testing"

	"github.com/gin-gonic/gin"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/suite"
)

type MockServiceInterface struct {
}

func (_m *MockServiceInterface) GetArticleService() []util.Article {
	return nil
}

type ControllerSuite struct {
	suite.Suite
	controller *Controller
	service    service.ServiceInterface
}

func (s *ControllerSuite) SetupTest() {
	s.controller = NewController(s.service)
	s.controller.service = &MockServiceInterface{}
}

func TestControllerSuite(t *testing.T) {
	suite.Run(t, new(ControllerSuite))
}

func (s *ControllerSuite) TestGetArticleController() {
	c, _ := gin.CreateTestContext(httptest.NewRecorder())
	s.controller.GetArticleController(c)
	assert.Equal(s.T(), 200, c.Writer.Status())
}

MockServiceInterfaceに上書きすることにより、モックを実現しています。
今回はstatus codeが正常であることだけが分かればいいテストにするので、GetArticleService()の返り値はnilにしています。

実際にテストを実行してみます。

terminal

$ go test -v
=== RUN   TestControllerSuite
=== RUN   TestControllerSuite/TestGetArticleController
[GIN-debug] [WARNING] Running in "debug" mode. Switch to "release" mode in production.
 - using env:   export GIN_MODE=release
 - using code:  gin.SetMode(gin.ReleaseMode)

--- PASS: TestControllerSuite (0.00s)
    --- PASS: TestControllerSuite/TestGetArticleController (0.00s)
PASS
ok      article/api/controller  0.044s

passしました。

全体像を把握したい場合は、以下のgithubをご参照ください。
https://github.com/jpskgc/article/tree/master/api