github.com/onflow/flow-go@v0.35.7-crescendo-preview.23-atree-inlining/fvm/environment/mock/address_generator.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 // AddressGenerator is an autogenerated mock type for the AddressGenerator type 11 type AddressGenerator struct { 12 mock.Mock 13 } 14 15 // AddressCount provides a mock function with given fields: 16 func (_m *AddressGenerator) AddressCount() uint64 { 17 ret := _m.Called() 18 19 var r0 uint64 20 if rf, ok := ret.Get(0).(func() uint64); ok { 21 r0 = rf() 22 } else { 23 r0 = ret.Get(0).(uint64) 24 } 25 26 return r0 27 } 28 29 // Bytes provides a mock function with given fields: 30 func (_m *AddressGenerator) Bytes() []byte { 31 ret := _m.Called() 32 33 var r0 []byte 34 if rf, ok := ret.Get(0).(func() []byte); ok { 35 r0 = rf() 36 } else { 37 if ret.Get(0) != nil { 38 r0 = ret.Get(0).([]byte) 39 } 40 } 41 42 return r0 43 } 44 45 // CurrentAddress provides a mock function with given fields: 46 func (_m *AddressGenerator) CurrentAddress() flow.Address { 47 ret := _m.Called() 48 49 var r0 flow.Address 50 if rf, ok := ret.Get(0).(func() flow.Address); ok { 51 r0 = rf() 52 } else { 53 if ret.Get(0) != nil { 54 r0 = ret.Get(0).(flow.Address) 55 } 56 } 57 58 return r0 59 } 60 61 // NextAddress provides a mock function with given fields: 62 func (_m *AddressGenerator) NextAddress() (flow.Address, error) { 63 ret := _m.Called() 64 65 var r0 flow.Address 66 var r1 error 67 if rf, ok := ret.Get(0).(func() (flow.Address, error)); ok { 68 return rf() 69 } 70 if rf, ok := ret.Get(0).(func() flow.Address); ok { 71 r0 = rf() 72 } else { 73 if ret.Get(0) != nil { 74 r0 = ret.Get(0).(flow.Address) 75 } 76 } 77 78 if rf, ok := ret.Get(1).(func() error); ok { 79 r1 = rf() 80 } else { 81 r1 = ret.Error(1) 82 } 83 84 return r0, r1 85 } 86 87 type mockConstructorTestingTNewAddressGenerator interface { 88 mock.TestingT 89 Cleanup(func()) 90 } 91 92 // NewAddressGenerator creates a new instance of AddressGenerator. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. 93 func NewAddressGenerator(t mockConstructorTestingTNewAddressGenerator) *AddressGenerator { 94 mock := &AddressGenerator{} 95 mock.Mock.Test(t) 96 97 t.Cleanup(func() { mock.AssertExpectations(t) }) 98 99 return mock 100 }