def test_mediator_task_view():
"""Same as above for mediator tasks."""
secret1 = factories.make_secret(1)
locked_amount1 = 11
payee_transfer = factories.create(
factories.LockedTransferUnsignedStateProperties(secret=secret1))
payer_transfer = factories.create(
factories.LockedTransferSignedStateProperties(
secret=secret1, payment_identifier=1,
locked_amount=locked_amount1))
secrethash1 = payee_transfer.lock.secrethash
initiator = payee_transfer.initiator
initiator_channel = factories.create(
factories.NettingChannelStateProperties(
partner_state=factories.NettingChannelEndStateProperties(
address=initiator, balance=100)))
routes = [factories.make_route_from_channel(initiator_channel)]
transfer_state1 = MediatorTransferState(secrethash=secrethash1,
routes=routes)
transfer_state1.transfers_pair.append(
MediationPairState(
payer_transfer=payer_transfer,
payee_transfer=payee_transfer,
payee_address=payee_transfer.target,
))
task1 = MediatorTask(
token_network_identifier=factories.UNIT_TOKEN_NETWORK_ADDRESS,
mediator_state=transfer_state1,
)
secret2 = factories.make_secret(2)
locked_amount2 = 13
transfer2 = factories.create(
factories.LockedTransferSignedStateProperties(
secret=secret2, payment_identifier=2,
locked_amount=locked_amount2))
secrethash2 = transfer2.lock.secrethash
transfer_state2 = MediatorTransferState(secrethash=secrethash2,
routes=routes)
transfer_state2.waiting_transfer = WaitingTransferState(transfer=transfer2)
task2 = MediatorTask(
token_network_identifier=factories.UNIT_TOKEN_NETWORK_ADDRESS,
mediator_state=transfer_state2,
)
payment_mapping = {secrethash1: task1, secrethash2: task2}
view = transfer_tasks_view(payment_mapping)
assert len(view) == 2
if view[0].get("payment_identifier") == "1":
pending_transfer, waiting_transfer = view
else:
waiting_transfer, pending_transfer = view
assert pending_transfer.get("role") == waiting_transfer.get(
"role") == "mediator"
assert pending_transfer.get("payment_identifier") == "1"
assert waiting_transfer.get("payment_identifier") == "2"
assert pending_transfer.get("locked_amount") == str(locked_amount1)
assert waiting_transfer.get("locked_amount") == str(locked_amount2)
def test_mediator_task_view():
"""Same as above for mediator tasks."""
secret1 = factories.make_secret(1)
locked_amount1 = TokenAmount(11)
payee_transfer = factories.create(
factories.LockedTransferUnsignedStateProperties(secret=secret1))
payer_transfer = factories.create(
factories.LockedTransferSignedStateProperties(
secret=secret1,
payment_identifier=PaymentID(1),
locked_amount=locked_amount1))
secrethash1 = payee_transfer.lock.secrethash
route_state = RouteState(route=[payee_transfer.target])
transfer_state1 = MediatorTransferState(secrethash=secrethash1,
routes=[route_state])
# pylint: disable=E1101
transfer_state1.transfers_pair.append(
MediationPairState(
payer_transfer=payer_transfer,
payee_transfer=payee_transfer,
payee_address=payee_transfer.target,
))
task1 = MediatorTask(
token_network_address=factories.UNIT_TOKEN_NETWORK_ADDRESS,
mediator_state=transfer_state1)
secret2 = factories.make_secret(2)
locked_amount2 = TokenAmount(13)
transfer2 = factories.create(
factories.LockedTransferSignedStateProperties(
secret=secret2,
payment_identifier=PaymentID(2),
locked_amount=locked_amount2))
secrethash2 = transfer2.lock.secrethash
transfer_state2 = MediatorTransferState(secrethash=secrethash2,
routes=[route_state])
transfer_state2.waiting_transfer = WaitingTransferState(transfer=transfer2)
task2 = MediatorTask(
token_network_address=factories.UNIT_TOKEN_NETWORK_ADDRESS,
mediator_state=transfer_state2)
payment_mapping = {
secrethash1: cast(TransferTask, task1),
secrethash2: cast(TransferTask, task2),
}
view = transfer_tasks_view(payment_mapping)
assert len(view) == 2
if view[0].get("payment_identifier") == "1":
pending_transfer, waiting_transfer = view
else:
waiting_transfer, pending_transfer = view
assert pending_transfer.get("role") == waiting_transfer.get(
"role") == "mediator"
assert pending_transfer.get("payment_identifier") == "1"
assert waiting_transfer.get("payment_identifier") == "2"
assert pending_transfer.get("locked_amount") == str(locked_amount1)
assert waiting_transfer.get("locked_amount") == str(locked_amount2)
def test_invalid_instantiation_locked_transfer_state():
valid_unsigned = factories.create(
factories.LockedTransferUnsignedStateProperties())
valid_signed = factories.create(
factories.LockedTransferSignedStateProperties())
# neither class can be instantiated with an empty locksroot
for valid in (valid_unsigned, valid_signed):
empty_balance_proof = replace(valid.balance_proof,
locksroot=LOCKSROOT_OF_NO_LOCKS)
with pytest.raises(ValueError):
replace(valid, balance_proof=empty_balance_proof)
# an unsigned locked transfer state cannot be instantiated with a signed balance proof
with pytest.raises(ValueError):
replace(valid_unsigned, balance_proof=valid_signed.balance_proof)
# and vice versa
with pytest.raises(ValueError):
replace(valid_signed, balance_proof=valid_unsigned.balance_proof)
# lock is typechecked
wrong_type_lock = object()
for valid in (valid_unsigned, valid_signed):
with pytest.raises(ValueError):
replace(valid, lock=wrong_type_lock)
def make_transfer_from_counter(counter):
return factories.create(
factories.LockedTransferUnsignedStateProperties(
amount=next(counter),
initiator=factories.make_initiator_address(),
target=factories.make_target_address(),
expiration=next(counter),
secret=factories.make_secret(next(counter)),
))
def test_invalid_instantiation_mediation_pair_state():
valid = MediationPairState(
payer_transfer=factories.create(factories.LockedTransferSignedStateProperties()),
payee_address=factories.make_address(),
payee_transfer=factories.create(factories.LockedTransferUnsignedStateProperties()),
)
unsigned_transfer = factories.create(factories.LockedTransferUnsignedStateProperties())
with pytest.raises(ValueError):
replace(valid, payer_transfer=unsigned_transfer)
signed_transfer = factories.create(factories.LockedTransferSignedStateProperties())
with pytest.raises(ValueError):
replace(valid, payee_transfer=signed_transfer)
hex_instead_of_binary = factories.make_checksum_address()
with pytest.raises(ValueError):
replace(valid, payee_address=hex_instead_of_binary)
def test_get_transfer_secret_none_for_none_transfer_state(chain_state):
secret = factories.make_secret()
transfer = factories.create(factories.LockedTransferUnsignedStateProperties(secret=secret))
secrethash = transfer.lock.secrethash
payment_state = InitiatorPaymentState(initiator_transfers={secrethash: None}, routes=[])
task = InitiatorTask(
token_network_address=factories.UNIT_TOKEN_NETWORK_ADDRESS, manager_state=payment_state
)
chain_state.payment_mapping.secrethashes_to_task[secrethash] = task
assert get_transfer_secret(chain_state=chain_state, secrethash=secrethash) is None
def test_send_refund_transfer_contains_balance_proof():
recipient = factories.make_address()
transfer = factories.create(factories.LockedTransferUnsignedStateProperties())
message_identifier = 1
event = SendRefundTransfer(
recipient=recipient,
message_identifier=message_identifier,
transfer=transfer,
canonical_identifier=factories.make_canonical_identifier(),
)
assert hasattr(event, "balance_proof")
assert JSONSerializer.deserialize(JSONSerializer.serialize(event)) == event
def test_send_refund_transfer_contains_balance_proof():
recipient = factories.make_address()
transfer = factories.create(factories.LockedTransferUnsignedStateProperties())
message_identifier = 1
channel_identifier = factories.make_channel_identifier()
event = SendRefundTransfer(
recipient=recipient,
channel_identifier=channel_identifier,
message_identifier=message_identifier,
transfer=transfer,
)
assert hasattr(event, "balance_proof")
assert SendRefundTransfer.from_dict(event.to_dict()) == event
def test_initiator_task_view():
"""Test transfer_tasks_view(), which is used to generate the output of the
pending transfers API, with an initiator task.
"""
channel_id = factories.UNIT_CHANNEL_ID
secret = factories.make_secret()
transfer = factories.create(
factories.LockedTransferUnsignedStateProperties(secret=secret))
secrethash = transfer.lock.secrethash
transfer_description = TransferDescriptionWithSecretState(
token_network_registry_address=factories.
UNIT_TOKEN_NETWORK_REGISTRY_ADDRESS,
payment_identifier=transfer.payment_identifier,
amount=transfer.balance_proof.locked_amount,
token_network_address=factories.UNIT_TOKEN_NETWORK_ADDRESS,
initiator=transfer.initiator,
target=transfer.target,
secret=secret,
secrethash=sha256_secrethash(secret),
)
transfer_state = InitiatorTransferState(
route=RouteState(route=[transfer.initiator, transfer.target],
forward_channel_id=channel_id),
transfer_description=transfer_description,
channel_identifier=channel_id,
transfer=transfer,
)
payment_state = InitiatorPaymentState(
routes=[], initiator_transfers={secrethash: transfer_state})
task = InitiatorTask(
token_network_address=factories.UNIT_TOKEN_NETWORK_ADDRESS,
manager_state=payment_state)
payment_mapping = {secrethash: cast(TransferTask, task)}
view = transfer_tasks_view(payment_mapping)
assert len(view) == 1
pending_transfer = view[0]
assert pending_transfer.get("role") == "initiator"
balance_proof = transfer.balance_proof
assert pending_transfer.get("channel_identifier") == str(
balance_proof.channel_identifier)
assert pending_transfer.get("locked_amount") == str(
balance_proof.locked_amount)
assert pending_transfer.get("transferred_amount") == str(
balance_proof.transferred_amount)
def test_initiator_task_view():
"""Test transfer_tasks_view(), which is used to generate the output of the
pending transfers API, with an initiator task.
"""
channel_id = factories.UNIT_CHANNEL_ID
secret = factories.make_secret()
payment_hash_invoice = factories.make_payment_hash_invoice()
transfer = factories.create(
factories.LockedTransferUnsignedStateProperties(
secret=secret, payment_hash_invoice=payment_hash_invoice))
secrethash = transfer.lock.secrethash
transfer_description = TransferDescriptionWithSecretState(
payment_network_identifier=factories.UNIT_PAYMENT_NETWORK_IDENTIFIER,
payment_identifier=transfer.payment_identifier,
payment_hash_invoice=transfer.payment_hash_invoice,
amount=transfer.balance_proof.locked_amount,
allocated_fee=0,
token_network_identifier=factories.UNIT_TOKEN_NETWORK_ADDRESS,
initiator=transfer.initiator,
target=transfer.target,
secret=secret,
)
transfer_state = InitiatorTransferState(
transfer_description=transfer_description,
channel_identifier=channel_id,
transfer=transfer,
revealsecret=None,
)
payment_state = InitiatorPaymentState({secrethash: transfer_state})
task = InitiatorTask(
token_network_identifier=factories.UNIT_TOKEN_NETWORK_ADDRESS,
manager_state=payment_state)
payment_mapping = {secrethash: task}
view = transfer_tasks_view(payment_mapping)
assert len(view) == 1
pending_transfer = view[0]
assert pending_transfer.get("role") == "initiator"
balance_proof = transfer.balance_proof
assert pending_transfer.get("channel_identifier") == str(
balance_proof.channel_identifier)
assert pending_transfer.get("locked_amount") == str(
balance_proof.locked_amount)
assert pending_transfer.get("transferred_amount") == str(
balance_proof.transferred_amount)
