github.com/koko1123/flow-go-1@v0.29.6/state/protocol/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 mock "github.com/stretchr/testify/mock" 8 9 protocol "github.com/koko1123/flow-go-1/state/protocol" 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) protocol.Snapshot { 19 ret := _m.Called(blockID) 20 21 var r0 protocol.Snapshot 22 if rf, ok := ret.Get(0).(func(flow.Identifier) protocol.Snapshot); ok { 23 r0 = rf(blockID) 24 } else { 25 if ret.Get(0) != nil { 26 r0 = ret.Get(0).(protocol.Snapshot) 27 } 28 } 29 30 return r0 31 } 32 33 // AtHeight provides a mock function with given fields: height 34 func (_m *State) AtHeight(height uint64) protocol.Snapshot { 35 ret := _m.Called(height) 36 37 var r0 protocol.Snapshot 38 if rf, ok := ret.Get(0).(func(uint64) protocol.Snapshot); ok { 39 r0 = rf(height) 40 } else { 41 if ret.Get(0) != nil { 42 r0 = ret.Get(0).(protocol.Snapshot) 43 } 44 } 45 46 return r0 47 } 48 49 // Final provides a mock function with given fields: 50 func (_m *State) Final() protocol.Snapshot { 51 ret := _m.Called() 52 53 var r0 protocol.Snapshot 54 if rf, ok := ret.Get(0).(func() protocol.Snapshot); ok { 55 r0 = rf() 56 } else { 57 if ret.Get(0) != nil { 58 r0 = ret.Get(0).(protocol.Snapshot) 59 } 60 } 61 62 return r0 63 } 64 65 // Params provides a mock function with given fields: 66 func (_m *State) Params() protocol.Params { 67 ret := _m.Called() 68 69 var r0 protocol.Params 70 if rf, ok := ret.Get(0).(func() protocol.Params); ok { 71 r0 = rf() 72 } else { 73 if ret.Get(0) != nil { 74 r0 = ret.Get(0).(protocol.Params) 75 } 76 } 77 78 return r0 79 } 80 81 // Sealed provides a mock function with given fields: 82 func (_m *State) Sealed() protocol.Snapshot { 83 ret := _m.Called() 84 85 var r0 protocol.Snapshot 86 if rf, ok := ret.Get(0).(func() protocol.Snapshot); ok { 87 r0 = rf() 88 } else { 89 if ret.Get(0) != nil { 90 r0 = ret.Get(0).(protocol.Snapshot) 91 } 92 } 93 94 return r0 95 } 96 97 type mockConstructorTestingTNewState interface { 98 mock.TestingT 99 Cleanup(func()) 100 } 101 102 // 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. 103 func NewState(t mockConstructorTestingTNewState) *State { 104 mock := &State{} 105 mock.Mock.Test(t) 106 107 t.Cleanup(func() { mock.AssertExpectations(t) }) 108 109 return mock 110 }