75a6ff269e
* add daemon Component and ComponentManager classes * convert directory and SQLiteStorage setup to be a Component * support callbacks to component setups * Fixed typo in ComponentManager * convert wallet to be Component * Use storage from session. * Remove create_session internal function and PEP8 * Starting to convert session to its own component. Removed ref to `self.storage` from Daemon.py * Making DHT component(broken) * Refactored classes to reduce redundancy in getting config setting * DHT is now it's own component * Fixed `test_streamify` test * Fixed regression caused by removing `peer_manager` from session * refactor ComponentManager and Component to use instance instead of class methods * Hash announcer, file manager, stream identifier components * Query Handler and server components * Reflector Component * Fixed test_streamify(well Jack did, but ¯\_(ツ)_/¯) * All tests now passing * Pylint fixes * Oops(That's all you're gonna get :-P) * Making decorators(WIP, commit so that I don't lose work) * Decorator made and decorating of functions done(some other changes) * import fixes and removed temporary test function * Fixed new broken tests from daemon refactor * Sanitization of modules * Reworded errors * wallet unlock condition checks, fixed breaking changes * Rebased on amster and other crazy stuff * Started writing tests * Tests for component manager * Fix Daemon Tests * Fixed passing mutable args in init * Using constants instead of strings. Added CHANGELOG.md * Now components can be skipped by setting relevant config in file. * P-Y-L-I-N-T #angry_emoji
132 lines
5.5 KiB
Python
132 lines
5.5 KiB
Python
from twisted.internet.task import Clock
|
|
from twisted.trial import unittest
|
|
|
|
from lbrynet.daemon.ComponentManager import ComponentManager
|
|
from lbrynet.daemon.Components import DATABASE_COMPONENT, DHT_COMPONENT, STREAM_IDENTIFIER_COMPONENT
|
|
from lbrynet.daemon.Components import HASH_ANNOUNCER_COMPONENT, REFLECTOR_COMPONENT, UPNP_COMPONENT
|
|
from lbrynet.daemon.Components import PEER_PROTOCOL_SERVER_COMPONENT
|
|
from lbrynet.daemon import Components
|
|
from lbrynet.tests import mocks
|
|
|
|
|
|
class TestComponentManager(unittest.TestCase):
|
|
def setUp(self):
|
|
mocks.mock_conf_settings(self)
|
|
self.default_components_sort = [
|
|
[Components.DatabaseComponent,
|
|
Components.UPnPComponent],
|
|
[Components.DHTComponent,
|
|
Components.WalletComponent],
|
|
[Components.HashAnnouncer],
|
|
[Components.SessionComponent],
|
|
[Components.PeerProtocolServer,
|
|
Components.StreamIdentifier],
|
|
[Components.FileManager],
|
|
[Components.ReflectorComponent]
|
|
]
|
|
self.component_manager = ComponentManager()
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
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.assertSetEqual(set([type(stage) for stage in stage_list]), set(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.assertSetEqual(set([type(stage) for stage in stage_list]), set(sorted_stage_list))
|
|
|
|
def test_get_component_not_exists(self):
|
|
|
|
with self.assertRaises(NameError):
|
|
self.component_manager.get_component("random_component")
|
|
|
|
|
|
class TestComponentManagerOverrides(unittest.TestCase):
|
|
def setUp(self):
|
|
mocks.mock_conf_settings(self)
|
|
|
|
def test_init_with_overrides(self):
|
|
class FakeWallet(object):
|
|
component_name = "wallet"
|
|
depends_on = []
|
|
|
|
def __init__(self, component_manager):
|
|
self.component_manager = component_manager
|
|
|
|
@property
|
|
def component(self):
|
|
return self
|
|
|
|
new_component_manager = ComponentManager(wallet=FakeWallet)
|
|
fake_wallet = new_component_manager.get_component("wallet")
|
|
# wallet should be an instance of FakeWallet and not WalletComponent from Components.py
|
|
self.assertEquals(type(fake_wallet), FakeWallet)
|
|
self.assertNotEquals(type(fake_wallet), Components.WalletComponent)
|
|
|
|
def test_init_with_wrong_overrides(self):
|
|
class FakeRandomComponent(object):
|
|
component_name = "someComponent"
|
|
depends_on = []
|
|
|
|
with self.assertRaises(SyntaxError):
|
|
ComponentManager(randomComponent=FakeRandomComponent)
|
|
|
|
|
|
class TestComponentManagerProperStart(unittest.TestCase):
|
|
def setUp(self):
|
|
self.reactor = Clock()
|
|
mocks.mock_conf_settings(self)
|
|
self.component_manager = ComponentManager(
|
|
skip_components=[DATABASE_COMPONENT, DHT_COMPONENT, HASH_ANNOUNCER_COMPONENT, STREAM_IDENTIFIER_COMPONENT,
|
|
PEER_PROTOCOL_SERVER_COMPONENT, REFLECTOR_COMPONENT, UPNP_COMPONENT],
|
|
reactor=self.reactor,
|
|
wallet=mocks.FakeDelayedWallet,
|
|
session=mocks.FakeDelayedSession,
|
|
fileManager=mocks.FakeDelayedFileManager
|
|
)
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def test_proper_starting_of_components(self):
|
|
self.component_manager.setup()
|
|
self.assertTrue(self.component_manager.get_component('wallet').running)
|
|
self.assertFalse(self.component_manager.get_component('session').running)
|
|
self.assertFalse(self.component_manager.get_component('file_manager').running)
|
|
|
|
self.reactor.advance(1)
|
|
self.assertTrue(self.component_manager.get_component('wallet').running)
|
|
self.assertTrue(self.component_manager.get_component('session').running)
|
|
self.assertFalse(self.component_manager.get_component('file_manager').running)
|
|
|
|
self.reactor.advance(1)
|
|
self.assertTrue(self.component_manager.get_component('wallet').running)
|
|
self.assertTrue(self.component_manager.get_component('session').running)
|
|
self.assertTrue(self.component_manager.get_component('file_manager').running)
|
|
|
|
def test_proper_stopping_of_components(self):
|
|
self.component_manager.setup()
|
|
self.reactor.advance(1)
|
|
self.reactor.advance(1)
|
|
self.component_manager.stop()
|
|
self.assertFalse(self.component_manager.get_component('file_manager').running)
|
|
self.assertTrue(self.component_manager.get_component('session').running)
|
|
self.assertTrue(self.component_manager.get_component('wallet').running)
|
|
|
|
self.reactor.advance(1)
|
|
self.assertFalse(self.component_manager.get_component('file_manager').running)
|
|
self.assertFalse(self.component_manager.get_component('session').running)
|
|
self.assertTrue(self.component_manager.get_component('wallet').running)
|
|
|
|
self.reactor.advance(1)
|
|
self.assertFalse(self.component_manager.get_component('file_manager').running)
|
|
self.assertFalse(self.component_manager.get_component('session').running)
|
|
self.assertFalse(self.component_manager.get_component('wallet').running)
|