Git-Mirrors/rnsh
1
0
Fork 0
mirror of https://github.com/markqvist/rnsh.git synced 2025-05-10 18:15:16 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Retry with Packet.resend() instead of Packet.send() for #8

Browse source
This commit is contained in:
Aaron Heise 2023-02-19 06:45:58 -06:00
parent b41bcbbc44
commit 07cf0e3d05
No known key found for this signature in database
GPG key ID: 6BA54088C41DE8BF
3 changed files with 29 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/test_protocol.py
View file

@ -22,6 +22,8 @@ class Receipt:
def __init__(self, state: rnsh.protocol.MessageState, raw: bytes):
self.state = state
self.raw = raw
self.msgid = uuid.uuid4()
self.tries = 1
class MessageOutletTest(rnsh.protocol.MessageOutletBase):
@ -39,6 +41,11 @@ class MessageOutletTest(rnsh.protocol.MessageOutletBase):
self.receipts.append(receipt)
return receipt
def resend(self, receipt: Receipt) -> Receipt:
receipt.tries += 1
receipt.state = rnsh.protocol.MessageState.MSGSTATE_SENT
return receipt
def set_packet_received_callback(self, cb: Callable[[rnsh.protocol.MessageOutletBase, bytes], None]):
self.packet_callback = cb
@ -104,12 +111,13 @@ def test_send_one_retry():
receipt.state = rnsh.protocol.MessageState.MSGSTATE_FAILED
module_logger.info("set failed")
time.sleep(retry_interval)
assert len(outlet.receipts) == 2
receipt = outlet.receipts[1]
assert len(outlet.receipts) == 1
assert receipt == outlet.receipts[0]
assert receipt.state == rnsh.protocol.MessageState.MSGSTATE_SENT
assert receipt.tries == 2
receipt.state = rnsh.protocol.MessageState.MSGSTATE_DELIVERED
time.sleep(retry_interval)
assert len(outlet.receipts) == 2
assert len(outlet.receipts) == 1
assert not message.tracked