fixing unit tests

This commit is contained in:
Lex Berezhny 2019-03-24 22:39:38 -04:00
parent d61e1e9d8c
commit e06916f5ae
2 changed files with 2 additions and 2 deletions

View file

@ -5,10 +5,10 @@ import asyncio
import logging import logging
from torba.testcase import AsyncioTestCase from torba.testcase import AsyncioTestCase
from lbrynet.conf import Config from lbrynet.conf import Config
from lbrynet.extras.daemon.storage import SQLiteStorage
from lbrynet.blob.blob_info import BlobInfo from lbrynet.blob.blob_info import BlobInfo
from lbrynet.blob.blob_manager import BlobFileManager from lbrynet.blob.blob_manager import BlobFileManager
from lbrynet.stream.descriptor import StreamDescriptor from lbrynet.stream.descriptor import StreamDescriptor
from lbrynet.extras.daemon.storage import SQLiteStorage
from tests.test_utils import random_lbry_hash from tests.test_utils import random_lbry_hash
log = logging.getLogger() log = logging.getLogger()

View file

@ -76,7 +76,7 @@ def get_mock_wallet(sd_hash, storage, balance=10.0, fee=None):
} }
mock_wallet = mock.Mock(spec=LbryWalletManager) mock_wallet = mock.Mock(spec=LbryWalletManager)
mock_wallet.resolve = mock_resolve mock_wallet.ledger.resolve = mock_resolve
async def get_balance(*_): async def get_balance(*_):
return balance return balance