diff --git a/tests/mocks.py b/tests/mocks.py index 96845ad43..ca321b464 100644 --- a/tests/mocks.py +++ b/tests/mocks.py @@ -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 diff --git a/tests/unit/lbryfilemanager/test_EncryptedFileCreator.py b/tests/unit/lbryfilemanager/test_EncryptedFileCreator.py index 60d1d4f39..704274444 100644 --- a/tests/unit/lbryfilemanager/test_EncryptedFileCreator.py +++ b/tests/unit/lbryfilemanager/test_EncryptedFileCreator.py @@ -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)()