forked from LBRYCommunity/lbry-sdk
fix unit tests
This commit is contained in:
parent
7b425eb2ac
commit
758f9deafe
4 changed files with 8 additions and 10 deletions
|
@ -54,7 +54,8 @@ class BlobExchangeTestBase(AsyncioTestCase):
|
||||||
download_dir=self.client_dir,
|
download_dir=self.client_dir,
|
||||||
wallet=self.client_wallet_dir,
|
wallet=self.client_wallet_dir,
|
||||||
save_files=True,
|
save_files=True,
|
||||||
fixed_peers=[]
|
fixed_peers=[],
|
||||||
|
tracker_servers=[]
|
||||||
)
|
)
|
||||||
self.client_config.transaction_cache_size = 10000
|
self.client_config.transaction_cache_size = 10000
|
||||||
self.client_storage = SQLiteStorage(self.client_config, os.path.join(self.client_dir, "lbrynet.sqlite"))
|
self.client_storage = SQLiteStorage(self.client_config, os.path.join(self.client_dir, "lbrynet.sqlite"))
|
||||||
|
|
|
@ -34,6 +34,7 @@ class TestComponentManager(AsyncioTestCase):
|
||||||
],
|
],
|
||||||
[
|
[
|
||||||
components.BackgroundDownloaderComponent,
|
components.BackgroundDownloaderComponent,
|
||||||
|
components.TrackerAnnouncerComponent
|
||||||
]
|
]
|
||||||
]
|
]
|
||||||
self.component_manager = ComponentManager(Config())
|
self.component_manager = ComponentManager(Config())
|
||||||
|
@ -150,6 +151,9 @@ class FakeDelayedFileManager(FakeComponent):
|
||||||
async def start(self):
|
async def start(self):
|
||||||
await asyncio.sleep(1)
|
await asyncio.sleep(1)
|
||||||
|
|
||||||
|
def get_filtered(self):
|
||||||
|
return []
|
||||||
|
|
||||||
|
|
||||||
class TestComponentManagerProperStart(AdvanceTimeTestCase):
|
class TestComponentManagerProperStart(AdvanceTimeTestCase):
|
||||||
|
|
||||||
|
|
|
@ -11,7 +11,7 @@ from lbry.extras.daemon.components import (
|
||||||
DATABASE_COMPONENT, DISK_SPACE_COMPONENT, BLOB_COMPONENT, WALLET_COMPONENT, DHT_COMPONENT,
|
DATABASE_COMPONENT, DISK_SPACE_COMPONENT, BLOB_COMPONENT, WALLET_COMPONENT, DHT_COMPONENT,
|
||||||
HASH_ANNOUNCER_COMPONENT, FILE_MANAGER_COMPONENT, PEER_PROTOCOL_SERVER_COMPONENT,
|
HASH_ANNOUNCER_COMPONENT, FILE_MANAGER_COMPONENT, PEER_PROTOCOL_SERVER_COMPONENT,
|
||||||
UPNP_COMPONENT, EXCHANGE_RATE_MANAGER_COMPONENT, WALLET_SERVER_PAYMENTS_COMPONENT,
|
UPNP_COMPONENT, EXCHANGE_RATE_MANAGER_COMPONENT, WALLET_SERVER_PAYMENTS_COMPONENT,
|
||||||
LIBTORRENT_COMPONENT, BACKGROUND_DOWNLOADER_COMPONENT
|
LIBTORRENT_COMPONENT, BACKGROUND_DOWNLOADER_COMPONENT, TRACKER_ANNOUNCER_COMPONENT
|
||||||
)
|
)
|
||||||
from lbry.extras.daemon.daemon import Daemon
|
from lbry.extras.daemon.daemon import Daemon
|
||||||
|
|
||||||
|
@ -72,7 +72,7 @@ class TestAccessHeaders(AsyncioTestCase):
|
||||||
DATABASE_COMPONENT, DISK_SPACE_COMPONENT, BLOB_COMPONENT, WALLET_COMPONENT, DHT_COMPONENT,
|
DATABASE_COMPONENT, DISK_SPACE_COMPONENT, BLOB_COMPONENT, WALLET_COMPONENT, DHT_COMPONENT,
|
||||||
HASH_ANNOUNCER_COMPONENT, FILE_MANAGER_COMPONENT, PEER_PROTOCOL_SERVER_COMPONENT,
|
HASH_ANNOUNCER_COMPONENT, FILE_MANAGER_COMPONENT, PEER_PROTOCOL_SERVER_COMPONENT,
|
||||||
UPNP_COMPONENT, EXCHANGE_RATE_MANAGER_COMPONENT, WALLET_SERVER_PAYMENTS_COMPONENT,
|
UPNP_COMPONENT, EXCHANGE_RATE_MANAGER_COMPONENT, WALLET_SERVER_PAYMENTS_COMPONENT,
|
||||||
LIBTORRENT_COMPONENT, BACKGROUND_DOWNLOADER_COMPONENT
|
LIBTORRENT_COMPONENT, BACKGROUND_DOWNLOADER_COMPONENT, TRACKER_ANNOUNCER_COMPONENT
|
||||||
)
|
)
|
||||||
Daemon.component_attributes = {}
|
Daemon.component_attributes = {}
|
||||||
self.daemon = Daemon(conf)
|
self.daemon = Daemon(conf)
|
||||||
|
|
|
@ -13,13 +13,6 @@ from lbry.stream.descriptor import StreamDescriptor
|
||||||
from tests.unit.blob_exchange.test_transfer_blob import BlobExchangeTestBase
|
from tests.unit.blob_exchange.test_transfer_blob import BlobExchangeTestBase
|
||||||
|
|
||||||
|
|
||||||
def get_mock_node(loop):
|
|
||||||
mock_node = mock.Mock(spec=Node)
|
|
||||||
mock_node.joined = asyncio.Event(loop=loop)
|
|
||||||
mock_node.joined.set()
|
|
||||||
return mock_node
|
|
||||||
|
|
||||||
|
|
||||||
class TestManagedStream(BlobExchangeTestBase):
|
class TestManagedStream(BlobExchangeTestBase):
|
||||||
async def create_stream(self, blob_count: int = 10, file_name='test_file'):
|
async def create_stream(self, blob_count: int = 10, file_name='test_file'):
|
||||||
self.stream_bytes = b''
|
self.stream_bytes = b''
|
||||||
|
|
Loading…
Reference in a new issue