聊聊Golang中的单元测试

2023-08-19 09:47:58 浏览数 (2)

对外接口

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

代码语言:javascript复制
go get github.com/golang/mock/gomock
go get github.com/golang/mock/mockgen

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

怎样mock

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

代码语言:javascript复制
ctrl := gomock.NewController(t)
defer ctrl.Finish()
mockcli := mockredis.NewMockClient(ctrl)
stubs := gostub.Stub(&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库,比如:

代码语言:javascript复制
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

0 人点赞