github.com/onflow/flow-go@v0.35.7-crescendo-preview.23-atree-inlining/consensus/hotstuff/mocks/timeout_collector_factory.go (about) 1 // Code generated by mockery v2.21.4. DO NOT EDIT. 2 3 package mocks 4 5 import ( 6 hotstuff "github.com/onflow/flow-go/consensus/hotstuff" 7 mock "github.com/stretchr/testify/mock" 8 ) 9 10 // TimeoutCollectorFactory is an autogenerated mock type for the TimeoutCollectorFactory type 11 type TimeoutCollectorFactory struct { 12 mock.Mock 13 } 14 15 // Create provides a mock function with given fields: view 16 func (_m *TimeoutCollectorFactory) Create(view uint64) (hotstuff.TimeoutCollector, error) { 17 ret := _m.Called(view) 18 19 var r0 hotstuff.TimeoutCollector 20 var r1 error 21 if rf, ok := ret.Get(0).(func(uint64) (hotstuff.TimeoutCollector, error)); ok { 22 return rf(view) 23 } 24 if rf, ok := ret.Get(0).(func(uint64) hotstuff.TimeoutCollector); ok { 25 r0 = rf(view) 26 } else { 27 if ret.Get(0) != nil { 28 r0 = ret.Get(0).(hotstuff.TimeoutCollector) 29 } 30 } 31 32 if rf, ok := ret.Get(1).(func(uint64) error); ok { 33 r1 = rf(view) 34 } else { 35 r1 = ret.Error(1) 36 } 37 38 return r0, r1 39 } 40 41 type mockConstructorTestingTNewTimeoutCollectorFactory interface { 42 mock.TestingT 43 Cleanup(func()) 44 } 45 46 // NewTimeoutCollectorFactory creates a new instance of TimeoutCollectorFactory. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. 47 func NewTimeoutCollectorFactory(t mockConstructorTestingTNewTimeoutCollectorFactory) *TimeoutCollectorFactory { 48 mock := &TimeoutCollectorFactory{} 49 mock.Mock.Test(t) 50 51 t.Cleanup(func() { mock.AssertExpectations(t) }) 52 53 return mock 54 }