mirror of
https://github.com/anyproto/any-sync.git
synced 2025-06-07 21:47:02 +09:00
anymock helper
This commit is contained in:
parent
3a3be11416
commit
c6ac38edb3
1 changed files with 22 additions and 0 deletions
22
testutil/anymock/anymock.go
Normal file
22
testutil/anymock/anymock.go
Normal file
|
@ -0,0 +1,22 @@
|
|||
package anymock
|
||||
|
||||
import "go.uber.org/mock/gomock"
|
||||
|
||||
type MockComp interface {
|
||||
Name() *gomock.Call
|
||||
Init(x any) *gomock.Call
|
||||
}
|
||||
|
||||
type MockCompRunnable interface {
|
||||
Run(x any) *gomock.Call
|
||||
Close(x any) *gomock.Call
|
||||
}
|
||||
|
||||
func ExpectComp(c MockComp, name string) {
|
||||
c.Name().Return(name).AnyTimes()
|
||||
c.Init(gomock.Any()).AnyTimes()
|
||||
if cr, ok := c.(MockCompRunnable); ok {
|
||||
cr.Run(gomock.Any()).AnyTimes()
|
||||
cr.Close(gomock.Any()).AnyTimes()
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue