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