欢迎光临散文网 会员登陆 & 注册

gomock 教程

2023-04-09 18:53 作者:罹心  | 我要投稿

gomock 教程


GoMock是Go官方出品的一款mock框架,可以Go语言内置的testing包很好集成,并且提供EXCEPT方法去指定期待的值。


## 安装

打开[https://github.com/golang/mock](https://github.com/golang/mock)

```sh

go install github.com/golang/mock/mockgen@v1.6.0

```


输入mockgen查看是否下载在$GOPATH/bin 目录下


## 基本用法步骤


首先选定一个mock的demo目录


比如说叫gomock-learn,然后在此目录下创建对应的mod,然后引入对应的gomock包


```

go mod init gomock-learn

go get github.com/golang/mock v1.6.0 // indirect

```

接下来创建两个目录person和student分别用来放对应的接口和代码。


```go

// person.go

package person


type Person interface {

Eat(food string) string

Sleep(name string) string

}

```


```

// student.go

package student


import "gomock-learn/person"


type Student struct {

p    person.Person

Name string

}


func (p *Student) Eat(food string) string {

return p.p.Eat(food)

}


func (p *Student) Sleep() string {

return p.p.Sleep(p.Name)

}

```


接着你要创建一个mocks目录,不然如果没有mock目录的话用mockgen命令行会失败


使用方法,直接在相应的目录下执行以下命令


```

mockgen -destination mocks/mock_person.go -package=mocks gomock-learn/person Person

```

这里需要注意的是我们必须自己创建mocks目录因为GoMock不会自动帮我们创建,当它发现mocks目录不存在时会返回一个错误。以下是对mockgen命令参数的说明:


````

 -destination=mocks/mock_person.go:将自动生成的mock代码存储到文件mocks/mock_person.go中。-


package=mocks:将生成的mock代码放置到mocks包中。


gomock-learn/person:为这个包生成mock代码。


Person:为这个接口生成mock代码。这个参数是个必填参数,我们需要显式地指定要生成mock代码的接口。如果需要指定多个接口,可以将接口通过逗号连接起来,比如:Person1,Person2。

````


## 结合go-generate使用GoMock

在对应的借口前加入注释


```

//go:generate mockgen -destination mocks/mock_person.go -package=mocks gomock-learn/person Person


type Person interface {

Eat(food string) string

Sleep(name string) string

}

```


然后在对应的目录下输入

```

go generate ./ 

```


可以发现就在对应的mocks目录下里有一个mock_xx.go函数,这个函数里面就是我们可以Mock的数据。


## 使用参数匹配


有时候你可能不太确定调用mock时指定的参数,所以有一个对应的Matcher来代表一个mock方法可以接受的参数范围,比如gomock.Eq(x)指定传入值必须等于x。


以下是GoMock中一些预定义的matcher:

```

    gomock.Any():匹配任何类型的任何值

    gomock.Eq(x):匹配使用反射reflect.DeepEqual与x相等的值gomock.Nil():匹配等于nil的值

    gomock.Not(m):(这里的m是一个Matcher)匹配同m不匹配的值

    gomock.Not(x):(这里的x不是Matcher)匹配使用反射reflect.DeepEqual与x不相等的值

```


如果我们希望第一个参数必须是x,那么我们就用

```

mockDoer.EXPECT().DoSomething(gomock.Eq(x), "Hello GoMock")

```


具体例子

```go

func Test_Eat(t *testing.T) {

ctrl := gomock.NewController(t)

mockPerson := mocks.NewMockPerson(ctrl)


mockPerson.

EXPECT().

Eat("Apple").Times(1)


testStudent := Student{Name: "lixin", p: mockPerson}

testStudent.Eat("Apple")

}

```


```go


func Test_Sleep(t *testing.T) {

ctrl := gomock.NewController(t)

mockPerson := mocks.NewMockPerson(ctrl)

testStudent := Student{Name: "lixin", p: mockPerson}


mockPerson.

EXPECT().

Sleep("lixin").Return("lixin is sleep").Do(func(name string) {

fmt.Printf("%s is sleep!\n", name)

})


if testStudent.Sleep() != "lixin is sleep" {

t.Error("Error!!!!!")

}

}

```


## 断言调用顺序


有时候我们期望控制一些mock流程的顺序,这里有一个例子调用After的方法

```go

func Test_Eat(t *testing.T) {

ctrl := gomock.NewController(t)

mockPerson := mocks.NewMockPerson(ctrl)


first := mockPerson.EXPECT().Eat("xxx")


mockPerson.

EXPECT().

Eat("Apple").

After(first)


testStudent := Student{Name: "lixin", p: mockPerson}

testStudent.Eat("xxx")

testStudent.Eat("Apple")

}

```


## 指定mock行为


比如说可以在执行完毕后加一个Do函数去做一些事情。


## 总结


安装

基本用法步骤

结合go-generate使用GoMock

使用参数匹配

断言调用顺序

指定mock的行为


gomock 教程的评论 (共 条)

分享到微博请遵守国家法律