Collect messages in WireMock.process ethereum#50

devp2p/tests/test_kademlia_protocol.py

@@ -60,16 +60,20 @@ def process(self, kademlia_protocols, steps=0):
         """
         process messages until none are left
         or if process steps messages if steps >0
+
+        also yields all messages for asserting
         """
         i = 0
         proto_by_node = dict((p.this_node, p) for p in kademlia_protocols)
         while self.messages:
             msg = self.messages.pop(0)
+            print(msg)
             assert isinstance(msg[2], kademlia.Node)
             target = proto_by_node[msg[0]]
             cmd = 'recv_' + msg[1]
             getattr(target, cmd)(*msg[2:])
             i += 1
+            yield msg[1:]
             if steps and i == steps:
                 return  # messages may be left
         assert not self.messages