p15693087
/
wheat-cache
forked from p53841790/wheat-cache
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat(dao-external): add external mock

This commit is contained in:
bandl 2021-11-21 23:10:14 +08:00
parent 0c8a17b21d
commit a5ef559e14
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
storage/external/define.go vendored
View File

@ -13,5 +13,7 @@ var (
) )
var ( var (
GateWayCtrl *gomock.Controller GateWayCtrl *gomock.Controller
mockGatewayClient proto.CommServerClient
oneMockGatewayClient sync.Once
) )

7
storage/external/gateway.go vendored
View File

@ -18,7 +18,12 @@ func NewGatewayClient() (proto.CommServerClient, error) {
if GateWayCtrl == nil { if GateWayCtrl == nil {
return nil, errorx.New("mock ctrl not init") return nil, errorx.New("mock ctrl not init")
} }
return mockClient.NewMockCommServerClient(GateWayCtrl), nil
oneMockGatewayClient.Do(func() {
mockGatewayClient = mockClient.NewMockCommServerClient(GateWayCtrl)
})
return mockGatewayClient, nil
} }
oneGatewayClient.Do(func() { oneGatewayClient.Do(func() {