Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

No need for clock now in test for DHTHashAnnouncer

Browse source
This commit is contained in:
Kay Kurokawa 2017-11-09 14:47:55 -05:00 committed by Jack Robison
parent 75b977dff9
commit 0425c95b68
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
1 changed files with 1 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
lbrynet/tests/unit/core/server/test_DHTHashAnnouncer.py
View file

@ -1,7 +1,6 @@
from twisted.trial import unittest from twisted.trial import unittest
from twisted.internet import defer, task, reactor from twisted.internet import defer, reactor
from lbrynet.core import utils
from lbrynet.tests.util import random_lbry_hash from lbrynet.tests.util import random_lbry_hash
from lbrynet.core.server.DHTHashAnnouncer import DHTHashAnnouncer from lbrynet.core.server.DHTHashAnnouncer import DHTHashAnnouncer
@ -41,9 +40,7 @@ class DHTHashAnnouncerTest(unittest.TestCase):
self.blobs_to_announce = [] self.blobs_to_announce = []
for i in range(0, self.num_blobs): for i in range(0, self.num_blobs):
self.blobs_to_announce.append(random_lbry_hash()) self.blobs_to_announce.append(random_lbry_hash())
self.clock = task.Clock()
self.dht_node = MocDHTNode() self.dht_node = MocDHTNode()
utils.call_later = self.clock.callLater
self.announcer = DHTHashAnnouncer(self.dht_node, peer_port=3333) self.announcer = DHTHashAnnouncer(self.dht_node, peer_port=3333)
self.supplier = MocSupplier(self.blobs_to_announce) self.supplier = MocSupplier(self.blobs_to_announce)
self.announcer.add_supplier(self.supplier) self.announcer.add_supplier(self.supplier)
@ -52,7 +49,6 @@ class DHTHashAnnouncerTest(unittest.TestCase):
def test_basic(self): def test_basic(self):
d = self.announcer._announce_available_hashes() d = self.announcer._announce_available_hashes()
self.assertEqual(self.announcer.hash_queue_size(), self.announcer.CONCURRENT_ANNOUNCERS) self.assertEqual(self.announcer.hash_queue_size(), self.announcer.CONCURRENT_ANNOUNCERS)
self.clock.advance(1)
yield d yield d
self.assertEqual(self.dht_node.blobs_announced, self.num_blobs) self.assertEqual(self.dht_node.blobs_announced, self.num_blobs)
self.assertEqual(self.announcer.hash_queue_size(), 0) self.assertEqual(self.announcer.hash_queue_size(), 0)