github.com/koko1123/flow-go-1@v0.29.6/state/cluster/mock/state.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 cluster "github.com/koko1123/flow-go-1/state/cluster" 8 9 mock "github.com/stretchr/testify/mock" 10 ) 11 12 // State is an autogenerated mock type for the State type 13 type State struct { 14 mock.Mock 15 } 16 17 // AtBlockID provides a mock function with given fields: blockID 18 func (_m *State) AtBlockID(blockID flow.Identifier) cluster.Snapshot { 19 ret := _m.Called(blockID) 20 21 var r0 cluster.Snapshot 22 if rf, ok := ret.Get(0).(func(flow.Identifier) cluster.Snapshot); ok { 23 r0 = rf(blockID) 24 } else { 25 if ret.Get(0) != nil { 26 r0 = ret.Get(0).(cluster.Snapshot) 27 } 28 } 29 30 return r0 31 } 32 33 // Final provides a mock function with given fields: 34 func (_m *State) Final() cluster.Snapshot { 35 ret := _m.Called() 36 37 var r0 cluster.Snapshot 38 if rf, ok := ret.Get(0).(func() cluster.Snapshot); ok { 39 r0 = rf() 40 } else { 41 if ret.Get(0) != nil { 42 r0 = ret.Get(0).(cluster.Snapshot) 43 } 44 } 45 46 return r0 47 } 48 49 // Params provides a mock function with given fields: 50 func (_m *State) Params() cluster.Params { 51 ret := _m.Called() 52 53 var r0 cluster.Params 54 if rf, ok := ret.Get(0).(func() cluster.Params); ok { 55 r0 = rf() 56 } else { 57 if ret.Get(0) != nil { 58 r0 = ret.Get(0).(cluster.Params) 59 } 60 } 61 62 return r0 63 } 64 65 type mockConstructorTestingTNewState interface { 66 mock.TestingT 67 Cleanup(func()) 68 } 69 70 // NewState creates a new instance of State. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. 71 func NewState(t mockConstructorTestingTNewState) *State { 72 mock := &State{} 73 mock.Mock.Test(t) 74 75 t.Cleanup(func() { mock.AssertExpectations(t) }) 76 77 return mock 78 }