def test_transfer_eth_from_contract(self):
contract_address_string = "0x0b55929f4095f677C9Ec1F4810C3E59CCD6D33C7"
contract_address = eth_utils.to_int(hexstr=contract_address_string)
evm_code = make_evm_transfer_code(dest_address_string, 2000)
contract_a = make_contract(evm_code, "address",
contract_address_string)
arbinfo = contract_templates.get_info_contract()
arbinfo_abi = ContractABI(arbinfo)
vm = create_evm_vm([contract_a], False, False)
output_handler = create_output_handler([contract_a, arbinfo_abi])
inbox = value.Tuple([])
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple(
[1, 2345,
value.Tuple([contract_address,
2000])]) # type # sender
)),
)
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple([0, address,
contract_a.testMethod(0,
0)]) # type # sender
)),
)
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple([
CALL_TX_TYPE,
address,
arbinfo_abi.call_getBalance(dest_address_string),
]) # type # sender
)),
)
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple([
CALL_TX_TYPE,
address,
arbinfo_abi.call_getBalance(contract_address_string),
]) # type # sender
)),
)
vm.env.messages = inbox
run_until_block(vm, self)
self.assertEqual(len(vm.logs), 4)
parsed_outs = [output_handler(log) for log in vm.logs]
self.assertIsInstance(parsed_outs[0], EVMStop)
self.assertIsInstance(parsed_outs[1], EVMReturn)
self.assertIsInstance(parsed_outs[2], EVMReturn)
self.assertIsInstance(parsed_outs[3], EVMReturn)
self.assertEqual(
eth_utils.to_int(hexstr=parsed_outs[1].output_values[0]), 1)
dest_balance = parsed_outs[2].output_values[0]
contract_balance = parsed_outs[3].output_values[0]
self.assertEqual(dest_balance, 2000)
self.assertEqual(contract_balance, 0)
def test_eth(self):
contract_a = make_contract("", "uint256")
vm = create_evm_vm([contract_a], False, False)
arbsys = contract_templates.get_arbsys()
arbsys_abi = ContractABI(arbsys)
arbinfo = contract_templates.get_info_contract()
arbinfo_abi = ContractABI(arbinfo)
output_handler = create_output_handler([contract_a, arbinfo_abi])
inbox = value.Tuple([])
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple(
[1, 2345, value.Tuple([address, 100000])]
) # type # sender
)
),
)
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple(
[
CALL_TX_TYPE,
address,
arbinfo_abi.call_getBalance(address_string),
]
) # type # sender
)
),
)
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple(
[
0,
address,
arbsys_abi.withdrawEth(0, 0, dest_address_string, 150000),
]
) # type # sender
)
),
)
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple(
[
0,
address,
arbsys_abi.withdrawEth(1, 0, dest_address_string, 50000),
]
) # type # sender
)
),
)
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple(
[
CALL_TX_TYPE,
address,
arbinfo_abi.call_getBalance(address_string),
]
) # type # sender
)
),
)
vm.env.messages = inbox
run_until_block(vm, self)
self.assertEqual(len(vm.logs), 5)
parsed_out0 = output_handler(vm.logs[0])
parsed_out1 = output_handler(vm.logs[1])
parsed_out2 = output_handler(vm.logs[2])
parsed_out3 = output_handler(vm.logs[3])
parsed_out4 = output_handler(vm.logs[4])
self.assertIsInstance(parsed_out0, EVMStop)
self.assertIsInstance(parsed_out1, EVMReturn)
self.assertIsInstance(parsed_out2, EVMRevert)
self.assertIsInstance(parsed_out3, EVMStop)
self.assertIsInstance(parsed_out4, EVMReturn)
self.assertEqual(parsed_out1.output_values[0], 100000)
self.assertEqual(parsed_out4.output_values[0], 50000)
self.assertEqual(len(vm.sent_messages), 1)
self.assertEqual(
vm.sent_messages[0],
value.Tuple([1, address, value.Tuple([dest_address, 50000])]),
)
def test_insuffient_balance(self):
contract_a = make_contract("", "uint256")
vm = create_evm_vm([contract_a], False, False)
arbinfo = contract_templates.get_info_contract()
arbinfo_abi = ContractABI(arbinfo)
output_handler = create_output_handler([contract_a, arbinfo_abi])
inbox = value.Tuple([])
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple([1, 2345,
value.Tuple([address,
5000])]) # type # sender
)),
)
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple([
CALL_TX_TYPE,
address,
arbinfo_abi.call_getBalance(address_string),
]) # type # sender
)),
)
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple([
CALL_TX_TYPE,
address,
arbinfo_abi.call_getBalance(dest_address_string),
]) # type # sender
)),
)
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple([
0,
address,
value.Tuple(
[dest_address, 0, 10000,
bytestack_frombytes(b"")]),
]) # type # sender
)),
)
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple([
CALL_TX_TYPE,
address,
arbinfo_abi.call_getBalance(address_string),
]) # type # sender
)),
)
inbox = messagestack.addMessage(
inbox,
value.Tuple(
make_msg_val(
value.Tuple([
CALL_TX_TYPE,
address,
arbinfo_abi.call_getBalance(dest_address_string),
]) # type # sender
)),
)
vm.env.messages = inbox
run_until_block(vm, self)
self.assertEqual(len(vm.logs), 6)
parsed_outs = [output_handler(log) for log in vm.logs]
self.assertIsInstance(parsed_outs[0], EVMStop)
self.assertIsInstance(parsed_outs[1], EVMReturn)
self.assertIsInstance(parsed_outs[2], EVMReturn)
self.assertIsInstance(parsed_outs[3], EVMRevert)
self.assertIsInstance(parsed_outs[4], EVMReturn)
self.assertIsInstance(parsed_outs[5], EVMReturn)
self.assertEqual(len(vm.sent_messages), 0)
self.assertEqual(parsed_outs[1].output_values[0],
parsed_outs[4].output_values[0])
self.assertEqual(parsed_outs[2].output_values[0],
parsed_outs[5].output_values[0])