github.com/status-im/status-go@v1.1.0/services/wallet/token/mock/balance_persistence/balance_persistence.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: services/wallet/token/balance_persistence.go
     3  
     4  // Package mock_balance_persistence is a generated GoMock package.
     5  package mock_balance_persistence
     6  
     7  import (
     8  	reflect "reflect"
     9  
    10  	gomock "github.com/golang/mock/gomock"
    11  
    12  	common "github.com/ethereum/go-ethereum/common"
    13  	token "github.com/status-im/status-go/services/wallet/token"
    14  )
    15  
    16  // MockTokenBalancesStorage is a mock of TokenBalancesStorage interface
    17  type MockTokenBalancesStorage struct {
    18  	ctrl     *gomock.Controller
    19  	recorder *MockTokenBalancesStorageMockRecorder
    20  }
    21  
    22  // MockTokenBalancesStorageMockRecorder is the mock recorder for MockTokenBalancesStorage
    23  type MockTokenBalancesStorageMockRecorder struct {
    24  	mock *MockTokenBalancesStorage
    25  }
    26  
    27  // NewMockTokenBalancesStorage creates a new mock instance
    28  func NewMockTokenBalancesStorage(ctrl *gomock.Controller) *MockTokenBalancesStorage {
    29  	mock := &MockTokenBalancesStorage{ctrl: ctrl}
    30  	mock.recorder = &MockTokenBalancesStorageMockRecorder{mock}
    31  	return mock
    32  }
    33  
    34  // EXPECT returns an object that allows the caller to indicate expected use
    35  func (m *MockTokenBalancesStorage) EXPECT() *MockTokenBalancesStorageMockRecorder {
    36  	return m.recorder
    37  }
    38  
    39  // SaveTokens mocks base method
    40  func (m *MockTokenBalancesStorage) SaveTokens(tokens map[common.Address][]token.StorageToken) error {
    41  	m.ctrl.T.Helper()
    42  	ret := m.ctrl.Call(m, "SaveTokens", tokens)
    43  	ret0, _ := ret[0].(error)
    44  	return ret0
    45  }
    46  
    47  // SaveTokens indicates an expected call of SaveTokens
    48  func (mr *MockTokenBalancesStorageMockRecorder) SaveTokens(tokens interface{}) *gomock.Call {
    49  	mr.mock.ctrl.T.Helper()
    50  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SaveTokens", reflect.TypeOf((*MockTokenBalancesStorage)(nil).SaveTokens), tokens)
    51  }
    52  
    53  // GetTokens mocks base method
    54  func (m *MockTokenBalancesStorage) GetTokens() (map[common.Address][]token.StorageToken, error) {
    55  	m.ctrl.T.Helper()
    56  	ret := m.ctrl.Call(m, "GetTokens")
    57  	ret0, _ := ret[0].(map[common.Address][]token.StorageToken)
    58  	ret1, _ := ret[1].(error)
    59  	return ret0, ret1
    60  }
    61  
    62  // GetTokens indicates an expected call of GetTokens
    63  func (mr *MockTokenBalancesStorageMockRecorder) GetTokens() *gomock.Call {
    64  	mr.mock.ctrl.T.Helper()
    65  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTokens", reflect.TypeOf((*MockTokenBalancesStorage)(nil).GetTokens))
    66  }