lbry-sdk/lbry/tests/unit/components/test_component_manager.py

201 lines
7.2 KiB
Python
Raw Normal View History

import asyncio
2019-01-24 02:57:56 +01:00
from torba.testcase import AsyncioTestCase, AdvanceTimeTestCase
from lbry.conf import Config
from lbry.extras.daemon.ComponentManager import ComponentManager
from lbry.extras.daemon.Components import DATABASE_COMPONENT, DHT_COMPONENT
from lbry.extras.daemon.Components import HASH_ANNOUNCER_COMPONENT, UPNP_COMPONENT
from lbry.extras.daemon.Components import PEER_PROTOCOL_SERVER_COMPONENT, EXCHANGE_RATE_MANAGER_COMPONENT
from lbry.extras.daemon.Components import HEADERS_COMPONENT
from lbry.extras.daemon import Components
2019-01-24 02:57:56 +01:00
class TestComponentManager(AsyncioTestCase):
2019-01-24 02:26:29 +01:00
def setUp(self):
self.default_components_sort = [
2018-08-02 21:54:43 +02:00
[
Components.HeadersComponent,
Components.DatabaseComponent,
Components.ExchangeRateManagerComponent,
Components.UPnPComponent
],
[
Components.BlobComponent,
2018-08-02 21:54:43 +02:00
Components.DHTComponent,
Components.WalletComponent
],
[
Components.HashAnnouncerComponent,
2019-01-24 02:26:29 +01:00
Components.PeerProtocolServerComponent,
Components.StreamManagerComponent,
2018-08-02 21:54:43 +02:00
]
]
2019-01-21 21:55:50 +01:00
self.component_manager = ComponentManager(Config())
def test_sort_components(self):
stages = self.component_manager.sort_components()
for stage_list, sorted_stage_list in zip(stages, self.default_components_sort):
self.assertEqual([type(stage) for stage in stage_list], sorted_stage_list)
def test_sort_components_reverse(self):
rev_stages = self.component_manager.sort_components(reverse=True)
reverse_default_components_sort = reversed(self.default_components_sort)
for stage_list, sorted_stage_list in zip(rev_stages, reverse_default_components_sort):
self.assertEqual([type(stage) for stage in stage_list], sorted_stage_list)
def test_get_component_not_exists(self):
with self.assertRaises(NameError):
self.component_manager.get_component("random_component")
2019-01-24 02:57:56 +01:00
class TestComponentManagerOverrides(AsyncioTestCase):
2019-01-24 02:26:29 +01:00
def test_init_with_overrides(self):
class FakeWallet:
component_name = "wallet"
depends_on = []
def __init__(self, component_manager):
self.component_manager = component_manager
@property
def component(self):
return self
2019-01-21 21:55:50 +01:00
new_component_manager = ComponentManager(Config(), wallet=FakeWallet)
fake_wallet = new_component_manager.get_component("wallet")
# wallet should be an instance of FakeWallet and not WalletComponent from Components.py
self.assertIsInstance(fake_wallet, FakeWallet)
self.assertNotIsInstance(fake_wallet, Components.WalletComponent)
def test_init_with_wrong_overrides(self):
class FakeRandomComponent:
component_name = "someComponent"
depends_on = []
with self.assertRaises(SyntaxError):
2019-01-21 21:55:50 +01:00
ComponentManager(Config(), randomComponent=FakeRandomComponent)
2019-01-24 02:26:29 +01:00
class FakeComponent:
depends_on = []
component_name = None
def __init__(self, component_manager):
self.component_manager = component_manager
self._running = False
@property
def running(self):
return self._running
async def start(self):
pass
async def stop(self):
pass
@property
def component(self):
return self
async def _setup(self):
result = await self.start()
self._running = True
return result
async def _stop(self):
result = await self.stop()
self._running = False
return result
async def get_status(self):
return {}
def __lt__(self, other):
return self.component_name < other.component_name
class FakeDelayedWallet(FakeComponent):
component_name = "wallet"
depends_on = []
async def stop(self):
await asyncio.sleep(1)
class FakeDelayedBlobManager(FakeComponent):
component_name = "blob_manager"
depends_on = [FakeDelayedWallet.component_name]
async def start(self):
await asyncio.sleep(1)
async def stop(self):
await asyncio.sleep(1)
class FakeDelayedStreamManager(FakeComponent):
component_name = "stream_manager"
depends_on = [FakeDelayedBlobManager.component_name]
async def start(self):
await asyncio.sleep(1)
class TestComponentManagerProperStart(AdvanceTimeTestCase):
2019-01-24 02:26:29 +01:00
def setUp(self):
self.component_manager = ComponentManager(
2019-01-21 21:55:50 +01:00
Config(),
2019-01-24 02:26:29 +01:00
skip_components=[
DATABASE_COMPONENT, DHT_COMPONENT, HASH_ANNOUNCER_COMPONENT,
PEER_PROTOCOL_SERVER_COMPONENT, UPNP_COMPONENT,
HEADERS_COMPONENT, EXCHANGE_RATE_MANAGER_COMPONENT],
wallet=FakeDelayedWallet,
stream_manager=FakeDelayedStreamManager,
blob_manager=FakeDelayedBlobManager
)
async def test_proper_starting_of_components(self):
2019-01-24 02:26:29 +01:00
asyncio.create_task(self.component_manager.start())
await self.advance(0)
self.assertTrue(self.component_manager.get_component('wallet').running)
2018-08-02 21:54:43 +02:00
self.assertFalse(self.component_manager.get_component('blob_manager').running)
2019-01-24 02:26:29 +01:00
self.assertFalse(self.component_manager.get_component('stream_manager').running)
await self.advance(1)
self.assertTrue(self.component_manager.get_component('wallet').running)
2018-08-02 21:54:43 +02:00
self.assertTrue(self.component_manager.get_component('blob_manager').running)
2019-01-24 02:26:29 +01:00
self.assertFalse(self.component_manager.get_component('stream_manager').running)
await self.advance(1)
self.assertTrue(self.component_manager.get_component('wallet').running)
2018-08-02 21:54:43 +02:00
self.assertTrue(self.component_manager.get_component('blob_manager').running)
2019-01-24 02:26:29 +01:00
self.assertTrue(self.component_manager.get_component('stream_manager').running)
async def test_proper_stopping_of_components(self):
2019-01-24 02:26:29 +01:00
asyncio.create_task(self.component_manager.start())
await self.advance(0)
await self.advance(1)
await self.advance(1)
self.assertTrue(self.component_manager.get_component('wallet').running)
self.assertTrue(self.component_manager.get_component('blob_manager').running)
2019-01-24 02:26:29 +01:00
self.assertTrue(self.component_manager.get_component('stream_manager').running)
asyncio.create_task(self.component_manager.stop())
await self.advance(0)
2019-01-24 02:26:29 +01:00
self.assertFalse(self.component_manager.get_component('stream_manager').running)
2018-08-02 21:54:43 +02:00
self.assertTrue(self.component_manager.get_component('blob_manager').running)
self.assertTrue(self.component_manager.get_component('wallet').running)
await self.advance(1)
2019-01-24 02:26:29 +01:00
self.assertFalse(self.component_manager.get_component('stream_manager').running)
2018-08-02 21:54:43 +02:00
self.assertFalse(self.component_manager.get_component('blob_manager').running)
self.assertTrue(self.component_manager.get_component('wallet').running)
await self.advance(1)
2019-01-24 02:26:29 +01:00
self.assertFalse(self.component_manager.get_component('stream_manager').running)
2018-08-02 21:54:43 +02:00
self.assertFalse(self.component_manager.get_component('blob_manager').running)
self.assertFalse(self.component_manager.get_component('wallet').running)