From f5cef14d8d3d595669aa31a07a6d941a9269e2b0 Mon Sep 17 00:00:00 2001 From: Victor Shyba Date: Sun, 12 May 2019 03:53:46 -0300 Subject: [PATCH] test get token on announce --- tests/integration/test_dht.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/integration/test_dht.py b/tests/integration/test_dht.py index 66f69d6a5..32ac27757 100644 --- a/tests/integration/test_dht.py +++ b/tests/integration/test_dht.py @@ -60,6 +60,7 @@ class DHTIntegrationTest(AsyncioTestCase): for network_node in self.nodes[:-1]: network_node.stop() await node.refresh_node(True) + await asyncio.sleep(.3) # let pending events settle self.assertFalse(node.protocol.routing_table.get_peers()) for network_node in self.nodes[:-1]: await network_node.start_listening('127.0.0.1') @@ -77,3 +78,11 @@ class DHTIntegrationTest(AsyncioTestCase): blob_hash = hexlify(constants.generate_id(1337)).decode() peers = await node.announce_blob(blob_hash) self.assertEqual(len(peers), 0) + + async def test_get_token_on_announce(self): + await self.setup_network(2, seed_nodes=2) + node1, node2 = self.nodes + node1.protocol.peer_manager.clear_token(node2.protocol.node_id) + blob_hash = hexlify(constants.generate_id(1337)).decode() + node_ids = await node1.announce_blob(blob_hash) + self.assertIn(node2.protocol.node_id, node_ids)