From e06916f5ae22c99cd2c8ecd5ef8efdc10ac35505 Mon Sep 17 00:00:00 2001 From: Lex Berezhny Date: Sun, 24 Mar 2019 22:39:38 -0400 Subject: [PATCH] fixing unit tests --- tests/unit/database/test_SQLiteStorage.py | 2 +- tests/unit/stream/test_stream_manager.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/database/test_SQLiteStorage.py b/tests/unit/database/test_SQLiteStorage.py index abd7882ae..71cea752c 100644 --- a/tests/unit/database/test_SQLiteStorage.py +++ b/tests/unit/database/test_SQLiteStorage.py @@ -5,10 +5,10 @@ import asyncio import logging from torba.testcase import AsyncioTestCase from lbrynet.conf import Config +from lbrynet.extras.daemon.storage import SQLiteStorage from lbrynet.blob.blob_info import BlobInfo from lbrynet.blob.blob_manager import BlobFileManager from lbrynet.stream.descriptor import StreamDescriptor -from lbrynet.extras.daemon.storage import SQLiteStorage from tests.test_utils import random_lbry_hash log = logging.getLogger() diff --git a/tests/unit/stream/test_stream_manager.py b/tests/unit/stream/test_stream_manager.py index 0011afe71..4de829cc1 100644 --- a/tests/unit/stream/test_stream_manager.py +++ b/tests/unit/stream/test_stream_manager.py @@ -76,7 +76,7 @@ def get_mock_wallet(sd_hash, storage, balance=10.0, fee=None): } mock_wallet = mock.Mock(spec=LbryWalletManager) - mock_wallet.resolve = mock_resolve + mock_wallet.ledger.resolve = mock_resolve async def get_balance(*_): return balance