mirror of
https://git.wownero.com/lza_menace/wownero-python.git
synced 2024-08-15 03:25:25 +00:00
Add mock tests for accounts, addresses and balance.
Add method for getting a single balance.
This commit is contained in:
parent
5e96380d7b
commit
56d7443616
5 changed files with 138 additions and 6 deletions
|
@ -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)
|
||||
|
|
|
@ -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']))
|
||||
|
||||
|
|
|
@ -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]
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue