github.com/koko1123/flow-go-1@v0.29.6/fvm/environment/mock/account_info.go (about) 1 // Code generated by mockery v2.13.1. DO NOT EDIT. 2 3 package mock 4 5 import ( 6 common "github.com/onflow/cadence/runtime/common" 7 8 flow "github.com/koko1123/flow-go-1/model/flow" 9 10 mock "github.com/stretchr/testify/mock" 11 ) 12 13 // AccountInfo is an autogenerated mock type for the AccountInfo type 14 type AccountInfo struct { 15 mock.Mock 16 } 17 18 // GetAccount provides a mock function with given fields: address 19 func (_m *AccountInfo) GetAccount(address flow.Address) (*flow.Account, error) { 20 ret := _m.Called(address) 21 22 var r0 *flow.Account 23 if rf, ok := ret.Get(0).(func(flow.Address) *flow.Account); ok { 24 r0 = rf(address) 25 } else { 26 if ret.Get(0) != nil { 27 r0 = ret.Get(0).(*flow.Account) 28 } 29 } 30 31 var r1 error 32 if rf, ok := ret.Get(1).(func(flow.Address) error); ok { 33 r1 = rf(address) 34 } else { 35 r1 = ret.Error(1) 36 } 37 38 return r0, r1 39 } 40 41 // GetAccountAvailableBalance provides a mock function with given fields: address 42 func (_m *AccountInfo) GetAccountAvailableBalance(address common.Address) (uint64, error) { 43 ret := _m.Called(address) 44 45 var r0 uint64 46 if rf, ok := ret.Get(0).(func(common.Address) uint64); ok { 47 r0 = rf(address) 48 } else { 49 r0 = ret.Get(0).(uint64) 50 } 51 52 var r1 error 53 if rf, ok := ret.Get(1).(func(common.Address) error); ok { 54 r1 = rf(address) 55 } else { 56 r1 = ret.Error(1) 57 } 58 59 return r0, r1 60 } 61 62 // GetAccountBalance provides a mock function with given fields: address 63 func (_m *AccountInfo) GetAccountBalance(address common.Address) (uint64, error) { 64 ret := _m.Called(address) 65 66 var r0 uint64 67 if rf, ok := ret.Get(0).(func(common.Address) uint64); ok { 68 r0 = rf(address) 69 } else { 70 r0 = ret.Get(0).(uint64) 71 } 72 73 var r1 error 74 if rf, ok := ret.Get(1).(func(common.Address) error); ok { 75 r1 = rf(address) 76 } else { 77 r1 = ret.Error(1) 78 } 79 80 return r0, r1 81 } 82 83 // GetStorageCapacity provides a mock function with given fields: address 84 func (_m *AccountInfo) GetStorageCapacity(address common.Address) (uint64, error) { 85 ret := _m.Called(address) 86 87 var r0 uint64 88 if rf, ok := ret.Get(0).(func(common.Address) uint64); ok { 89 r0 = rf(address) 90 } else { 91 r0 = ret.Get(0).(uint64) 92 } 93 94 var r1 error 95 if rf, ok := ret.Get(1).(func(common.Address) error); ok { 96 r1 = rf(address) 97 } else { 98 r1 = ret.Error(1) 99 } 100 101 return r0, r1 102 } 103 104 // GetStorageUsed provides a mock function with given fields: address 105 func (_m *AccountInfo) GetStorageUsed(address common.Address) (uint64, error) { 106 ret := _m.Called(address) 107 108 var r0 uint64 109 if rf, ok := ret.Get(0).(func(common.Address) uint64); ok { 110 r0 = rf(address) 111 } else { 112 r0 = ret.Get(0).(uint64) 113 } 114 115 var r1 error 116 if rf, ok := ret.Get(1).(func(common.Address) error); ok { 117 r1 = rf(address) 118 } else { 119 r1 = ret.Error(1) 120 } 121 122 return r0, r1 123 } 124 125 type mockConstructorTestingTNewAccountInfo interface { 126 mock.TestingT 127 Cleanup(func()) 128 } 129 130 // NewAccountInfo creates a new instance of AccountInfo. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. 131 func NewAccountInfo(t mockConstructorTestingTNewAccountInfo) *AccountInfo { 132 mock := &AccountInfo{} 133 mock.Mock.Test(t) 134 135 t.Cleanup(func() { mock.AssertExpectations(t) }) 136 137 return mock 138 }