github.com/onflow/flow-go@v0.35.7-crescendo-preview.23-atree-inlining/fvm/environment/mock/account_info.go (about)

     1  // Code generated by mockery v2.21.4. DO NOT EDIT.
     2  
     3  package mock
     4  
     5  import (
     6  	common "github.com/onflow/cadence/runtime/common"
     7  
     8  	flow "github.com/onflow/flow-go/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  	var r1 error
    24  	if rf, ok := ret.Get(0).(func(flow.Address) (*flow.Account, error)); ok {
    25  		return rf(address)
    26  	}
    27  	if rf, ok := ret.Get(0).(func(flow.Address) *flow.Account); ok {
    28  		r0 = rf(address)
    29  	} else {
    30  		if ret.Get(0) != nil {
    31  			r0 = ret.Get(0).(*flow.Account)
    32  		}
    33  	}
    34  
    35  	if rf, ok := ret.Get(1).(func(flow.Address) error); ok {
    36  		r1 = rf(address)
    37  	} else {
    38  		r1 = ret.Error(1)
    39  	}
    40  
    41  	return r0, r1
    42  }
    43  
    44  // GetAccountAvailableBalance provides a mock function with given fields: runtimeAddress
    45  func (_m *AccountInfo) GetAccountAvailableBalance(runtimeAddress common.Address) (uint64, error) {
    46  	ret := _m.Called(runtimeAddress)
    47  
    48  	var r0 uint64
    49  	var r1 error
    50  	if rf, ok := ret.Get(0).(func(common.Address) (uint64, error)); ok {
    51  		return rf(runtimeAddress)
    52  	}
    53  	if rf, ok := ret.Get(0).(func(common.Address) uint64); ok {
    54  		r0 = rf(runtimeAddress)
    55  	} else {
    56  		r0 = ret.Get(0).(uint64)
    57  	}
    58  
    59  	if rf, ok := ret.Get(1).(func(common.Address) error); ok {
    60  		r1 = rf(runtimeAddress)
    61  	} else {
    62  		r1 = ret.Error(1)
    63  	}
    64  
    65  	return r0, r1
    66  }
    67  
    68  // GetAccountBalance provides a mock function with given fields: runtimeAddress
    69  func (_m *AccountInfo) GetAccountBalance(runtimeAddress common.Address) (uint64, error) {
    70  	ret := _m.Called(runtimeAddress)
    71  
    72  	var r0 uint64
    73  	var r1 error
    74  	if rf, ok := ret.Get(0).(func(common.Address) (uint64, error)); ok {
    75  		return rf(runtimeAddress)
    76  	}
    77  	if rf, ok := ret.Get(0).(func(common.Address) uint64); ok {
    78  		r0 = rf(runtimeAddress)
    79  	} else {
    80  		r0 = ret.Get(0).(uint64)
    81  	}
    82  
    83  	if rf, ok := ret.Get(1).(func(common.Address) error); ok {
    84  		r1 = rf(runtimeAddress)
    85  	} else {
    86  		r1 = ret.Error(1)
    87  	}
    88  
    89  	return r0, r1
    90  }
    91  
    92  // GetStorageCapacity provides a mock function with given fields: runtimeAddress
    93  func (_m *AccountInfo) GetStorageCapacity(runtimeAddress common.Address) (uint64, error) {
    94  	ret := _m.Called(runtimeAddress)
    95  
    96  	var r0 uint64
    97  	var r1 error
    98  	if rf, ok := ret.Get(0).(func(common.Address) (uint64, error)); ok {
    99  		return rf(runtimeAddress)
   100  	}
   101  	if rf, ok := ret.Get(0).(func(common.Address) uint64); ok {
   102  		r0 = rf(runtimeAddress)
   103  	} else {
   104  		r0 = ret.Get(0).(uint64)
   105  	}
   106  
   107  	if rf, ok := ret.Get(1).(func(common.Address) error); ok {
   108  		r1 = rf(runtimeAddress)
   109  	} else {
   110  		r1 = ret.Error(1)
   111  	}
   112  
   113  	return r0, r1
   114  }
   115  
   116  // GetStorageUsed provides a mock function with given fields: runtimeaddress
   117  func (_m *AccountInfo) GetStorageUsed(runtimeaddress common.Address) (uint64, error) {
   118  	ret := _m.Called(runtimeaddress)
   119  
   120  	var r0 uint64
   121  	var r1 error
   122  	if rf, ok := ret.Get(0).(func(common.Address) (uint64, error)); ok {
   123  		return rf(runtimeaddress)
   124  	}
   125  	if rf, ok := ret.Get(0).(func(common.Address) uint64); ok {
   126  		r0 = rf(runtimeaddress)
   127  	} else {
   128  		r0 = ret.Get(0).(uint64)
   129  	}
   130  
   131  	if rf, ok := ret.Get(1).(func(common.Address) error); ok {
   132  		r1 = rf(runtimeaddress)
   133  	} else {
   134  		r1 = ret.Error(1)
   135  	}
   136  
   137  	return r0, r1
   138  }
   139  
   140  type mockConstructorTestingTNewAccountInfo interface {
   141  	mock.TestingT
   142  	Cleanup(func())
   143  }
   144  
   145  // 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.
   146  func NewAccountInfo(t mockConstructorTestingTNewAccountInfo) *AccountInfo {
   147  	mock := &AccountInfo{}
   148  	mock.Mock.Test(t)
   149  
   150  	t.Cleanup(func() { mock.AssertExpectations(t) })
   151  
   152  	return mock
   153  }