github.com/onflow/flow-go@v0.35.7-crescendo-preview.23-atree-inlining/network/mocknetwork/disallow_list_oracle.go (about) 1 // Code generated by mockery v2.21.4. DO NOT EDIT. 2 3 package mocknetwork 4 5 import ( 6 mock "github.com/stretchr/testify/mock" 7 8 network "github.com/onflow/flow-go/network" 9 10 peer "github.com/libp2p/go-libp2p/core/peer" 11 ) 12 13 // DisallowListOracle is an autogenerated mock type for the DisallowListOracle type 14 type DisallowListOracle struct { 15 mock.Mock 16 } 17 18 // GetAllDisallowedListCausesFor provides a mock function with given fields: _a0 19 func (_m *DisallowListOracle) GetAllDisallowedListCausesFor(_a0 peer.ID) []network.DisallowListedCause { 20 ret := _m.Called(_a0) 21 22 var r0 []network.DisallowListedCause 23 if rf, ok := ret.Get(0).(func(peer.ID) []network.DisallowListedCause); ok { 24 r0 = rf(_a0) 25 } else { 26 if ret.Get(0) != nil { 27 r0 = ret.Get(0).([]network.DisallowListedCause) 28 } 29 } 30 31 return r0 32 } 33 34 type mockConstructorTestingTNewDisallowListOracle interface { 35 mock.TestingT 36 Cleanup(func()) 37 } 38 39 // NewDisallowListOracle creates a new instance of DisallowListOracle. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. 40 func NewDisallowListOracle(t mockConstructorTestingTNewDisallowListOracle) *DisallowListOracle { 41 mock := &DisallowListOracle{} 42 mock.Mock.Test(t) 43 44 t.Cleanup(func() { mock.AssertExpectations(t) }) 45 46 return mock 47 }