def test_dict_in_ctx(self):
hex_contract_address = '6690b6638251be951dded8c537678200a470c679'
bool_value = True
int_value = 100
str_value = 'value3'
dict_value = {'key': 'value'}
list_value = [1, 10, 1024, [1, 10, 1024, [1, 10, 1024]]]
dict_msg = {
'key': dict_value,
'key1': int_value,
'key2': str_value,
'key3': bool_value,
'key4': list_value
}
func = NeoInvokeFunction('testMapInMap')
func.set_params_value(dict_msg)
tx_hash = sdk.rpc.send_neo_vm_transaction(hex_contract_address, None,
acct1, self.gas_price,
self.gas_limit, func, False)
time.sleep(randint(10, 15))
event = sdk.rpc.get_contract_event_by_tx_hash(tx_hash)
states = Event.get_states_by_contract_address(event,
hex_contract_address)
states[0] = NeoData.to_utf8_str(states[0])
self.assertEqual('mapInfo', states[0])
states[1] = NeoData.to_dict(states[1])
self.assertTrue(isinstance(states[1], dict))
def test_notify(self):
hex_contract_address = '6690b6638251be951dded8c537678200a470c679'
notify_args = NeoInvokeFunction('testHello')
bool_msg = True
int_msg = 1
bytes_msg = b'Hello'
str_msg = 'Hello'
bytes_address_msg = acct1.get_address().to_bytes()
notify_args.set_params_value(bool_msg, int_msg, bytes_msg, str_msg,
bytes_address_msg)
tx_hash = self.send_tx(hex_contract_address, None, acct1, notify_args)
if len(tx_hash) == 0:
return
time.sleep(randint(10, 15))
event = sdk.rpc.get_contract_event_by_tx_hash(tx_hash)
states = Event.get_states_by_contract_address(event,
hex_contract_address)
states[0] = NeoData.to_utf8_str(states[0])
self.assertEqual('testHello', states[0])
states[1] = NeoData.to_bool(states[1])
self.assertEqual(bool_msg, states[1])
states[2] = NeoData.to_int(states[2])
self.assertEqual(int_msg, states[2])
states[3] = NeoData.to_bytes(states[3])
self.assertEqual(bytes_msg, states[3])
states[4] = NeoData.to_utf8_str(states[4])
self.assertEqual(str_msg, states[4])
states[5] = NeoData.to_b58_address(states[5])
self.assertEqual(acct1.get_address_base58(), states[5])
def new_transfer_multi_tx(self, transfer_list: list,
payer: Union[str, bytes,
Address], gas_price: int,
gas_limit: int) -> InvokeTransaction:
"""
This interface is used to generate a transaction which can
transfer amount of token from from-account to to-account multiple times.
"""
func = NeoInvokeFunction('transferMulti')
for index, item in enumerate(transfer_list):
if not isinstance(item[2], int):
raise SDKException(
ErrorCode.param_err(
'the data type of value should be int.'))
if item[2] < 0:
raise SDKException(
ErrorCode.param_err(
'the value should be equal or great than 0.'))
transfer_list[index] = [
Address.b58decode(item[0]),
Address.b58decode(item[1]), item[2]
]
for item in transfer_list:
func.add_params_value(item)
params = InvokeTransaction.generate_neo_vm_invoke_code(
self._contract_address, func)
tx = InvokeTransaction(payer, gas_price, gas_limit, params)
return tx
def test_transfer_multi_args_0(self):
transfer_1 = [
acct1.get_address().to_bytes(),
acct2.get_address().to_bytes(), 10
]
transfer_2 = [
acct2.get_address().to_bytes(),
acct3.get_address().to_bytes(), 100
]
transfer_list = [transfer_1, transfer_2]
hex_contract_address = 'ca91a73433c016fbcbcf98051d385785a6a5d9be'
func = NeoInvokeFunction('transfer_multi')
func.set_params_value(transfer_list)
tx_hash = sdk.rpc.send_neo_vm_transaction(hex_contract_address, acct1,
acct2, self.gas_price,
self.gas_limit, func, False)
self.assertEqual(64, len(tx_hash))
time.sleep(randint(10, 15))
event = sdk.rpc.get_contract_event_by_tx_hash(tx_hash)
states = Event.get_states_by_contract_address(event,
hex_contract_address)
states[0] = NeoData.to_utf8_str(states[0])
states[1][0][0] = NeoData.to_b58_address(states[1][0][0])
self.assertEqual(acct1.get_address_base58(), states[1][0][0])
states[1][0][1] = NeoData.to_b58_address(states[1][0][1])
self.assertEqual(acct2.get_address_base58(), states[1][0][1])
states[1][0][2] = NeoData.to_int(states[1][0][2])
self.assertEqual(10, states[1][0][2])
states[1][1][0] = NeoData.to_b58_address(states[1][1][0])
self.assertEqual(acct2.get_address_base58(), states[1][1][0])
states[1][1][1] = NeoData.to_b58_address(states[1][1][1])
self.assertEqual(acct3.get_address_base58(), states[1][1][1])
states[1][1][2] = NeoData.to_int(states[1][1][2])
self.assertEqual(100, states[1][1][2])
def test_notify_pre_exec(self):
bool_msg = True
int_msg = 1024
list_msg = [1, 1024, 2048]
str_msg = 'Hello'
bytes_address_msg = acct1.get_address().to_bytes()
hex_contract_address = '4855735ffadad50e7000d73e1c4e96f38d225f70'
func = NeoInvokeFunction('notify_args')
func.set_params_value(bool_msg, int_msg, list_msg, str_msg,
bytes_address_msg)
sdk.rpc.set_address('http://polaris5.ont.io:20336')
response = sdk.rpc.send_neo_vm_tx_pre_exec(hex_contract_address, func)
self.assertEqual(1, response['State'])
self.assertEqual(20000, response['Gas'])
notify = response['Notify'][0]
self.assertEqual(hex_contract_address, notify['ContractAddress'])
states = notify['States']
states[0] = NeoData.to_utf8_str(states[0])
self.assertEqual('notify args', states[0])
states[1] = NeoData.to_bool(states[1])
self.assertEqual(True, states[1])
states[2] = NeoData.to_int(states[2])
self.assertEqual(int_msg, states[2])
states[3] = NeoData.to_int_list(states[3])
self.assertEqual(list_msg, states[3])
states[4] = NeoData.to_utf8_str(states[4])
self.assertEqual(str_msg, states[4])
states[5] = NeoData.to_b58_address(states[5])
self.assertEqual(acct1.get_address_base58(), states[5])
def new_revoke_tx(self, claim_id: str, issuer: Union[str, bytes, Address],
payer: Union[str, bytes,
Address], gas_price: int, gas_limit: int):
func = NeoInvokeFunction('Revoke')
func.set_params_value(claim_id, Address.b58decode(issuer))
tx = InvokeTransaction(Address.b58decode(payer), gas_price, gas_limit)
tx.add_invoke_code(self.__hex_contract_address, func)
return tx
def test_get_dict_in_ctx(self):
hex_contract_address = '6690b6638251be951dded8c537678200a470c679'
key = 'key'
func = NeoInvokeFunction('testGetMapInMap')
func.set_params_value(key)
result = sdk.rpc.send_neo_vm_tx_pre_exec(hex_contract_address, func)
value = result['Result']
value = NeoData.to_utf8_str(value)
self.assertEqual('value', value)
def test_oep4_name(self):
hex_contract_address = '1ddbb682743e9d9e2b71ff419e97a9358c5c4ee9'
func = NeoInvokeFunction('name')
self.assertEqual(bytearray(b'\x00\xc1\x04name'),
func.create_invoke_code())
result = sdk.rpc.send_neo_vm_tx_pre_exec(hex_contract_address, func)
name = result['Result']
name = NeoData.to_utf8_str(name)
self.assertEqual('DXToken', name)
def new_allowance_tx(
self, owner: Union[str, bytes, Address],
spender: Union[str, bytes, Address]) -> InvokeTransaction:
func = NeoInvokeFunction('allowance')
func.set_params_value(Address.b58decode(owner),
Address.b58decode(spender))
tx = InvokeTransaction()
tx.add_invoke_code(self._contract_address, func)
return tx
def test_oep4_symbol(self):
hex_contract_address = '1ddbb682743e9d9e2b71ff419e97a9358c5c4ee9'
func = NeoInvokeFunction('symbol')
self.assertEqual(bytearray(b'\x00\xc1\x06symbol'),
func.create_invoke_code())
result = sdk.rpc.send_neo_vm_tx_pre_exec(hex_contract_address, func)
symbol = result['Result']
symbol = NeoData.to_utf8_str(symbol)
self.assertEqual('DX', symbol)
def new_balance_of_tx(
self, owner: Union[str, bytes, Address]) -> InvokeTransaction:
"""
This interface is used to generate transaction which can get the account balance of another account with owner address.
"""
func = NeoInvokeFunction('balanceOf')
func.set_params_value(Address.b58decode(owner))
tx = InvokeTransaction()
tx.add_invoke_code(self._contract_address, func)
return tx
def test_invoke_transaction(self):
"""
from dna.interop.System.Runtime import Notify
def main(operation, args):
if operation == 'hello':
return hello(args[0])
return False
def hello(msg):
Notify(["hello", msg])
return msg
"""
avm_code = '51c56b6c58c56b6a00527ac46a51527ac46a52527ac46a51c30568656c6c6f7d9c7c756427' \
'00006a53527ac46a52c300c3516a53c3936a53527ac46a53c36a00c365f2006c7566620300' \
'006c75660111c56b6a00527ac46a51527ac46a51c300947600a0640c00c16a52527ac4620e' \
'007562030000c56a52527ac46a52c3c0517d9c7c75641c00006a53527ac46a52c300c36a54' \
'527ac4516a55527ac4625c006a52c3c0527d9c7c756421006a52c300c36a53527ac46a52c3' \
'51c36a54527ac4516a55527ac4616232006a52c3c0537d9c7c756424006a52c300c36a5352' \
'7ac46a52c351c36a54527ac46a52c352c36a55527ac462050000f100c176c96a56527ac46a' \
'53c36a57527ac46a57c36a54c37d9f7c756419006a56c36a57c3c86a57c36a55c3936a5752' \
'7ac462e0ff6a56c36c756656c56b6a00527ac46a51527ac46a52527ac46203000568656c6c' \
'6f6a52c352c176c9681553797374656d2e52756e74696d652e4e6f746966796a52c36c7566'
contract_address = sdk.neo_vm.address_from_avm_code(avm_code).hex()
self.assertEqual('f7b9970fd6def5229c1f30ad15372bd1c20bb260',
contract_address)
hello = NeoInvokeFunction('hello')
hello.set_params_value('dna')
tx = sdk.neo_vm.make_invoke_transaction(contract_address, hello,
acct1.get_address_base58(),
500, 20000)
response = sdk.rpc.send_raw_transaction_pre_exec(tx)
self.assertEqual(1, response['State'])
response['Result'] = NeoData.to_utf8_str(response['Result'])
self.assertEqual('dna', response['Result'])
tx.sign_transaction(acct1)
tx_hash = sdk.rpc.send_raw_transaction(tx)
sleep(10)
for _ in range(5):
try:
event = sdk.rpc.get_contract_event_by_tx_hash(tx_hash)
if isinstance(event, dict) and event.get('Notify', '') != '':
notify = Event.get_notify_by_contract_address(
event, contract_address)
self.assertEqual(contract_address,
notify['ContractAddress'])
self.assertEqual('hello',
NeoData.to_utf8_str(notify['States'][0]))
self.assertEqual('dna',
NeoData.to_utf8_str(notify['States'][1]))
break
except SDKException:
continue
sleep(2)
def new_commit_tx(self, claim_id: str, issuer_address: Union[str, bytes,
Address],
owner_ont_id: str, payer_address: Union[str, bytes,
Address],
gas_price: int, gas_limit: int) -> InvokeTransaction:
func = NeoInvokeFunction('Commit')
func.set_params_value(claim_id, Address.b58decode(issuer_address),
owner_ont_id)
tx = InvokeTransaction(Address.b58decode(payer_address), gas_price,
gas_limit)
tx.add_invoke_code(self.__hex_contract_address, func)
return tx
def new_transfer_tx(self, from_address: Union[str, Address],
to_address: Union[str, Address], amount: int,
payer: Union[str, Address], gas_price: int,
gas_limit: int) -> InvokeTransaction:
"""
This interface is used to generate a transaction which can transfer amount of tokens to to_address.
"""
func = NeoInvokeFunction('transfer')
func.set_params_value(Address.b58decode(from_address),
Address.b58decode(to_address), amount)
params = InvokeTransaction.generate_neo_vm_invoke_code(
self._contract_address, func)
tx = InvokeTransaction(payer, gas_price, gas_limit, params)
return tx
def test_oep4_decimal(self):
hex_contract_address = '1ddbb682743e9d9e2b71ff419e97a9358c5c4ee9'
func = NeoInvokeFunction('decimals')
result = sdk.rpc.send_neo_vm_tx_pre_exec(hex_contract_address, func)
decimals = result['Result']
decimals = NeoData.to_int(decimals)
self.assertEqual(10, decimals)
def new_transfer_from_tx(self, spender: Union[str, bytes, Address],
owner: Union[str, bytes, Address],
to_address: Union[str, bytes, Address],
value: int, payer: Union[str, bytes,
Address], gas_price: int,
gas_limit: int) -> InvokeTransaction:
func = NeoInvokeFunction('transferFrom')
if not isinstance(value, int):
raise SDKException(
ErrorCode.param_err('the data type of value should be int.'))
func.set_params_value(Address.b58decode(spender),
Address.b58decode(owner),
Address.b58decode(to_address), value)
tx = InvokeTransaction(payer, gas_price, gas_limit)
tx.add_invoke_code(self._contract_address, func)
return tx
def test_list(self):
hex_contract_address = '6690b6638251be951dded8c537678200a470c679'
list_msg = [1, 10, 1024, [1, 10, 1024, [1, 10, 1024]]]
func = NeoInvokeFunction('testList')
func.set_params_value(list_msg)
tx_hash = self.send_tx(hex_contract_address, None, acct1, func)
if len(tx_hash) == 0:
return
time.sleep(randint(10, 15))
event = sdk.rpc.get_contract_event_by_tx_hash(tx_hash)
states = Event.get_states_by_contract_address(event,
hex_contract_address)
states[0] = NeoData.to_utf8_str(states[0])
self.assertEqual('testMsgList', states[0])
states[1] = NeoData.to_int_list(states[1])
self.assertEqual(list_msg, states[1])
def test_oep4_total_supply(self):
hex_contract_address = '1ddbb682743e9d9e2b71ff419e97a9358c5c4ee9'
func = NeoInvokeFunction('totalSupply')
result = sdk.default_network.send_neo_vm_tx_pre_exec(
hex_contract_address, func)
total_supply = result['Result']
total_supply = NeoData.to_int(total_supply)
self.assertEqual(10000000000000000000, total_supply)
def new_total_supply_tx(self) -> InvokeTransaction:
"""
This interface is used to generate transaction which can get the total token supply.
"""
tx = InvokeTransaction()
tx.add_invoke_code(self._contract_address,
NeoInvokeFunction('totalSupply'))
return tx
def new_init_tx(self, payer: Union[str, bytes, Address], gas_price: int,
gas_limit: int) -> InvokeTransaction:
"""
This interface is used to call the TotalSupply method in ope4
that initialize smart contract parameter.
"""
tx = InvokeTransaction(payer, gas_price, gas_limit)
tx.add_invoke_code(self._contract_address, NeoInvokeFunction('init'))
return tx
def test_oep4_transfer(self):
hex_contract_address = '1ddbb682743e9d9e2b71ff419e97a9358c5c4ee9'
func = NeoInvokeFunction('transfer')
bytes_from_address = acct1.get_address().to_bytes()
bytes_to_address = acct2.get_address().to_bytes()
value = 1
func.set_params_value(bytes_from_address, bytes_to_address, value)
tx_hash = self.send_tx(hex_contract_address, acct1, acct2, func)
if len(tx_hash) == 0:
return
time.sleep(randint(10, 15))
event = sdk.rpc.get_contract_event_by_tx_hash(tx_hash)
states = Event.get_states_by_contract_address(event,
hex_contract_address)
states[0] = NeoData.to_utf8_str(states[0])
self.assertEqual('transfer', states[0])
states[1] = NeoData.to_b58_address(states[1])
self.assertEqual(acct1.get_address().b58encode(), states[1])
states[2] = NeoData.to_b58_address(states[2])
self.assertEqual(acct2.get_address().b58encode(), states[2])
states[3] = NeoData.to_int(states[3])
self.assertEqual(value, states[3])
def new_approve_tx(self, owner: Union[str, bytes,
Address], spender: Union[str, bytes,
Address],
amount: int, payer: Union[str, bytes, Address],
gas_price: int, gas_limit: int) -> InvokeTransaction:
"""
This interface is used to generate a transaction which allows spender to
withdraw from owner account multiple times, up to the _value amount.
If this function is called again it overwrites the current allowance with amount value.
"""
if not isinstance(amount, int):
raise SDKException(
ErrorCode.param_err('the data type of amount should be int.'))
if amount < 0:
raise SDKException(
ErrorCode.param_err(
'the amount should be equal or great than 0.'))
func = NeoInvokeFunction('approve')
func.set_params_value(Address.b58decode(owner),
Address.b58decode(spender), amount)
params = InvokeTransaction.generate_neo_vm_invoke_code(
self._contract_address, func)
tx = InvokeTransaction(payer, gas_price, gas_limit, params)
return tx
def test_dict(self):
hex_contract_address = '6690b6638251be951dded8c537678200a470c679'
dict_msg = {'key': 'value'}
func = NeoInvokeFunction('testMap')
func.set_params_value(dict_msg)
result = sdk.rpc.send_neo_vm_tx_pre_exec(hex_contract_address, func)
dict_value = result['Result']
dict_value = NeoData.to_utf8_str(dict_value)
self.assertEqual('value', dict_value)
list_value = [1, 10, 1024, [1, 10, 1024, [1, 10, 1024]]]
dict_msg = {'key': list_value}
func = NeoInvokeFunction('testMap')
func.set_params_value(dict_msg)
result = sdk.rpc.send_neo_vm_tx_pre_exec(hex_contract_address, func)
dict_value = result['Result']
dict_value = NeoData.to_int_list(dict_value)
self.assertEqual(list_value, dict_value)
def test_oep4_transfer_multi(self):
hex_contract_address = '1ddbb682743e9d9e2b71ff419e97a9358c5c4ee9'
bytes_from_address1 = acct1.get_address().to_bytes()
bytes_to_address1 = acct2.get_address().to_bytes()
value1 = 2
transfer1 = [bytes_from_address1, bytes_to_address1, value1]
bytes_from_address2 = acct2.get_address().to_bytes()
bytes_to_address2 = acct3.get_address().to_bytes()
value2 = 1
transfer2 = [bytes_from_address2, bytes_to_address2, value2]
func = NeoInvokeFunction('transferMulti')
func.set_params_value(transfer1, transfer2)
tx_hash = sdk.default_network.send_neo_vm_transaction(
hex_contract_address, acct1, acct2, self.gas_price, self.gas_limit,
func, False)
time.sleep(randint(10, 15))
event = sdk.rpc.get_contract_event_by_tx_hash(tx_hash)
states_list = Event.get_states_by_contract_address(
event, hex_contract_address)
states_list[0][0] = NeoData.to_utf8_str(states_list[0][0])
self.assertEqual('transfer', states_list[0][0])
states_list[0][1] = NeoData.to_b58_address(states_list[0][1])
self.assertEqual(acct1.get_address().b58encode(), states_list[0][1])
states_list[0][2] = NeoData.to_b58_address(states_list[0][2])
self.assertEqual(acct2.get_address().b58encode(), states_list[0][2])
states_list[0][3] = NeoData.to_int(states_list[0][3])
self.assertEqual(value1, states_list[0][3])
states_list[1][0] = NeoData.to_utf8_str(states_list[1][0])
self.assertEqual('transfer', states_list[1][0])
states_list[1][1] = NeoData.to_b58_address(states_list[1][1])
self.assertEqual(acct2.get_address().b58encode(), states_list[1][1])
states_list[1][2] = NeoData.to_b58_address(states_list[1][2])
self.assertEqual(acct3.get_address().b58encode(), states_list[1][2])
states_list[1][3] = NeoData.to_int(states_list[1][3])
self.assertEqual(value2, states_list[1][3])
def test_oep4_balance_of(self):
hex_contract_address = '1ddbb682743e9d9e2b71ff419e97a9358c5c4ee9'
func = NeoInvokeFunction('balanceOf')
self.assertEqual(bytearray(b'\x00\xc1\tbalanceOf'),
func.create_invoke_code())
bytes_address = acct1.get_address().to_bytes()
func.set_params_value(bytes_address)
target = bytearray(
b'\x14F\xb1\xa1\x8a\xf6\xb7\xc9\xf8\xa4`/\x9fs\xee\xb3\x03\x0f\x0c)\xb7Q\xc1\tbalanceOf'
)
self.assertEqual(target, func.create_invoke_code())
result = sdk.rpc.send_neo_vm_tx_pre_exec(hex_contract_address, func)
balance = result['Result']
balance = NeoData.to_int(balance)
self.assertGreater(balance, 100)
def new_get_status_tx(self, claim_id: str) -> InvokeTransaction:
func = NeoInvokeFunction('GetStatus')
func.set_params_value(claim_id)
tx = InvokeTransaction()
tx.add_invoke_code(self.__hex_contract_address, func)
return tx
def __new_token_setting_tx(self, func_name: str) -> InvokeTransaction:
func = NeoInvokeFunction(func_name)
tx = InvokeTransaction()
tx.add_invoke_code(self._contract_address, func)
return tx