def test_delivered_processed_message_cleanup():
recipient = factories.make_address()
channel_identifier = 1
secret = factories.random_secret()
first_message = events.SendSecretReveal(recipient, channel_identifier,
random.randint(0, 2**16), secret)
second_message = events.SendSecretReveal(recipient, channel_identifier,
random.randint(0, 2**16), secret)
message_queue = [first_message, second_message]
fake_message_identifier = random.randint(0, 2**16)
node.inplace_delete_message(
message_queue,
state_change.ReceiveDelivered(recipient, fake_message_identifier))
assert first_message in message_queue, "invalid message id must be ignored"
assert second_message in message_queue, "invalid message id must be ignored"
invalid_sender_address = factories.make_address()
node.inplace_delete_message(
message_queue,
state_change.ReceiveDelivered(invalid_sender_address,
first_message.message_identifier),
)
assert first_message in message_queue, "invalid sender id must be ignored"
assert second_message in message_queue, "invalid sender id must be ignored"
node.inplace_delete_message(
message_queue,
state_change.ReceiveProcessed(recipient,
first_message.message_identifier))
msg = "message must be cleared when a valid delivered is received"
assert first_message not in message_queue, msg
assert second_message in message_queue, msg
def test_delivered_message_must_clean_unordered_messages(chain_id):
pseudo_random_generator = random.Random()
block_number = 10
our_address = factories.make_address()
recipient = factories.make_address()
channel_identifier = 1
message_identifier = random.randint(0, 2**16)
secret = factories.random_secret()
chain_state = state.ChainState(
pseudo_random_generator=pseudo_random_generator,
block_number=block_number,
block_hash=factories.make_block_hash(),
our_address=our_address,
chain_id=chain_id,
)
queue_identifier = QueueIdentifier(
recipient,
events.CHANNEL_IDENTIFIER_GLOBAL_QUEUE,
)
# Regression test:
# The code delivered_message handler worked only with a queue of one
# element
first_message = events.SendSecretReveal(
recipient,
channel_identifier,
message_identifier,
secret,
)
second_message = events.SendSecretReveal(
recipient,
channel_identifier,
random.randint(0, 2**16),
secret,
)
chain_state.queueids_to_queues[queue_identifier] = [
first_message, second_message
]
delivered_message = state_change.ReceiveDelivered(recipient,
message_identifier)
iteration = node.handle_delivered(chain_state, delivered_message)
new_queue = iteration.new_state.queueids_to_queues.get(
queue_identifier, [])
assert first_message not in new_queue