Return Seed object from the wallet
This commit is contained in:
parent
94656dc771
commit
a91b975e4d
|
@ -10,6 +10,7 @@ from .. import exceptions
|
||||||
from ..account import Account
|
from ..account import Account
|
||||||
from ..address import address, Address, SubAddress
|
from ..address import address, Address, SubAddress
|
||||||
from ..numbers import from_atomic, to_atomic, PaymentID
|
from ..numbers import from_atomic, to_atomic, PaymentID
|
||||||
|
from ..seed import Seed
|
||||||
from ..transaction import Transaction, IncomingPayment, OutgoingPayment
|
from ..transaction import Transaction, IncomingPayment, OutgoingPayment
|
||||||
|
|
||||||
_log = logging.getLogger(__name__)
|
_log = logging.getLogger(__name__)
|
||||||
|
@ -133,7 +134,7 @@ class JSONRPCWallet(object):
|
||||||
return self.raw_request('query_key', {'key_type': 'view_key'})['key']
|
return self.raw_request('query_key', {'key_type': 'view_key'})['key']
|
||||||
|
|
||||||
def seed(self):
|
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):
|
def accounts(self):
|
||||||
accounts = []
|
accounts = []
|
||||||
|
|
|
@ -10,6 +10,7 @@ from monero.wallet import Wallet
|
||||||
from monero.account import Account
|
from monero.account import Account
|
||||||
from monero.address import Address, address
|
from monero.address import Address, address
|
||||||
from monero.numbers import PaymentID
|
from monero.numbers import PaymentID
|
||||||
|
from monero.seed import Seed
|
||||||
from monero.transaction import IncomingPayment, OutgoingPayment, Transaction
|
from monero.transaction import IncomingPayment, OutgoingPayment, Transaction
|
||||||
from monero.backends.jsonrpc import JSONRPCWallet
|
from monero.backends.jsonrpc import JSONRPCWallet
|
||||||
|
|
||||||
|
@ -34,6 +35,22 @@ class SubaddrWalletTestCase(unittest.TestCase):
|
||||||
'total_balance': 236153709446071,
|
'total_balance': 236153709446071,
|
||||||
'total_unlocked_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')
|
@patch('monero.backends.jsonrpc.requests.post')
|
||||||
def test_balance(self, mock_post):
|
def test_balance(self, mock_post):
|
||||||
mock_post.return_value.status_code = 200
|
mock_post.return_value.status_code = 200
|
||||||
|
|
Loading…
Reference in New Issue