Return Seed object from the wallet

This commit is contained in:
Michał Sałaban 2018-02-19 23:06:16 +01:00
parent 94656dc771
commit a91b975e4d
2 changed files with 19 additions and 1 deletions

View File

@ -10,6 +10,7 @@ from .. import exceptions
from ..account import Account
from ..address import address, Address, SubAddress
from ..numbers import from_atomic, to_atomic, PaymentID
from ..seed import Seed
from ..transaction import Transaction, IncomingPayment, OutgoingPayment
_log = logging.getLogger(__name__)
@ -133,7 +134,7 @@ class JSONRPCWallet(object):
return self.raw_request('query_key', {'key_type': 'view_key'})['key']
def seed(self):
return self.raw_request('query_key', {'key_type': 'mnemonic'})['key']
return Seed(self.raw_request('query_key', {'key_type': 'mnemonic'})['key'])
def accounts(self):
accounts = []

View File

@ -10,6 +10,7 @@ from monero.wallet import Wallet
from monero.account import Account
from monero.address import Address, address
from monero.numbers import PaymentID
from monero.seed import Seed
from monero.transaction import IncomingPayment, OutgoingPayment, Transaction
from monero.backends.jsonrpc import JSONRPCWallet
@ -34,6 +35,22 @@ class SubaddrWalletTestCase(unittest.TestCase):
'total_balance': 236153709446071,
'total_unlocked_balance': 236153709446071}}
@patch('monero.backends.jsonrpc.requests.post')
def test_seed(self, mock_post):
mock_post.return_value.status_code = 200
mock_post.return_value.json.return_value = self.accounts_result
self.wallet = Wallet(JSONRPCWallet())
phrase = 'phrases petals speedy fuming ascend weird duplex identity ' \
'yearbook masterful elope omission height empty react hope ' \
'left iceberg leisure bobsled pyramid ammo sorry tiers ' \
'pyramid'
mock_post.return_value.json.return_value = {'id': 0,
'jsonrpc': '2.0',
'result': {'key': phrase}}
seed = self.wallet.seed()
self.assertIsInstance(seed, Seed)
self.assertEqual(seed.phrase, phrase)
@patch('monero.backends.jsonrpc.requests.post')
def test_balance(self, mock_post):
mock_post.return_value.status_code = 200