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