github.com/onflow/flow-go@v0.35.7-crescendo-preview.23-atree-inlining/state/cluster/mock/snapshot.go (about) 1 // Code generated by mockery v2.21.4. DO NOT EDIT. 2 3 package mock 4 5 import ( 6 flow "github.com/onflow/flow-go/model/flow" 7 mock "github.com/stretchr/testify/mock" 8 ) 9 10 // Snapshot is an autogenerated mock type for the Snapshot type 11 type Snapshot struct { 12 mock.Mock 13 } 14 15 // Collection provides a mock function with given fields: 16 func (_m *Snapshot) Collection() (*flow.Collection, error) { 17 ret := _m.Called() 18 19 var r0 *flow.Collection 20 var r1 error 21 if rf, ok := ret.Get(0).(func() (*flow.Collection, error)); ok { 22 return rf() 23 } 24 if rf, ok := ret.Get(0).(func() *flow.Collection); ok { 25 r0 = rf() 26 } else { 27 if ret.Get(0) != nil { 28 r0 = ret.Get(0).(*flow.Collection) 29 } 30 } 31 32 if rf, ok := ret.Get(1).(func() error); ok { 33 r1 = rf() 34 } else { 35 r1 = ret.Error(1) 36 } 37 38 return r0, r1 39 } 40 41 // Head provides a mock function with given fields: 42 func (_m *Snapshot) Head() (*flow.Header, error) { 43 ret := _m.Called() 44 45 var r0 *flow.Header 46 var r1 error 47 if rf, ok := ret.Get(0).(func() (*flow.Header, error)); ok { 48 return rf() 49 } 50 if rf, ok := ret.Get(0).(func() *flow.Header); ok { 51 r0 = rf() 52 } else { 53 if ret.Get(0) != nil { 54 r0 = ret.Get(0).(*flow.Header) 55 } 56 } 57 58 if rf, ok := ret.Get(1).(func() error); ok { 59 r1 = rf() 60 } else { 61 r1 = ret.Error(1) 62 } 63 64 return r0, r1 65 } 66 67 // Pending provides a mock function with given fields: 68 func (_m *Snapshot) Pending() ([]flow.Identifier, error) { 69 ret := _m.Called() 70 71 var r0 []flow.Identifier 72 var r1 error 73 if rf, ok := ret.Get(0).(func() ([]flow.Identifier, error)); ok { 74 return rf() 75 } 76 if rf, ok := ret.Get(0).(func() []flow.Identifier); ok { 77 r0 = rf() 78 } else { 79 if ret.Get(0) != nil { 80 r0 = ret.Get(0).([]flow.Identifier) 81 } 82 } 83 84 if rf, ok := ret.Get(1).(func() error); ok { 85 r1 = rf() 86 } else { 87 r1 = ret.Error(1) 88 } 89 90 return r0, r1 91 } 92 93 type mockConstructorTestingTNewSnapshot interface { 94 mock.TestingT 95 Cleanup(func()) 96 } 97 98 // NewSnapshot creates a new instance of Snapshot. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. 99 func NewSnapshot(t mockConstructorTestingTNewSnapshot) *Snapshot { 100 mock := &Snapshot{} 101 mock.Mock.Test(t) 102 103 t.Cleanup(func() { mock.AssertExpectations(t) }) 104 105 return mock 106 }