github.com/onflow/flow-go@v0.35.7-crescendo-preview.23-atree-inlining/state/protocol/mock/epoch_query.go (about) 1 // Code generated by mockery v2.21.4. DO NOT EDIT. 2 3 package mock 4 5 import ( 6 protocol "github.com/onflow/flow-go/state/protocol" 7 mock "github.com/stretchr/testify/mock" 8 ) 9 10 // EpochQuery is an autogenerated mock type for the EpochQuery type 11 type EpochQuery struct { 12 mock.Mock 13 } 14 15 // Current provides a mock function with given fields: 16 func (_m *EpochQuery) Current() protocol.Epoch { 17 ret := _m.Called() 18 19 var r0 protocol.Epoch 20 if rf, ok := ret.Get(0).(func() protocol.Epoch); ok { 21 r0 = rf() 22 } else { 23 if ret.Get(0) != nil { 24 r0 = ret.Get(0).(protocol.Epoch) 25 } 26 } 27 28 return r0 29 } 30 31 // Next provides a mock function with given fields: 32 func (_m *EpochQuery) Next() protocol.Epoch { 33 ret := _m.Called() 34 35 var r0 protocol.Epoch 36 if rf, ok := ret.Get(0).(func() protocol.Epoch); ok { 37 r0 = rf() 38 } else { 39 if ret.Get(0) != nil { 40 r0 = ret.Get(0).(protocol.Epoch) 41 } 42 } 43 44 return r0 45 } 46 47 // Previous provides a mock function with given fields: 48 func (_m *EpochQuery) Previous() protocol.Epoch { 49 ret := _m.Called() 50 51 var r0 protocol.Epoch 52 if rf, ok := ret.Get(0).(func() protocol.Epoch); ok { 53 r0 = rf() 54 } else { 55 if ret.Get(0) != nil { 56 r0 = ret.Get(0).(protocol.Epoch) 57 } 58 } 59 60 return r0 61 } 62 63 type mockConstructorTestingTNewEpochQuery interface { 64 mock.TestingT 65 Cleanup(func()) 66 } 67 68 // NewEpochQuery creates a new instance of EpochQuery. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. 69 func NewEpochQuery(t mockConstructorTestingTNewEpochQuery) *EpochQuery { 70 mock := &EpochQuery{} 71 mock.Mock.Test(t) 72 73 t.Cleanup(func() { mock.AssertExpectations(t) }) 74 75 return mock 76 }