Add mock tests for accounts, addresses and balance.

Add method for getting a single balance.
This commit is contained in:
Michał Sałaban 2017-12-01 06:06:15 +01:00
parent 5e96380d7b
commit 56d7443616
5 changed files with 138 additions and 6 deletions

View file

@ -9,14 +9,17 @@ class Account(object):
self.index = index
self._backend = backend
def get_balance(self):
return self._backend.get_balance(account=self.index)
def get_balances(self):
return self._backend.get_balances(account=self.index)
def get_balance(self, unlocked=False):
return self._backend.get_balances(account=self.index)[1 if unlocked else 0]
def get_address(self):
"""
Return account's main address.
"""
return self._backend.get_address(account=self.index)[0]
return self._backend.get_addresses(account=self.index)[0]
def get_addresses(self):
return self._backend.get_addresses(account=self.index)

View file

@ -49,7 +49,7 @@ class JSONRPC(object):
addresses[_addr['address_index']] = address(_addr['address'])
return addresses
def get_balance(self, account=0):
def get_balances(self, account=0):
_balance = self.raw_request('getbalance', {'account_index': account})
return (from_atomic(_balance['balance']), from_atomic(_balance['unlocked_balance']))

View file

@ -22,8 +22,11 @@ class Wallet(object):
idx += 1
# Following methods operate on default account (index=0)
def get_balance(self):
return self.accounts[0].get_balance()
def get_balances(self):
return self.accounts[0].get_balances()
def get_balance(self, unlocked=False):
return self.accounts[0].get_balance(unlocked=unlocked)
def get_address(self, index=0):
return self.accounts[0].get_addresses()[0]