def test_sendmany_good_simple(self):
nodemgr = NodeManager()
nodemgr.reset_for_test()
nodemgr.nodes = [NeoNode(object, object)]
with patch('neo.Network.node.NeoNode.relay',
return_value=self.async_return(True)):
with patch('sys.stdout', new=StringIO()) as mock_print:
with patch('neo.Prompt.Commands.Send.prompt',
side_effect=[
"neo", self.watch_addr_str, "1", "gas",
self.watch_addr_str, "1",
UserWalletTestCase.wallet_1_pass()
]):
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['sendmany', '2']
res = Wallet.CommandWallet().execute(args)
self.assertTrue(res) # verify successful tx
self.assertIn("Sending with fee: 0", mock_print.getvalue())
json_res = res.ToJson()
# check for 2 transfers
transfers = 0
for info in json_res['vout']:
if info['address'] == self.watch_addr_str:
transfers += 1
self.assertEqual(2, transfers)
def test_sendmany_no_wallet(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
args = ['sendmany', '2']
Wallet.CommandWallet().execute(args)
self.assertIn("Please open a wallet", mock_print.getvalue())
def test_send_with_fee_and_from_addr(self):
nodemgr = NodeManager()
nodemgr.reset_for_test()
nodemgr.nodes = [NeoNode(object, object)]
with patch('neo.Network.node.NeoNode.relay',
return_value=self.async_return(True)):
with patch('sys.stdout', new=StringIO()) as mock_print:
with patch('neo.Prompt.Commands.Send.prompt',
side_effect=[UserWalletTestCase.wallet_1_pass()]):
PromptData.Wallet = self.GetWallet1(recreate=True)
args = [
'send', 'neo', self.watch_addr_str, '1',
'--from-addr=AJQ6FoaSXDFzA6wLnyZ1nFN7SGSN2oNTc3',
'--fee=0.005'
]
res = Wallet.CommandWallet().execute(args)
self.assertTrue(res) # verify successful tx
json_res = res.ToJson()
self.assertEqual(self.watch_addr_str, json_res['vout'][0]
['address']) # verify correct address_to
self.assertEqual(
self.wallet_1_addr, json_res['vout'][1]
['address']) # verify correct address_from
self.assertEqual(json_res['net_fee'],
"0.005") # verify correct fee
self.assertIn("Sending with fee: 0.005",
mock_print.getvalue())
def test_send_no_wallet(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
args = ["send", "neo", self.wallet_1_addr, '5']
Wallet.CommandWallet().execute(args)
self.assertIn("Please open a wallet", mock_print.getvalue())
def test_send_token_ok(self):
nodemgr = NodeManager()
nodemgr.reset_for_test()
nodemgr.nodes = [NeoNode(object, object)]
with patch('neo.Network.node.NeoNode.relay',
return_value=self.async_return(True)):
with patch('neo.Prompt.Commands.Tokens.prompt',
side_effect=[UserWalletTestCase.wallet_1_pass()]):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
token_hash = '31730cc9a1844891a3bafd1aa929a4142860d8d3'
ImportToken(PromptData.Wallet, token_hash)
args = [
'send', 'NXT4', self.watch_addr_str, '30',
'--from-addr=%s' % self.wallet_1_addr
]
res = Wallet.CommandWallet().execute(args)
self.assertTrue(res)
self.assertIn(
"Will transfer 30.00000000 NXT4 from AJQ6FoaSXDFzA6wLnyZ1nFN7SGSN2oNTc3 to AGYaEi3W6ndHPUmW7T12FFfsbQ6DWymkEm",
mock_print.getvalue())
def test_insufficient_funds(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['send', 'gas', self.watch_addr_str, '72620']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("Insufficient funds", mock_print.getvalue())
def test_send_bad_precision_amount(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['send', 'neo', self.watch_addr_str, '12.01']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("incorrect amount precision", mock_print.getvalue())
def test_send_zero_amount(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['send', 'neo', self.watch_addr_str, '0']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("Amount cannot be 0", mock_print.getvalue())
def test_send_bad_assetid(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['send', 'blah', self.watch_addr_str, '12']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("Asset id not found", mock_print.getvalue())
def test_sendmany_weird_outgoing(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['sendmany', '0.5'] # weird number outgoing
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("Invalid outgoing number", mock_print.getvalue())
def test_send_weird_fee(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['send', 'neo', self.watch_addr_str, '12', '--fee=0.0abc']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("invalid amount format", mock_print.getvalue())
def test_sendmany_bad_outgoing(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['sendmany', '0'] # too few outgoing
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("Outgoing number must be >= 1",
mock_print.getvalue())
def test_bad_password(self):
with patch('neo.Prompt.Commands.Send.prompt', side_effect=['blah']):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['send', 'neo', self.watch_addr_str, '50']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("Incorrect password", mock_print.getvalue())
def test_sendmany_bad_args(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['sendmany'] # too few args
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("Please specify the required parameter",
mock_print.getvalue())
def test_keyboard_interrupt(self):
with patch('neo.Prompt.Commands.Send.prompt',
side_effect=[KeyboardInterrupt]):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['send', 'neo', self.watch_addr_str, '50']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("Transaction cancelled", mock_print.getvalue())
def test_send_gas(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
with patch('neo.Prompt.Commands.Send.prompt',
side_effect=[UserWalletTestCase.wallet_1_pass()]):
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['send', 'gas', self.watch_addr_str, '5']
res = Wallet.CommandWallet().execute(args)
self.assertTrue(res)
self.assertIn("Sending with fee: 0.0", mock_print.getvalue())
def test_send_bad_address_to(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
address_to = 'AGYaEi3W6ndHPUmW7T12FFfsbQ6DWymkE' # address_to is too short causing ToScriptHash to fail
args = ['send', 'neo', address_to, '12']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("Not correct Address, wrong length",
mock_print.getvalue())
def test_send_bad_address_from(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
address_from = '--from-addr=AJQ6FoaSXDFzA6wLnyZ1nFN7SGSN2oNTc' # address_from is too short causing ToScriptHash to fail
args = ['send', 'neo', self.watch_addr_str, '12', address_from]
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("Not correct Address, wrong length",
mock_print.getvalue())
def test_bad_attributes(self):
with patch('neo.Prompt.Commands.Send.prompt',
side_effect=[UserWalletTestCase.wallet_1_pass()]):
PromptData.Wallet = self.GetWallet1(recreate=True)
args = [
'send', 'gas', self.watch_addr_str, '2',
'--tx-attr=[{"usa:241"data":his is a remark"}]'
]
res = Wallet.CommandWallet().execute(args)
self.assertTrue(res)
self.assertEqual(1, len(res.Attributes))
def test_fails_to_relay_tx(self):
with patch('neo.Prompt.Commands.Send.prompt',
side_effect=[UserWalletTestCase.wallet_1_pass()]):
with patch('neo.Prompt.Commands.Send.NodeLeader.Relay',
return_value=False):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['send', 'gas', self.watch_addr_str, '2']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("Could not relay tx", mock_print.getvalue())
def test_owners(self, mock):
with patch('neo.Prompt.Commands.Send.prompt',
side_effect=[UserWalletTestCase.wallet_1_pass()]):
PromptData.Wallet = self.GetWallet1(recreate=True)
args = [
'send', 'gas', self.wallet_1_addr, '2',
"--owners=['AXjaFSP23Jkbe6Pk9pPGT6NBDs1HVdqaXK','APRgMZHZubii29UXF9uFa6sohrsYupNAvx']"
]
Wallet.CommandWallet().execute(args)
self.assertTrue(mock.called)
def test_could_not_send(self):
# mocking traceback module to avoid stacktrace printing during test run
with patch('neo.Prompt.Commands.Send.traceback'):
with patch('sys.stdout', new=StringIO()) as mock_print:
with patch('neo.Prompt.Commands.Send.prompt',
side_effect=[UserWalletTestCase.wallet_1_pass()]):
with patch('neo.Wallets.Wallet.Wallet.GetStandardAddress',
side_effect=[Exception]):
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['send', 'gas', self.watch_addr_str, '2']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("Could not send:", mock_print.getvalue())
def test_fails_to_sign_tx(self):
with patch('neo.Prompt.Commands.Send.prompt',
side_effect=[UserWalletTestCase.wallet_1_pass()]):
with patch('neo.Wallets.Wallet.Wallet.Sign', return_value=False):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['send', 'gas', self.watch_addr_str, '2']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn(
"Transaction initiated, but the signature is incomplete. Use the `sign` command with the information below to complete signing",
mock_print.getvalue())
def test_transaction_size_2(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
with patch('neo.Core.TX.Transaction.Transaction.Size',
return_value=1411): # returns a size of 1411
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['send', 'gas', self.watch_addr_str, '5', '--fee=0.001']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn(
'Transaction cancelled. The tx size (1411) exceeds the max free tx size (1024).\nA network fee of 0.00387 GAS is required.',
mock_print.getvalue()
) # the required fee is equal to (1411-1024) * 0.0001 (FEE_PER_EXTRA_BYTE) = 0.00387
def test_send_token_bad(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
PromptData.Wallet = self.GetWallet1(recreate=True)
token_hash = 'f8d448b227991cf07cb96a6f9c0322437f1599b9'
ImportToken(PromptData.Wallet, token_hash)
args = ['send', 'NEP5', self.watch_addr_str, '32']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("Could not find the contract hash",
mock_print.getvalue())
def test_sendmany_negative_fee(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
with patch('neo.Prompt.Commands.Send.prompt',
side_effect=[
"neo", self.watch_addr_str, "1", "gas",
self.watch_addr_str, "1"
]):
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['sendmany', '2', '--fee=-0.005']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("invalid amount format", mock_print.getvalue())
def test_transaction_size_1(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
with patch('neo.Core.TX.Transaction.Transaction.Size',
return_value=1026): # returns a size of 1026
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['send', 'gas', self.watch_addr_str, '5']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn(
'Transaction cancelled. The tx size (1026) exceeds the max free tx size (1024).\nA network fee of 0.001 GAS is required.',
mock_print.getvalue()
) # notice the required fee is equal to the low priority threshold
def test_sendmany_zero_amount(self):
with patch('sys.stdout', new=StringIO()) as mock_print:
with patch('neo.Prompt.Commands.Send.prompt',
side_effect=[
"neo", self.watch_addr_str, "1", "gas",
self.watch_addr_str, "0"
]):
PromptData.Wallet = self.GetWallet1(recreate=True)
args = ['sendmany', '2']
res = Wallet.CommandWallet().execute(args)
self.assertFalse(res)
self.assertIn("Amount cannot be 0", mock_print.getvalue())
def test_attributes(self):
with patch('neo.Prompt.Commands.Send.prompt',
side_effect=[UserWalletTestCase.wallet_1_pass()]):
PromptData.Wallet = self.GetWallet1(recreate=True)
args = [
'send', 'gas', self.watch_addr_str, '2',
'--tx-attr={"usage":241,"data":"This is a remark"}'
]
res = Wallet.CommandWallet().execute(args)
self.assertTrue(res)
self.assertEqual(
2, len(res.Attributes)
) # By default the script_hash of the transaction sender is added to the TransactionAttribute list, therefore the Attributes length is `count` + 1
def test_sendmany_good_complex(self):
nodemgr = NodeManager()
nodemgr.reset_for_test()
nodemgr.nodes = [NeoNode(object, object)]
with patch('neo.Network.node.NeoNode.relay',
return_value=self.async_return(True)):
with patch('sys.stdout', new=StringIO()) as mock_print:
with patch('neo.Prompt.Commands.Send.prompt',
side_effect=[
"neo", "AXjaFSP23Jkbe6Pk9pPGT6NBDs1HVdqaXK",
"1", "gas",
"AXjaFSP23Jkbe6Pk9pPGT6NBDs1HVdqaXK", "1",
UserWalletTestCase.wallet_1_pass()
]):
PromptData.Wallet = self.GetWallet1(recreate=True)
args = [
'sendmany', '2',
'--from-addr=%s' % self.wallet_1_addr,
'--change-addr=%s' % self.watch_addr_str, '--fee=0.005'
]
address_from_account_state = Blockchain.Default(
).GetAccountState(self.wallet_1_addr).ToJson()
address_from_gas = next(
filter(
lambda b: b['asset'] ==
'0x602c79718b16e442de58778e148d0b1084e3b2dffd5de6b7b16cee7969282de7',
address_from_account_state['balances']))
address_from_gas_bal = address_from_gas['value']
res = Wallet.CommandWallet().execute(args)
self.assertTrue(res) # verify successful tx
json_res = res.ToJson()
self.assertEqual("AXjaFSP23Jkbe6Pk9pPGT6NBDs1HVdqaXK",
json_res['vout'][0]
['address']) # verify correct address_to
self.assertEqual(
self.watch_addr_str, json_res['vout'][2]
['address']) # verify correct change address
self.assertEqual(
float(address_from_gas_bal) - 1 - 0.005,
float(json_res['vout'][3]['value']))
self.assertEqual('0.005', json_res['net_fee'])
self.assertIn("Sending with fee: 0.005",
mock_print.getvalue())