聊聊Golang中的单元测试

tunsuy

发布于 2023-08-19 09:47:58

33800

代码可运行

文章被收录于专栏：有文化的技术人有文化的技术人

运行总次数：0

代码可运行

对外接口

对于向外提供的接口来说，一般需要提供mock接口给使用方：安装这两个库

go get github.com/golang/mock/gomock
go get github.com/golang/mock/mockgen

然后通过mockgen 命令生成mock接口文件

怎样mock

1、对于提供了mock接口文件的接口mock，可以直接使用：

ctrl := gomock.NewController(t)
defer ctrl.Finish()
mockcli := mockredis.NewMockClient(ctrl)
stubs := gostub.Stub(&amp;GetClient,
   func() redis.Client { return mockcli })
defer stubs.Reset()
mockcli.EXPECT().Do(context.Background(),
   "set", info.Key, info.Value, "nx", "ex", info.TimeTick).Return(1, nil).Times(1)
mockcli.EXPECT().Do(context.Background(),
   "set", info.Key, info.Value, "nx", "ex", info.TimeTick).Return(0, redis.ErrNil).AnyTimes()

使用这种方式mock方法的话，有个很不好的地方就是，所有的运行时入参必须完全一样。

但是这种方式支持接口实现类自动识别

2. 一般mock，可以使用gomonkey库，比如：

defer gomonkey.ApplyFunc(coreUtils.CountModel, func(_ *mongo.Collection, _ any) (uint32, error) {
   return 1, nil
}).Reset()

但是该库无法mock泛型 https://buaq.net/go-122496.html https://taoshu.in/go/monkey/generic.html 使用该库还有一个不好的地方就是，对于applymethod来说，如果对于接口来说，无法实现接口实现类的自动识别，也就是说无法直接使用接口，必须显示的使用实现类，也就是必须要求实现类可导出

https://medium.com/@victor.neuret/mocking-the-official-mongo-golang-driver-5aad5b226a78

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2023-07-18，如有侵权请联系 cloudcommunity@tencent.com 删除

单元测试