From 60ee53071db42d4d9dbc1d20c432d4b371fede4a Mon Sep 17 00:00:00 2001 From: Konrad Feldmeier Date: Wed, 18 Jan 2017 12:47:12 +0100 Subject: [PATCH] Return `poll`ed cmd_ids in test #50 --- devp2p/tests/test_discovery.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/devp2p/tests/test_discovery.py b/devp2p/tests/test_discovery.py index eb9beb2..67c45e6 100644 --- a/devp2p/tests/test_discovery.py +++ b/devp2p/tests/test_discovery.py @@ -82,11 +82,16 @@ def receive(self, address, message): self.protocol.receive(address, message) def poll(self): + """try to receive a message + Returns: + cmd_id: the command id of the received message + """ # try to receive a message for i, (to_address, from_address, message) in enumerate(self.messages): if to_address == self.address: del self.messages[i] self.receive(from_address, message) + return self.protocol.unpack(message)[1] def test_packing():