github.com/onflow/flow-go@v0.35.7-crescendo-preview.23-atree-inlining/network/p2p/mock/disallow_list_notification_consumer.go (about) 1 // Code generated by mockery v2.21.4. DO NOT EDIT. 2 3 package mockp2p 4 5 import ( 6 network "github.com/onflow/flow-go/network" 7 mock "github.com/stretchr/testify/mock" 8 9 peer "github.com/libp2p/go-libp2p/core/peer" 10 ) 11 12 // DisallowListNotificationConsumer is an autogenerated mock type for the DisallowListNotificationConsumer type 13 type DisallowListNotificationConsumer struct { 14 mock.Mock 15 } 16 17 // OnAllowListNotification provides a mock function with given fields: id, cause 18 func (_m *DisallowListNotificationConsumer) OnAllowListNotification(id peer.ID, cause network.DisallowListedCause) { 19 _m.Called(id, cause) 20 } 21 22 // OnDisallowListNotification provides a mock function with given fields: id, cause 23 func (_m *DisallowListNotificationConsumer) OnDisallowListNotification(id peer.ID, cause network.DisallowListedCause) { 24 _m.Called(id, cause) 25 } 26 27 type mockConstructorTestingTNewDisallowListNotificationConsumer interface { 28 mock.TestingT 29 Cleanup(func()) 30 } 31 32 // NewDisallowListNotificationConsumer creates a new instance of DisallowListNotificationConsumer. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. 33 func NewDisallowListNotificationConsumer(t mockConstructorTestingTNewDisallowListNotificationConsumer) *DisallowListNotificationConsumer { 34 mock := &DisallowListNotificationConsumer{} 35 mock.Mock.Test(t) 36 37 t.Cleanup(func() { mock.AssertExpectations(t) }) 38 39 return mock 40 }