fixing unit tests
This commit is contained in:
parent
d61e1e9d8c
commit
e06916f5ae
2 changed files with 2 additions and 2 deletions
|
@ -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()
|
||||||
|
|
|
@ -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
|
||||||
|
|
Loading…
Reference in a new issue