带你走进Golang 单元测试（下） - 51Testing软件测试论坛

// 通过 redis 拉取对应用户的资料信息
　　func getPersonDetailRedis(username string) (*PersonDetail, error) {
　　 result := &PersonDetail{}
　　 client, err := redis.Dial("tcp", ":6379")
　　 defer client.Close()
　　 data, err := redis.Bytes(client.Do("GET", username))
　　 if err != nil {
　　 return nil, err
　　 }
　　 err = json.Unmarshal(data, result)
　　 if err != nil {
　　 return nil, err
　　 }
　　 return result, nil
}

复制代码

// redis 包中关于 Conn 的定义
　　// Conn represents a connection to a Redis server.
　　type Conn interface {
　　 // Close closes the connection.
　　 Close() error
　　 // Err returns a non-nil value when the connection is not usable.
　　 Err() error
　　 // Do sends a command to the server and returns the received reply.
　　 Do(commandName string, args ...interface{}) (reply interface{}, err error)
　　 // Send writes the command to the client's output buffer.
　　 Send(commandName string, args ...interface{}) error
　　 // Flush flushes the output buffer to the Redis server.
　　 Flush() error
　　 // Receive receives a single reply from the Redis server
　　 Receive() (reply interface{}, err error)
　　}
　　// 实现接口
　　type Client struct {}
　　func (c *Client) Close() error {
　　 return nil
　　}
　　func (c *Client) Err() error {
　　 return nil
　　}
　　func (c *Client) Do(commandName string, args ...interface{}) (interface{}, error) {
　　 return nil, nil
　　}
　　func (c *Client) Send(commandName string, args ...interface{}) error {
　　 return nil
　　}
　　func (c *Client) Flush() error {
　　 return nil
　　}
　　func (c *Client) Receive() (interface{}, error) {
　　 return nil, nil
　　}
　　// 实现接口
　　type Client struct {}
　　func (c *Client) Close() error {
　　 return nil
　　}
　　func (c *Client) Err() error {
　　 return nil
　　}
　　func (c *Client) Do(commandName string, args ...interface{}) (interface{}, error) {
　　 return nil, nil
　　}
　　func (c *Client) Send(commandName string, args ...interface{}) error {
　　 return nil
　　}
　　func (c *Client) Flush() error {
　　 return nil
　　}
　　func (c *Client) Receive() (interface{}, error) {
　　 return nil, nil
　　}
　　// 进行测试
　　func test() {
　　 c := &Client{}
　　 gomonkey.ApplyFunc(redis.Dial, func(_ string, _ string, _ ...redis.DialOption) (redis.Conn, error) {
　　 return c, nil
　　 })
　　 gomonkey.ApplyMethod(reflect.TypeOf(c), "Do", func(commandName string, args ...interface{}) (interface{}, error) {
　　 var result interface{}
　　 return result, nil
　　 })
}

复制代码

# 打开终端后依次执行
　　go get -u github.com/golang/mock/gomock
　　go install github.com/golang/mock/mockgen
　　# 备注说明，很重要！！！
　　# 安装完成之后，执行 mockgen 看命令是否生效 # 如果显示命令无效，请找到本机的 GOPATH 安装目录下的 bin 文件夹是否有 mockgen 二进制文件
　　# GOPATH 可以执行 go env 命令找到
# 如果命令无效但是 GOPATH 路径下的 bin 文件夹中存在 mockgen，请将 GOPATH 下 bin 文件夹的绝对路径添加到全局 PATH 中

复制代码

mockgen -destination=mock_redis.go -package=unit github.com/gomodule/redigo/redis Conn
# 更多指令参考：https://github.com/golang/mock#flags

复制代码

func Test_getPersonDetailRedis(t *testing.T) {
　　 tests := []struct {
　　 name string
　　 want *PersonDetail
　　 wantErr bool
　　 }{
　　 {name: "redis.Do err", want: nil, wantErr: true},
　　 {name: "json.Unmarshal err", want: nil, wantErr: true},
　　 {name: "success", want: &PersonDetail{
　　 Username: "steven",
　　 Email: "1234567@qq.com",
　　 }, wantErr: false},
　　 }
　　 ctrl := gomock.NewController(t)
　　 defer ctrl.Finish()
　　 // 1. 生成符合 redis.Conn 接口的 mockConn
　　 mockConn := NewMockConn(ctrl)
　　 // 2. 给接口打桩序列
　　 gomock.InOrder(
　　 mockConn.EXPECT().Do("GET", gomock.Any()).Return("", errors.New("redis.Do err")),
　　 mockConn.EXPECT().Close().Return(nil),
　　 mockConn.EXPECT().Do("GET", gomock.Any()).Return("123", nil),
　　 mockConn.EXPECT().Close().Return(nil),
　　 mockConn.EXPECT().Do("GET", gomock.Any()).Return([]byte(`{"username": "steven", "email": "1234567@qq.com"}`), nil),
　　 mockConn.EXPECT().Close().Return(nil),
　　 )
　　 // 3. 给 redis.Dail 函数打桩
　　 outputs := []gomonkey.OutputCell{
　　 {
　　 Values: gomonkey.Params{mockConn, nil},
　　 Times: 3, // 3 个用例
　　 },
　　 }
　　 patches := gomonkey.ApplyFuncSeq(redis.Dial, outputs)
　　 // 执行完毕之后释放桩序列
　　 defer patches.Reset()
　　 // 4. 断言
　　 for _, tt := range tests {
　　 actual, err := getPersonDetailRedis(tt.name)
　　 // 注意，equal 函数能够对结构体进行 deap diff
　　 assert.Equal(t, tt.want, actual)
　　 assert.Equal(t, tt.wantErr, err != nil)
　　 }
}

复制代码

go_unit_test [master] go test -v -cover
　　=== RUN TestGetPersonDetail
　　--- PASS: TestGetPersonDetail (0.00s)
　　=== RUN Test_checkEmail
　　--- PASS: Test_checkEmail (0.00s)
　　=== RUN Test_checkUsername
　　--- PASS: Test_checkUsername (0.00s)
　　=== RUN Test_getPersonDetailRedis
　　--- PASS: Test_getPersonDetailRedis (0.00s)
　　PASS
　　coverage: 60.8% of statements
ok unit 0.131s

复制代码

go_unit_test [master] go test -cover -v -run Test_getPersonDetailRedis
　　=== RUN Test_getPersonDetailRedis
　　--- PASS: Test_getPersonDetailRedis (0.00s)
　　PASS
　　coverage: 41.9% of statements
ok unit 0.369s

复制代码