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