def test_list_address(self):
with set_qrl_dir("wallet_ver1"):
walletd = WalletD()
qaddress = walletd.add_new_address(height=4)
self.assertEqual(qaddress[0], 'Q')
self.assertEqual(len(walletd.list_address()), 1)
list_address = walletd.list_address()
self.assertEqual(list_address[0], qaddress)
def test_remove_address(self):
with set_qrl_dir("wallet_ver1"):
walletd = WalletD()
qaddress = walletd.add_new_address(height=4)
self.assertEqual(qaddress[0], 'Q')
self.assertEqual(len(walletd.list_address()), 1)
result = walletd.remove_address(qaddress)
self.assertTrue(result)
self.assertEqual(len(walletd.list_address()), 0)
def test_add_address_from_seed(self):
with set_qrl_dir("wallet_ver1"):
walletd = WalletD()
qaddress1 = walletd.add_address_from_seed(
seed=self.hex_seed) # Using hexseed
self.assertEqual(self.qaddress, qaddress1)
self.assertEqual(len(walletd.list_address()), 1)
def test_add_address_from_seed2(self):
with set_qrl_dir("wallet_ver1"):
walletd = WalletD()
qaddress1 = walletd.add_address_from_seed(
seed=self.hex_seed) # Using hexseed
self.assertEqual(self.qaddress, qaddress1)
self.assertEqual(len(walletd.list_address()), 1)
walletd.remove_address(self.qaddress)
self.assertEqual(len(walletd.list_address()), 0)
qaddress2 = walletd.add_address_from_seed(
seed=self.mnemonic) # Using mnemonic
self.assertEqual(self.qaddress, qaddress2)
self.assertEqual(len(walletd.list_address()), 1)
def test_unlock_wallet(self):
with set_qrl_dir("wallet_ver1"):
walletd = WalletD()
walletd.add_new_address()
walletd.encrypt_wallet(passphrase=self.passphrase)
walletd.lock_wallet()
with self.assertRaises(ValueError):
walletd.add_new_address()
with self.assertRaises(WalletDecryptionError):
walletd.unlock_wallet(passphrase='pass123')
walletd.unlock_wallet(passphrase=self.passphrase)
walletd.add_new_address()
self.assertEqual(len(walletd.list_address()), 2)
def test_get_recovery_seeds(self):
with set_qrl_dir("wallet_ver1"):
walletd = WalletD()
qaddress = walletd.add_new_address(height=4)
self.assertEqual(qaddress[0], 'Q')
seeds = walletd.get_recovery_seeds(qaddress)
self.assertIsInstance(seeds, tuple)
walletd.remove_address(qaddress)
self.assertEqual(len(walletd.list_address()), 0)
qaddress2 = walletd.add_address_from_seed(
seeds[0]) # Using Hex Seed
self.assertEqual(qaddress, qaddress2)
walletd.remove_address(qaddress2)
self.assertEqual(len(walletd.list_address()), 0)
qaddress2 = walletd.add_address_from_seed(
seeds[1]) # Using Mnemonic
self.assertEqual(qaddress, qaddress2)
walletd.remove_address(qaddress2)
self.assertEqual(len(walletd.list_address()), 0)
def test_change_passphrase(self):
with set_qrl_dir("wallet_ver1"):
walletd = WalletD()
qaddress = walletd.add_new_address()
walletd.encrypt_wallet(passphrase=self.passphrase)
walletd.lock_wallet()
passphrase2 = 'pass000'
with self.assertRaises(ValueError):
walletd.change_passphrase(old_passphrase='pass123',
new_passphrase='pass234')
walletd.change_passphrase(old_passphrase=self.passphrase,
new_passphrase=passphrase2)
with self.assertRaises(WalletDecryptionError):
walletd.unlock_wallet(passphrase=self.passphrase)
walletd.unlock_wallet(passphrase=passphrase2)
qaddresses = walletd.list_address()
self.assertEqual(len(qaddresses), 1)
self.assertEqual(qaddresses[0], qaddress)