status-go/services/wallet/token/mock/balance_persistence/balance_persistence.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))
}