github.com/koko1123/flow-go-1@v0.29.6/state/protocol/mock/consumer.go (about) 1 // Code generated by mockery v2.13.1. DO NOT EDIT. 2 3 package mock 4 5 import ( 6 flow "github.com/koko1123/flow-go-1/model/flow" 7 mock "github.com/stretchr/testify/mock" 8 ) 9 10 // Consumer is an autogenerated mock type for the Consumer type 11 type Consumer struct { 12 mock.Mock 13 } 14 15 // BlockFinalized provides a mock function with given fields: block 16 func (_m *Consumer) BlockFinalized(block *flow.Header) { 17 _m.Called(block) 18 } 19 20 // BlockProcessable provides a mock function with given fields: block 21 func (_m *Consumer) BlockProcessable(block *flow.Header) { 22 _m.Called(block) 23 } 24 25 // EpochCommittedPhaseStarted provides a mock function with given fields: currentEpochCounter, first 26 func (_m *Consumer) EpochCommittedPhaseStarted(currentEpochCounter uint64, first *flow.Header) { 27 _m.Called(currentEpochCounter, first) 28 } 29 30 // EpochSetupPhaseStarted provides a mock function with given fields: currentEpochCounter, first 31 func (_m *Consumer) EpochSetupPhaseStarted(currentEpochCounter uint64, first *flow.Header) { 32 _m.Called(currentEpochCounter, first) 33 } 34 35 // EpochTransition provides a mock function with given fields: newEpochCounter, first 36 func (_m *Consumer) EpochTransition(newEpochCounter uint64, first *flow.Header) { 37 _m.Called(newEpochCounter, first) 38 } 39 40 type mockConstructorTestingTNewConsumer interface { 41 mock.TestingT 42 Cleanup(func()) 43 } 44 45 // NewConsumer creates a new instance of Consumer. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. 46 func NewConsumer(t mockConstructorTestingTNewConsumer) *Consumer { 47 mock := &Consumer{} 48 mock.Mock.Test(t) 49 50 t.Cleanup(func() { mock.AssertExpectations(t) }) 51 52 return mock 53 }