From 0badea874d467af5c184c42f4251d2d597cac096 Mon Sep 17 00:00:00 2001 From: Lex Berezhny Date: Sat, 4 Aug 2018 21:08:22 -0400 Subject: [PATCH] test_Downloder tests fixed --- tests/unit/lbrynet_daemon/test_Downloader.py | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/tests/unit/lbrynet_daemon/test_Downloader.py b/tests/unit/lbrynet_daemon/test_Downloader.py index 2c0c6f2af..d834c1c61 100644 --- a/tests/unit/lbrynet_daemon/test_Downloader.py +++ b/tests/unit/lbrynet_daemon/test_Downloader.py @@ -3,22 +3,21 @@ import mock from twisted.trial import unittest from twisted.internet import defer, task -from lbrynet.wallet.manager import LbryWalletManager from lbrynet.core import PaymentRateManager from lbrynet.core.Error import DownloadDataTimeout, DownloadSDTimeout -from lbrynet.daemon import Downloader from lbrynet.core.StreamDescriptor import StreamDescriptorIdentifier -from lbrynet.database.storage import SQLiteStorage from lbrynet.core.BlobManager import DiskBlobManager -from lbrynet.dht.peerfinder import DummyPeerFinder from lbrynet.core.RateLimiter import DummyRateLimiter +from lbrynet.daemon import Downloader +from lbrynet.daemon.ExchangeRateManager import ExchangeRateManager +from lbrynet.database.storage import SQLiteStorage +from lbrynet.dht.peerfinder import DummyPeerFinder from lbrynet.file_manager.EncryptedFileStatusReport import EncryptedFileStatusReport from lbrynet.file_manager.EncryptedFileDownloader import ManagedEncryptedFileDownloader -from lbrynet.daemon.ExchangeRateManager import ExchangeRateManager +from lbrynet.wallet.manager import LbryWalletManager -from lbrynet.tests.mocks import ExchangeRateManager as DummyExchangeRateManager -from lbrynet.tests.mocks import mock_conf_settings +from tests.mocks import mock_conf_settings class MocDownloader(object): @@ -71,7 +70,7 @@ class GetStreamTests(unittest.TestCase): mock_conf_settings(self) sd_identifier = mock.Mock(spec=StreamDescriptorIdentifier) - wallet = mock.Mock(spec=LbryWalletmanager) + wallet = mock.Mock(spec=LbryWalletManager) prm = mock.Mock(spec=PaymentRateManager.NegotiatedPaymentRateManager) exchange_rate_manager = mock.Mock(spec=ExchangeRateManager) storage = mock.Mock(spec=SQLiteStorage)