67 lines
2.3 KiB
Go
67 lines
2.3 KiB
Go
|
// Code generated by MockGen. DO NOT EDIT.
|
||
|
// Source: services/wallet/token/balance_persistence.go
|
||
|
|
||
|
// Package mock_balance_persistence is a generated GoMock package.
|
||
|
package mock_balance_persistence
|
||
|
|
||
|
import (
|
||
|
reflect "reflect"
|
||
|
|
||
|
gomock "github.com/golang/mock/gomock"
|
||
|
|
||
|
common "github.com/ethereum/go-ethereum/common"
|
||
|
token "github.com/status-im/status-go/services/wallet/token"
|
||
|
)
|
||
|
|
||
|
// MockTokenBalancesStorage is a mock of TokenBalancesStorage interface
|
||
|
type MockTokenBalancesStorage struct {
|
||
|
ctrl *gomock.Controller
|
||
|
recorder *MockTokenBalancesStorageMockRecorder
|
||
|
}
|
||
|
|
||
|
// MockTokenBalancesStorageMockRecorder is the mock recorder for MockTokenBalancesStorage
|
||
|
type MockTokenBalancesStorageMockRecorder struct {
|
||
|
mock *MockTokenBalancesStorage
|
||
|
}
|
||
|
|
||
|
// NewMockTokenBalancesStorage creates a new mock instance
|
||
|
func NewMockTokenBalancesStorage(ctrl *gomock.Controller) *MockTokenBalancesStorage {
|
||
|
mock := &MockTokenBalancesStorage{ctrl: ctrl}
|
||
|
mock.recorder = &MockTokenBalancesStorageMockRecorder{mock}
|
||
|
return mock
|
||
|
}
|
||
|
|
||
|
// EXPECT returns an object that allows the caller to indicate expected use
|
||
|
func (m *MockTokenBalancesStorage) EXPECT() *MockTokenBalancesStorageMockRecorder {
|
||
|
return m.recorder
|
||
|
}
|
||
|
|
||
|
// SaveTokens mocks base method
|
||
|
func (m *MockTokenBalancesStorage) SaveTokens(tokens map[common.Address][]token.StorageToken) error {
|
||
|
m.ctrl.T.Helper()
|
||
|
ret := m.ctrl.Call(m, "SaveTokens", tokens)
|
||
|
ret0, _ := ret[0].(error)
|
||
|
return ret0
|
||
|
}
|
||
|
|
||
|
// SaveTokens indicates an expected call of SaveTokens
|
||
|
func (mr *MockTokenBalancesStorageMockRecorder) SaveTokens(tokens interface{}) *gomock.Call {
|
||
|
mr.mock.ctrl.T.Helper()
|
||
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SaveTokens", reflect.TypeOf((*MockTokenBalancesStorage)(nil).SaveTokens), tokens)
|
||
|
}
|
||
|
|
||
|
// GetTokens mocks base method
|
||
|
func (m *MockTokenBalancesStorage) GetTokens() (map[common.Address][]token.StorageToken, error) {
|
||
|
m.ctrl.T.Helper()
|
||
|
ret := m.ctrl.Call(m, "GetTokens")
|
||
|
ret0, _ := ret[0].(map[common.Address][]token.StorageToken)
|
||
|
ret1, _ := ret[1].(error)
|
||
|
return ret0, ret1
|
||
|
}
|
||
|
|
||
|
// GetTokens indicates an expected call of GetTokens
|
||
|
func (mr *MockTokenBalancesStorageMockRecorder) GetTokens() *gomock.Call {
|
||
|
mr.mock.ctrl.T.Helper()
|
||
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTokens", reflect.TypeOf((*MockTokenBalancesStorage)(nil).GetTokens))
|
||
|
}
|