[object Object]

GoでのMock作り方

予めインターフェースを定義

package api

import "github.com/tubone24/what-is-your-color/models"

type GitHub struct {
	Client Client
}

type Client interface {
	GetColor(username string) (error, []models.GitHubLang)
	CallApi(username string) (error, *models.Query)
}

func (github *GitHub) DoGetColor(username string) (error, []models.GitHubLang) {
	return github.Client.GetColor(username)
}

func (github *GitHub) DoCallApi(username string) (error, *models.Query) {
	return github.Client.CallApi(username)
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

mockgenコマンド打つ

mockgen -source github.go -destination github_mock.go
1

mock使う

func TestGetColor(t *testing.T) {
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()
	m := mock.NewMockClient(ctrl)
	m.EXPECT().CallApi("tubone24").Return(nil)
	m.GetColor("tubone24")
}
1
2
3
4
5
6
7