github.com/koko1123/flow-go-1@v0.29.6/network/p2p/mock/peer_manager.go (about) 1 // Code generated by mockery v2.13.1. DO NOT EDIT. 2 3 package mockp2p 4 5 import ( 6 context "context" 7 8 irrecoverable "github.com/koko1123/flow-go-1/module/irrecoverable" 9 mock "github.com/stretchr/testify/mock" 10 11 p2p "github.com/koko1123/flow-go-1/network/p2p" 12 ) 13 14 // PeerManager is an autogenerated mock type for the PeerManager type 15 type PeerManager struct { 16 mock.Mock 17 } 18 19 // Done provides a mock function with given fields: 20 func (_m *PeerManager) Done() <-chan struct{} { 21 ret := _m.Called() 22 23 var r0 <-chan struct{} 24 if rf, ok := ret.Get(0).(func() <-chan struct{}); ok { 25 r0 = rf() 26 } else { 27 if ret.Get(0) != nil { 28 r0 = ret.Get(0).(<-chan struct{}) 29 } 30 } 31 32 return r0 33 } 34 35 // ForceUpdatePeers provides a mock function with given fields: _a0 36 func (_m *PeerManager) ForceUpdatePeers(_a0 context.Context) { 37 _m.Called(_a0) 38 } 39 40 // Ready provides a mock function with given fields: 41 func (_m *PeerManager) Ready() <-chan struct{} { 42 ret := _m.Called() 43 44 var r0 <-chan struct{} 45 if rf, ok := ret.Get(0).(func() <-chan struct{}); ok { 46 r0 = rf() 47 } else { 48 if ret.Get(0) != nil { 49 r0 = ret.Get(0).(<-chan struct{}) 50 } 51 } 52 53 return r0 54 } 55 56 // RequestPeerUpdate provides a mock function with given fields: 57 func (_m *PeerManager) RequestPeerUpdate() { 58 _m.Called() 59 } 60 61 // SetPeersProvider provides a mock function with given fields: _a0 62 func (_m *PeerManager) SetPeersProvider(_a0 p2p.PeersProvider) { 63 _m.Called(_a0) 64 } 65 66 // Start provides a mock function with given fields: _a0 67 func (_m *PeerManager) Start(_a0 irrecoverable.SignalerContext) { 68 _m.Called(_a0) 69 } 70 71 type mockConstructorTestingTNewPeerManager interface { 72 mock.TestingT 73 Cleanup(func()) 74 } 75 76 // NewPeerManager creates a new instance of PeerManager. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. 77 func NewPeerManager(t mockConstructorTestingTNewPeerManager) *PeerManager { 78 mock := &PeerManager{} 79 mock.Mock.Test(t) 80 81 t.Cleanup(func() { mock.AssertExpectations(t) }) 82 83 return mock 84 }