async def test_no_balance():
"""Test no balance."""
dispatcher = LedgerApiRequestDispatcher(AsyncState())
mock_api = Mock()
message = LedgerApiMessage(
performative=LedgerApiMessage.Performative.GET_BALANCE,
dialogue_reference=dispatcher.dialogues.
new_self_initiated_dialogue_reference(),
ledger_id=ETHEREUM,
address="test",
)
message.to = dispatcher.dialogues.self_address
message.sender = "test"
dialogue = dispatcher.dialogues.update(message)
assert dialogue is not None
mock_api.get_balance.return_value = None
msg = dispatcher.get_balance(mock_api, message, dialogue)
assert msg.performative == LedgerApiMessage.Performative.ERROR
async def test_attempts_get_transaction_receipt():
"""Test retry and sleep."""
dispatcher = LedgerApiRequestDispatcher(
AsyncState(ConnectionStates.connected))
mock_api = Mock()
message = LedgerApiMessage(
performative=LedgerApiMessage.Performative.GET_TRANSACTION_RECEIPT,
dialogue_reference=dispatcher.dialogues.
new_self_initiated_dialogue_reference(),
transaction_digest=TransactionDigest("asdad", "sdfdsf"),
)
message.to = dispatcher.dialogues.self_address
message.sender = "test"
dialogue = dispatcher.dialogues.update(message)
assert dialogue is not None
mock_api.get_transaction.return_value = None
mock_api.is_transaction_settled.return_value = True
with patch.object(dispatcher, "MAX_ATTEMPTS", 2):
with patch.object(dispatcher, "TIMEOUT", 0.001):
msg = dispatcher.get_transaction_receipt(mock_api, message,
dialogue)
assert msg.performative == LedgerApiMessage.Performative.ERROR