def test_monitor_downloads_recreate_ip(self):
"""
Test whether an old introduction point is recreated
"""
tribler_session = MockObject()
tribler_session.notifier = MockObject()
tribler_session.notifier.notify = lambda *_: None
tribler_session.lm = MockObject()
tribler_session.lm.ltmgr = MockObject()
tribler_session.lm.ltmgr.get_libtorrent_version = lambda: "1.1.0.0"
self.nodes[0].overlay.tribler_session = tribler_session
mock_state = MockObject()
mock_download = MockObject()
mock_tdef = MockObject()
mock_tdef.get_infohash = lambda: 'a'
mock_download.get_hops = lambda: 1
mock_download.get_def = lambda: mock_tdef
mock_download.add_peer = lambda x: None
mock_download.get_state = lambda: mock_state
mock_state.get_status = lambda: 1
mock_state.get_download = lambda: mock_download
tribler_session.get_downloads = lambda: [mock_download, ]
real_ih = self.nodes[0].overlay.get_lookup_info_hash('a')
self.nodes[0].overlay.infohash_ip_circuits[real_ih] = [(3, 0)]
self.nodes[0].overlay.download_states['a'] = 3
self.nodes[0].overlay.monitor_downloads([mock_state])
self.assertNotEqual(self.nodes[0].overlay.infohash_ip_circuits[real_ih][0][1], 0)
def test_readd_bittorrent_peers(self):
"""
Test the readd bittorrent peers method
"""
mock_torrent = MockObject()
mock_torrent.add_peer = lambda _: None
mock_torrent.tdef = MockObject()
mock_torrent.tdef.get_infohash = lambda: 'a' * 20
self.nodes[0].overlay.bittorrent_peers = {mock_torrent: [None]}
self.nodes[0].overlay.readd_bittorrent_peers()
self.assertNotIn(mock_torrent, self.nodes[0].overlay.bittorrent_peers)
