def create_raw_transaction_send_token_payload(pub_key_to, amount=1):
client = AccountClient()
signer = client._signer
address = client.make_address_from_data(pub_key_to)
node_address = client.get_user_address()
transfer = TransferPayload()
transfer.address_to = address
transfer.value = amount
tr = TransactionPayload()
tr.method = AccountMethod.TRANSFER
tr.data = transfer.SerializeToString()
payload = tr.SerializeToString()
header = TransactionHeader(
signer_public_key=signer.get_public_key().as_hex(),
family_name=client._family_handler.family_name,
family_version=client._family_handler.family_versions[-1],
inputs=[node_address, address],
outputs=[node_address, address],
dependencies=[],
payload_sha512=hash512(payload),
batcher_public_key=signer.get_public_key().as_hex(),
nonce=time.time().hex().encode()).SerializeToString()
signature = signer.sign(header)
transaction = Transaction(header=header,
payload=payload,
header_signature=signature)
return transaction
def test_account_handler_apply():
"""
Case: send transaction request, to send tokens to address, to the account handler.
Expect: addresses data, stored in state, are changed according to transfer amount.
"""
expected_account_from_balance = ACCOUNT_FROM_BALANCE - TOKENS_AMOUNT_TO_SEND
expected_account_to_balance = ACCOUNT_TO_BALANCE + TOKENS_AMOUNT_TO_SEND
account_protobuf = Account()
account_protobuf.balance = expected_account_from_balance
expected_serialized_account_from_balance = account_protobuf.SerializeToString()
account_protobuf.balance = expected_account_to_balance
expected_serialized_account_to_balance = account_protobuf.SerializeToString()
expected_state = {
ACCOUNT_ADDRESS_FROM: expected_serialized_account_from_balance,
ACCOUNT_ADDRESS_TO: expected_serialized_account_to_balance,
}
transfer_payload = TransferPayload()
transfer_payload.address_to = ACCOUNT_ADDRESS_TO
transfer_payload.value = TOKENS_AMOUNT_TO_SEND
transaction_payload = TransactionPayload()
transaction_payload.method = AccountMethod.TRANSFER
transaction_payload.data = transfer_payload.SerializeToString()
serialized_transaction_payload = transaction_payload.SerializeToString()
transaction_header = TransactionHeader(
signer_public_key=RANDOM_NODE_PUBLIC_KEY,
family_name=TRANSACTION_REQUEST_ACCOUNT_HANDLER_PARAMS.get('family_name'),
family_version=TRANSACTION_REQUEST_ACCOUNT_HANDLER_PARAMS.get('family_version'),
inputs=INPUTS,
outputs=OUTPUTS,
dependencies=[],
payload_sha512=hash512(data=serialized_transaction_payload),
batcher_public_key=RANDOM_NODE_PUBLIC_KEY,
nonce=time.time().hex().encode(),
)
serialized_header = transaction_header.SerializeToString()
transaction_request = TpProcessRequest(
header=transaction_header,
payload=serialized_transaction_payload,
signature=create_signer(private_key=ACCOUNT_FROM_PRIVATE_KEY).sign(serialized_header),
)
mock_context = create_context(account_from_balance=ACCOUNT_FROM_BALANCE, account_to_balance=ACCOUNT_TO_BALANCE)
AccountHandler().apply(transaction=transaction_request, context=mock_context)
state_as_list = mock_context.get_state(addresses=[ACCOUNT_ADDRESS_TO, ACCOUNT_ADDRESS_FROM])
state_as_dict = {entry.address: entry.data for entry in state_as_list}
assert expected_state == state_as_dict
def _charge_tokens_for_storing(context, address_from, address_to):
"""
Send fixed tokens value from address that want to store public key to node's storage address.
"""
if ZERO_ADDRESS == address_from:
raise InvalidTransaction('Transactions from zero address is used only for internal purposes.')
transfer_payload = TransferPayload()
transfer_payload.address_to = address_to
transfer_payload.value = PUB_KEY_STORE_PRICE
transfer_state = AccountHandler()._transfer_from_address(
context=context, address_from=address_from, transfer_payload=transfer_payload,
)
return transfer_state
def test_account_transfer_from_address_without_previous_usage():
"""
Case: transfer tokens from address to address when them have never been used before.
Expect: invalid transaction error is raised with not enough transferable balance error message.
"""
initial_state = {
ACCOUNT_ADDRESS_FROM: None,
ACCOUNT_ADDRESS_TO: None,
}
mock_context = StubContext(inputs=INPUTS, outputs=OUTPUTS, initial_state=initial_state)
transfer_payload = TransferPayload()
transfer_payload.address_to = ACCOUNT_ADDRESS_TO
transfer_payload.value = TOKENS_AMOUNT_TO_SEND
transaction_payload = TransactionPayload()
transaction_payload.method = AccountMethod.TRANSFER
transaction_payload.data = transfer_payload.SerializeToString()
serialized_transaction_payload = transaction_payload.SerializeToString()
transaction_header = TransactionHeader(
signer_public_key=RANDOM_NODE_PUBLIC_KEY,
family_name=TRANSACTION_REQUEST_ACCOUNT_HANDLER_PARAMS.get('family_name'),
family_version=TRANSACTION_REQUEST_ACCOUNT_HANDLER_PARAMS.get('family_version'),
inputs=INPUTS,
outputs=OUTPUTS,
dependencies=[],
payload_sha512=hash512(data=serialized_transaction_payload),
batcher_public_key=RANDOM_NODE_PUBLIC_KEY,
nonce=time.time().hex().encode(),
)
serialized_header = transaction_header.SerializeToString()
transaction_request = TpProcessRequest(
header=transaction_header,
payload=serialized_transaction_payload,
signature=create_signer(private_key=ACCOUNT_FROM_PRIVATE_KEY).sign(serialized_header),
)
with pytest.raises(InvalidTransaction) as error:
AccountHandler().apply(transaction=transaction_request, context=mock_context)
assert f'Not enough transferable balance. Sender\'s current balance: 0.' == str(error.value)
def test_account_transfer_from_address_to_address_not_account_type():
"""
Case: transfer tokens from address to address that is not account type.
Expect: invalid transaction error is raised with receiver address is not account type error message.
"""
mock_context = create_context(account_from_balance=ACCOUNT_FROM_BALANCE, account_to_balance=ACCOUNT_TO_BALANCE)
transfer_payload = TransferPayload()
transfer_payload.address_to = ADDRESS_NOT_ACCOUNT_TYPE
transfer_payload.value = TOKENS_AMOUNT_TO_SEND
transaction_payload = TransactionPayload()
transaction_payload.method = AccountMethod.TRANSFER
transaction_payload.data = transfer_payload.SerializeToString()
serialized_transaction_payload = transaction_payload.SerializeToString()
transaction_header = TransactionHeader(
signer_public_key=RANDOM_NODE_PUBLIC_KEY,
family_name=TRANSACTION_REQUEST_ACCOUNT_HANDLER_PARAMS.get('family_name'),
family_version=TRANSACTION_REQUEST_ACCOUNT_HANDLER_PARAMS.get('family_version'),
inputs=INPUTS,
outputs=OUTPUTS,
dependencies=[],
payload_sha512=hash512(data=serialized_transaction_payload),
batcher_public_key=RANDOM_NODE_PUBLIC_KEY,
nonce=time.time().hex().encode(),
)
serialized_header = transaction_header.SerializeToString()
transaction_request = TpProcessRequest(
header=transaction_header,
payload=serialized_transaction_payload,
signature=create_signer(private_key=ACCOUNT_FROM_PRIVATE_KEY).sign(serialized_header),
)
with pytest.raises(InvalidTransaction) as error:
AccountHandler().apply(transaction=transaction_request, context=mock_context)
assert proto_error_msg(
TransferPayload,
{
'address_to': ['Address is not of a blockchain token type.'],
}
) == str(error.value)
def test_account_handler_with_empty_proto():
"""
Case: send transaction request with empty proto
Expect: invalid transaction error
"""
transfer_payload = TransferPayload()
transaction_payload = TransactionPayload()
transaction_payload.method = AccountMethod.TRANSFER
transaction_payload.data = transfer_payload.SerializeToString()
serialized_transaction_payload = transaction_payload.SerializeToString()
transaction_header = TransactionHeader(
signer_public_key=RANDOM_NODE_PUBLIC_KEY,
family_name=TRANSACTION_REQUEST_ACCOUNT_HANDLER_PARAMS.get('family_name'),
family_version=TRANSACTION_REQUEST_ACCOUNT_HANDLER_PARAMS.get('family_version'),
inputs=INPUTS,
outputs=OUTPUTS,
dependencies=[],
payload_sha512=hash512(data=serialized_transaction_payload),
batcher_public_key=RANDOM_NODE_PUBLIC_KEY,
nonce=time.time().hex().encode(),
)
serialized_header = transaction_header.SerializeToString()
transaction_request = TpProcessRequest(
header=transaction_header,
payload=serialized_transaction_payload,
signature=create_signer(private_key=ACCOUNT_FROM_PRIVATE_KEY).sign(serialized_header),
)
mock_context = StubContext(inputs=INPUTS, outputs=OUTPUTS, initial_state={})
with pytest.raises(InvalidTransaction) as error:
AccountHandler().apply(transaction=transaction_request, context=mock_context)
assert proto_error_msg(
TransferPayload,
{
'address_to': ['Missed address'],
'value': ['Could not transfer with zero amount.'],
}
) == str(error.value)
def test_account_handler_apply_invalid_transfer_method():
"""
Case: send transaction request, to send tokens to address, to account handler with invalid transfer method value.
Expect: invalid transaction error is raised with invalid account method value error message.
"""
account_method_impossible_value = 5347
transfer_payload = TransferPayload()
transfer_payload.address_to = ACCOUNT_ADDRESS_TO
transfer_payload.value = TOKENS_AMOUNT_TO_SEND
transaction_payload = TransactionPayload()
transaction_payload.method = account_method_impossible_value
transaction_payload.data = transfer_payload.SerializeToString()
serialized_transaction_payload = transaction_payload.SerializeToString()
transaction_header = TransactionHeader(
signer_public_key=RANDOM_NODE_PUBLIC_KEY,
family_name=TRANSACTION_REQUEST_ACCOUNT_HANDLER_PARAMS.get('family_name'),
family_version=TRANSACTION_REQUEST_ACCOUNT_HANDLER_PARAMS.get('family_version'),
inputs=INPUTS,
outputs=OUTPUTS,
dependencies=[],
payload_sha512=hash512(data=serialized_transaction_payload),
batcher_public_key=RANDOM_NODE_PUBLIC_KEY,
nonce=time.time().hex().encode(),
)
serialized_header = transaction_header.SerializeToString()
transaction_request = TpProcessRequest(
header=transaction_header,
payload=serialized_transaction_payload,
signature=create_signer(private_key=ACCOUNT_FROM_PRIVATE_KEY).sign(serialized_header),
)
mock_context = create_context(account_from_balance=ACCOUNT_FROM_BALANCE, account_to_balance=ACCOUNT_TO_BALANCE)
with pytest.raises(InvalidTransaction) as error:
AccountHandler().apply(transaction=transaction_request, context=mock_context)
assert f'Invalid account method value ({account_method_impossible_value}) has been set.' == str(error.value)
def get_transfer_payload(self, address_to, value):
transfer = TransferPayload()
transfer.address_to = address_to
transfer.value = value
return transfer
def create_transfer(cls, context, address_from, address_to, amount):
transfer_payload = TransferPayload(address_to=address_to, value=amount)
return cls() \
._transfer_from_address(context, address_from, transfer_payload)