对外接口
对于向外提供的接口来说,一般需要提供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