Fixes for other tests now that BlobManager relies on DHTHashAnnouncer.get_next_announce_time()

This commit is contained in:
Kay Kurokawa 2017-02-13 13:53:55 -05:00
parent 8e0f71ac74
commit aa45b0e121
2 changed files with 4 additions and 1 deletions

View file

@ -97,6 +97,9 @@ class Announcer(object):
def __init__(self, *args):
pass
def hash_queue_size(self):
return 0
def add_supplier(self, supplier):
pass

View file

@ -34,7 +34,7 @@ class CreateEncryptedFileTest(unittest.TestCase):
def create_file(self, filename):
session = mock.Mock(spec=Session.Session)(None, None)
hash_announcer = mock.Mock(spec=DHTHashAnnouncer.DHTHashAnnouncer)(None, None)
hash_announcer = DHTHashAnnouncer.DHTHashAnnouncer(None, None)
session.blob_manager = BlobManager.TempBlobManager(hash_announcer)
session.db_dir = self.tmp_dir
manager = mock.Mock(spec=EncryptedFileManager.EncryptedFileManager)()