def test_submit_earn_batch_rejected(self, grpc_channel, executor, app_index_client):
sender = PrivateKey.random()
earns = [
Earn(PrivateKey.random().public_key, 100000),
Earn(PrivateKey.random().public_key, 100000),
]
future = executor.submit(app_index_client.submit_earn_batch, sender, earns)
account_req = self._set_successful_get_account_info_response(grpc_channel, sender, 10)
resp = tx_pb.SubmitTransactionResponse(
result=tx_pb.SubmitTransactionResponse.Result.REJECTED,
hash=model_pb2.TransactionHash(value=b'somehash'),
)
submit_req = self._set_submit_transaction_response(grpc_channel, resp)
batch_earn_result = future.result()
assert len(batch_earn_result.succeeded) == 0
assert len(batch_earn_result.failed) == 2
for idx, earn_result in enumerate(batch_earn_result.failed):
assert earn_result.earn == earns[idx]
assert not earn_result.tx_hash
assert isinstance(earn_result.error, TransactionRejectedError)
assert account_req.account_id.value == sender.public_key.stellar_address
expected_memo = memo.HashMemo(AgoraMemo.new(1, TransactionType.EARN, 1, b'').val)
self._assert_earn_batch_envelope(submit_req.envelope_xdr, [sender], sender, 100, 11, expected_memo, sender,
earns)
assert len(submit_req.invoice_list.invoices) == 0
def test_from_proto_text_memo(self):
op_result = gen_payment_op_result(xdr_const.PAYMENT_UNDERFUNDED)
result_xdr = gen_result_xdr(xdr_const.txFAILED, [op_result])
tx_src = gen_account_id()
dest = gen_account_id()
operations = [gen_payment_op(dest, amount=20)]
envelope_xdr = gen_tx_envelope_xdr(tx_src, 1, operations,
gen_text_memo(b'somememo'))
history_item = tx_pb.HistoryItem(
hash=model_pb2.TransactionHash(value=b'somehash'),
result_xdr=result_xdr,
envelope_xdr=envelope_xdr,
cursor=tx_pb.Cursor(value=b'cursor1'),
)
data = TransactionData.from_proto(history_item)
assert data.tx_hash == b'somehash'
assert len(data.payments) == 1
payment = data.payments[0]
assert payment.sender.raw == tx_src.ed25519
assert payment.destination.raw == dest.ed25519
assert payment.tx_type == TransactionType.UNKNOWN
assert payment.quarks == 20
assert not payment.invoice
assert payment.memo == 'somememo'
def test_submit_payment_with_nonce_retry(self, grpc_channel, executor, nonce_retry_client):
sender = PrivateKey.random()
dest = PrivateKey.random().public_key
payment = Payment(sender, dest, TransactionType.EARN, 100000)
future = executor.submit(nonce_retry_client.submit_payment, payment)
result_xdr = gen_result_xdr(xdr_const.txBAD_SEQ, [])
resp = tx_pb.SubmitTransactionResponse(
result=tx_pb.SubmitTransactionResponse.Result.FAILED,
hash=model_pb2.TransactionHash(value=b'somehash'),
ledger=10,
result_xdr=result_xdr,
)
account_reqs = []
submit_reqs = []
for i in range(_config_with_nonce_retry.max_nonce_refreshes + 1):
# this blocks until the system under test invokes the RPC, so if the test completes then the RPC was called
# the expected number of times.
account_reqs.append(self._set_successful_get_account_info_response(grpc_channel, sender, 10))
submit_reqs.append(self._set_submit_transaction_response(grpc_channel, resp))
with pytest.raises(BadNonceError):
future.result()
for account_req in account_reqs:
assert account_req.account_id.value == sender.public_key.stellar_address
expected_memo = memo.HashMemo(AgoraMemo.new(1, TransactionType.EARN, 1, b'').val)
for submit_req in submit_reqs:
self._assert_payment_envelope(submit_req.envelope_xdr, [sender], sender, 100, 11, expected_memo, payment)
assert len(submit_req.invoice_list.invoices) == 0
def test_submit_payment_tx_failed(self, grpc_channel, executor, app_index_client):
sender = PrivateKey.random()
dest = PrivateKey.random().public_key
payment = Payment(sender, dest, TransactionType.EARN, 100000)
future = executor.submit(app_index_client.submit_payment, payment)
account_req = self._set_successful_get_account_info_response(grpc_channel, sender, 10)
result_xdr = gen_result_xdr(xdr_const.txFAILED, [gen_payment_op_result(xdr_const.PAYMENT_UNDERFUNDED)])
resp = tx_pb.SubmitTransactionResponse(
result=tx_pb.SubmitTransactionResponse.Result.FAILED,
hash=model_pb2.TransactionHash(value=b'somehash'),
ledger=10,
result_xdr=result_xdr,
)
submit_req = self._set_submit_transaction_response(grpc_channel, resp)
with pytest.raises(InsufficientBalanceError):
future.result()
assert account_req.account_id.value == sender.public_key.stellar_address
expected_memo = memo.HashMemo(AgoraMemo.new(1, TransactionType.EARN, 1, b'').val)
self._assert_payment_envelope(submit_req.envelope_xdr, [sender], sender, 100, 11, expected_memo, payment)
assert len(submit_req.invoice_list.invoices) == 0
def _set_successful_submit_transaction_response(channel: grpc_testing.Channel, tx_hash: bytes, result_xdr: bytes):
resp = tx_pb.SubmitTransactionResponse(
result=tx_pb.SubmitTransactionResponse.Result.OK,
hash=model_pb2.TransactionHash(value=tx_hash),
ledger=10,
result_xdr=result_xdr,
)
return TestAgoraClient._set_submit_transaction_response(channel, resp)
def test_get_transaction(self, grpc_channel, executor, app_index_client):
tx_hash = b'somehash'
future = executor.submit(app_index_client.get_transaction, tx_hash)
_, request, rpc = grpc_channel.take_unary_unary(
tx_pb.DESCRIPTOR.services_by_name['Transaction'].methods_by_name['GetTransaction']
)
# Create full response
op_result = gen_payment_op_result(xdr_const.PAYMENT_SUCCESS)
result_xdr = gen_result_xdr(xdr_const.txSUCCESS, [op_result, op_result])
il = model_pb2.InvoiceList(invoices=[
model_pb2.Invoice(
items=[
model_pb2.Invoice.LineItem(title='t1', amount=15),
]
),
])
fk = InvoiceList.from_proto(il).get_sha_224_hash()
memo = AgoraMemo.new(1, TransactionType.EARN, 1, fk)
hash_memo = gen_hash_memo(memo.val)
acc1 = gen_account_id()
acc2 = gen_account_id()
operations = [gen_payment_op(acc2, amount=15)]
envelope_xdr = gen_tx_envelope_xdr(acc1, 1, operations, hash_memo)
history_item = tx_pb.HistoryItem(
hash=model_pb2.TransactionHash(value=tx_hash),
result_xdr=result_xdr,
envelope_xdr=envelope_xdr,
cursor=tx_pb.Cursor(value=b'cursor1'),
invoice_list=il,
)
resp = tx_pb.GetTransactionResponse(
state=tx_pb.GetTransactionResponse.State.SUCCESS,
ledger=10,
item=history_item,
)
rpc.terminate(resp, (), grpc.StatusCode.OK, '')
tx_data = future.result()
assert tx_data.tx_hash == tx_hash
assert len(tx_data.payments) == 1
assert not tx_data.error
payment1 = tx_data.payments[0]
assert payment1.sender.raw == acc1.ed25519
assert payment1.destination.raw == acc2.ed25519
assert payment1.tx_type == memo.tx_type()
assert payment1.quarks == 15
assert (payment1.invoice.to_proto().SerializeToString() == il.invoices[0].SerializeToString())
assert not payment1.memo
assert request.transaction_hash.value == tx_hash
def test_from_proto_agora_memo(self):
op_result = gen_payment_op_result(xdr_const.PAYMENT_SUCCESS)
result_xdr = gen_result_xdr(xdr_const.txSUCCESS,
[op_result, op_result])
il = model_pb2.InvoiceList(invoices=[
model_pb2.Invoice(items=[
model_pb2.Invoice.LineItem(title='t1', amount=10),
]),
model_pb2.Invoice(items=[
model_pb2.Invoice.LineItem(title='t1', amount=15),
]),
])
fk = InvoiceList.from_proto(il).get_sha_224_hash()
memo = AgoraMemo.new(1, TransactionType.P2P, 0, fk)
hash_memo = gen_hash_memo(memo.val)
acc1 = gen_account_id()
acc2 = gen_account_id()
acc3 = gen_account_id()
operations = [
gen_payment_op(acc2, src=acc1, amount=10),
gen_payment_op(acc1, src=acc2, amount=15),
]
envelope_xdr = gen_tx_envelope_xdr(acc3, 1, operations, hash_memo)
history_item = tx_pb.HistoryItem(
hash=model_pb2.TransactionHash(value=b'somehash'),
result_xdr=result_xdr,
envelope_xdr=envelope_xdr,
cursor=tx_pb.Cursor(value=b'cursor1'),
invoice_list=il,
)
data = TransactionData.from_proto(history_item)
assert data.tx_hash == b'somehash'
assert len(data.payments) == 2
payment1 = data.payments[0]
assert payment1.sender.raw == acc1.ed25519
assert payment1.destination.raw == acc2.ed25519
assert payment1.tx_type == memo.tx_type()
assert payment1.quarks == 10
assert (payment1.invoice.to_proto().SerializeToString() ==
il.invoices[0].SerializeToString())
assert not payment1.memo
payment2 = data.payments[1]
assert payment2.sender.raw == acc2.ed25519
assert payment2.destination.raw == acc1.ed25519
assert payment2.tx_type == TransactionType.P2P
assert payment2.quarks == 15
assert (payment2.invoice.to_proto().SerializeToString() ==
il.invoices[1].SerializeToString())
assert not payment2.memo
def get_transaction(self, tx_hash: bytes) -> TransactionData:
resp = self.transaction_stub.GetTransaction(
tx_pb.GetTransactionRequest(
transaction_hash=model_pb2.TransactionHash(value=tx_hash)),
timeout=_GRPC_TIMEOUT_SECONDS)
if resp.state is tx_pb.GetTransactionResponse.State.UNKNOWN:
raise TransactionNotFound()
if resp.state == tx_pb.GetTransactionResponse.State.SUCCESS:
return TransactionData.from_proto(resp.item)
raise Error("Unexpected transaction state from Agora: %d", resp.state)
def test_submit_earn_batch_tx_failed(self, grpc_channel, executor, app_index_client):
sender = PrivateKey.random()
earns = [
Earn(PrivateKey.random().public_key, 100000),
Earn(PrivateKey.random().public_key, 100000),
]
future = executor.submit(app_index_client.submit_earn_batch, sender, earns)
account_req = self._set_successful_get_account_info_response(grpc_channel, sender, 10)
result_xdr = gen_result_xdr(xdr_const.txFAILED, [gen_payment_op_result(xdr_const.PAYMENT_UNDERFUNDED),
gen_payment_op_result(xdr_const.PAYMENT_NO_DESTINATION)])
resp = tx_pb.SubmitTransactionResponse(
result=tx_pb.SubmitTransactionResponse.Result.FAILED,
hash=model_pb2.TransactionHash(value=b'somehash'),
ledger=10,
result_xdr=result_xdr,
)
submit_req = self._set_submit_transaction_response(grpc_channel, resp)
batch_earn_result = future.result()
assert len(batch_earn_result.succeeded) == 0
assert len(batch_earn_result.failed) == 2
expected_errors = [InsufficientBalanceError, DestinationDoesNotExistError]
for idx, earn_result in enumerate(batch_earn_result.failed):
assert earn_result.earn == earns[idx]
assert earn_result.tx_hash # make sure it's set
assert isinstance(earn_result.error, expected_errors[idx])
assert account_req.account_id.value == sender.public_key.stellar_address
expected_memo = memo.HashMemo(AgoraMemo.new(1, TransactionType.EARN, 1, b'').val)
self._assert_earn_batch_envelope(submit_req.envelope_xdr, [sender], sender, 100, 11, expected_memo, sender,
earns)
assert len(submit_req.invoice_list.invoices) == 0
def test_submit_earn_batch_with_retry(self, grpc_channel, executor, retry_client):
sender = PrivateKey.random()
earns = [Earn(PrivateKey.random().public_key, 100000)]
future = executor.submit(retry_client.submit_earn_batch, sender, earns)
account_req = self._set_successful_get_account_info_response(grpc_channel, sender, 10)
resp = tx_pb.SubmitTransactionResponse(
result=5, # invalid result code, should throw an error
hash=model_pb2.TransactionHash(value=b'somehash'),
)
submit_reqs = []
for i in range(_config_with_retry.max_retries + 1):
# this blocks until the system under test invokes the RPC, so if the test completes then the RPC was called
# the expected number of times.
submit_reqs.append(self._set_submit_transaction_response(grpc_channel, resp))
batch_earn_result = future.result()
assert len(batch_earn_result.succeeded) == 0
assert len(batch_earn_result.failed) == 1
earn_result = batch_earn_result.failed[0]
assert not earn_result.tx_hash
assert earn_result.earn == earns[0]
assert isinstance(earn_result.error, Error)
assert account_req.account_id.value == sender.public_key.stellar_address
expected_memo = memo.HashMemo(AgoraMemo.new(1, TransactionType.EARN, 1, b'').val)
for submit_req in submit_reqs:
self._assert_earn_batch_envelope(submit_req.envelope_xdr, [sender], sender, 100, 11, expected_memo,
sender,
earns)
assert len(submit_req.invoice_list.invoices) == 0