def test_relaySlaveTxn(self):
with set_xrd_dir("wallet_ver1"):
walletd = WalletD()
service = WalletAPIService(walletd)
resp = service.AddNewAddress(xrdwallet_pb2.AddNewAddressReq(), context=None)
qaddress = resp.address
addr_state = AddressState.get_default(walletd.qaddress_to_address(qaddress))
walletd._public_stub.GetAddressState = Mock(
return_value=xrd_pb2.GetAddressStateResp(state=addr_state.pbdata))
walletd._public_stub.IsSlave = Mock(
return_value=xrd_pb2.IsSlaveResp(result=True))
walletd._public_stub.GetOTS = Mock(
return_value=xrd_pb2.GetOTSResp(next_unused_ots_index=0,
unused_ots_index_found=True))
alice_xmss = get_alice_xmss(4)
slave_pks = [alice_xmss.pk]
access_types = [0]
walletd._public_stub.PushTransaction = Mock(
return_value=xrd_pb2.PushTransactionResp(error_code=xrd_pb2.PushTransactionResp.SUBMITTED))
resp = service.RelaySlaveTxn(xrdwallet_pb2.RelaySlaveTxnReq(slave_pks=slave_pks,
access_types=access_types,
fee=100000000,
master_address=None,
signer_address=qaddress,
ots_index=0), context=None)
self.assertEqual(resp.code, 0)
self.assertIsNotNone(resp.tx)
def test_relayTokenTxnBySlave(self):
with set_xrd_dir("wallet_ver1"):
walletd = WalletD()
walletd._public_stub.PushTransaction = Mock(
return_value=xrd_pb2.PushTransactionResp(error_code=xrd_pb2.PushTransactionResp.SUBMITTED))
service = WalletAPIService(walletd)
resp = service.AddNewAddressWithSlaves(xrdwallet_pb2.AddNewAddressWithSlavesReq(), context=None)
qaddress = resp.address
addr_state = AddressState.get_default(walletd.qaddress_to_address(qaddress))
slaves = walletd.get_slave_list(qaddress)
addr_state.add_slave_pks_access_type(bytes(hstr2bin(slaves[0][0].pk)), 0)
walletd._public_stub.GetAddressState = Mock(
return_value=xrd_pb2.GetAddressStateResp(state=addr_state.pbdata))
alice_xmss = get_alice_xmss(4)
bob_xmss = get_bob_xmss(4)
qaddresses = [alice_xmss.qaddress, bob_xmss.qaddress]
amounts = [1000000000, 1000000000]
resp = service.RelayTokenTxnBySlave(
xrdwallet_pb2.RelayTokenTxnBySlaveReq(symbol=b'xrd',
name=b'Quantum Resistant Ledger',
owner=alice_xmss.qaddress,
decimals=5,
addresses=qaddresses,
amounts=amounts,
fee=100000000,
master_address=qaddress), context=None)
self.assertEqual(resp.code, 0)
self.assertIsNotNone(resp.tx)
def test_addNewAddress(self):
with set_xrd_dir("wallet_ver1"):
walletd = WalletD()
service = WalletAPIService(walletd)
resp = service.AddNewAddress(xrdwallet_pb2.AddNewAddressReq(), context=None)
self.assertEqual(resp.code, 0)
self.assertEqual(resp.address[0], 'Q')
def test_encrypt_save_reload(self):
with set_xrd_dir("no_data"):
wallet = Wallet()
wallet.add_new_address(height=4)
wallet.save()
wallet_b = Wallet()
self.assertEqual(1, len(wallet_b.address_items))
self.assertEqual(wallet.address_items[0],
wallet_b.address_items[0])
TEST_KEY = 'mytestkey'
wallet_b.encrypt(TEST_KEY)
wallet_b.save()
self.assertEqual(1, len(wallet_b.address_items))
self.assertNotEqual(wallet.address_items[0],
wallet_b.address_items[0])
wallet_c = Wallet()
self.assertEqual(1, len(wallet_c.address_items))
self.assertTrue(wallet_c.address_items[0].encrypted)
wallet_c.decrypt(TEST_KEY)
self.assertFalse(wallet_c.address_items[0].encrypted)
self.assertEqual(wallet.address_items[0],
wallet_c.address_items[0])
def test_relayMessageTxn(self):
with set_xrd_dir("wallet_ver1"):
walletd = WalletD()
walletd._public_stub.PushTransaction = Mock(
return_value=xrd_pb2.PushTransactionResp(error_code=xrd_pb2.PushTransactionResp.SUBMITTED))
service = WalletAPIService(walletd)
resp = service.AddNewAddress(xrdwallet_pb2.AddNewAddressReq(), context=None)
qaddress = resp.address
addr_state = AddressState.get_default(walletd.qaddress_to_address(qaddress))
walletd._public_stub.GetAddressState = Mock(
return_value=xrd_pb2.GetAddressStateResp(state=addr_state.pbdata))
walletd._public_stub.GetOTS = Mock(
return_value=xrd_pb2.GetOTSResp(next_unused_ots_index=0,
unused_ots_index_found=True))
resp = service.RelayMessageTxn(xrdwallet_pb2.RelayMessageTxnReq(message=b'Hello xrd!',
fee=100000000,
master_address=None,
signer_address=qaddress,
ots_index=0), context=None)
self.assertEqual(0, resp.code)
self.assertIsNotNone(resp.tx)
def setUp(self):
with set_xrd_dir('no_data'):
self.state = State()
self.alice = get_alice_xmss()
self.bob = get_bob_xmss()
alice_address_state = OptimizedAddressState.get_default(
self.alice.address)
alice_address_state.pbdata.balance = 100
self.addresses_state = {
self.alice.address: alice_address_state,
self.bob.address:
OptimizedAddressState.get_default(self.bob.address)
}
self.params = {
"token_txhash": b'I declare the TEST token',
"addrs_to": [self.bob.address],
"amounts": [100],
"fee": 1,
"xmss_pk": self.alice.pk
}
self.unused_chain_manager_mock = Mock(autospec=ChainManager,
name='unused ChainManager')
def setUp(self):
with set_xrd_dir('no_data'):
self.state = State()
self.alice = get_alice_xmss()
self.bob = get_bob_xmss()
self.alice.set_ots_index(10)
self.maxDiff = None
def test_getWalletInfo(self):
with set_xrd_dir("wallet_ver1"):
walletd = WalletD()
service = WalletAPIService(walletd)
resp = service.GetWalletInfo(xrdwallet_pb2.GetWalletInfoReq(), context=None)
self.assertEqual(resp.version, 1)
self.assertEqual(resp.address_count, 0)
self.assertFalse(resp.is_encrypted)
def test_getRecoverySeeds(self):
with set_xrd_dir("wallet_ver1"):
walletd = WalletD()
service = WalletAPIService(walletd)
service.AddAddressFromSeed(xrdwallet_pb2.AddAddressFromSeedReq(seed=self.mnemonic), context=None)
resp = service.GetRecoverySeeds(xrdwallet_pb2.GetRecoverySeedsReq(address=self.qaddress), context=None)
self.assertEqual(resp.hexseed, self.hex_seed)
self.assertEqual(resp.mnemonic, self.mnemonic)
def test_read_wallet_ver0_saves_wallet_ver1(self):
with set_xrd_dir("wallet_ver0"):
wallet = Wallet()
self.assertEqual(wallet.version, 0)
wallet.version = 1
wallet.save()
wallet_reloaded = Wallet()
self.assertEqual(wallet_reloaded.version, 1)
