mirror of
https://github.com/anyproto/any-sync.git
synced 2025-06-08 05:57:03 +09:00
22 lines
442 B
Go
22 lines
442 B
Go
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()
|
|
}
|
|
}
|