From aa45b0e121b238a3d31b9a8d1137bb20ef43cbbd Mon Sep 17 00:00:00 2001 From: Kay Kurokawa Date: Mon, 13 Feb 2017 13:53:55 -0500 Subject: [PATCH] Fixes for other tests now that BlobManager relies on DHTHashAnnouncer.get_next_announce_time() --- tests/mocks.py | 3 +++ tests/unit/lbryfilemanager/test_EncryptedFileCreator.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) 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)()