def test_remove_all_plugins():
assert get_plugins() == OrderedDict()
manually_set_plugins(OrderedDict([
('p1', LightbusPlugin()),
]))
remove_all_plugins()
assert get_plugins() == OrderedDict()
def test_autoload_plugins():
assert get_plugins() == OrderedDict()
assert autoload_plugins()
assert [(name, p.__class__) for name, p in get_plugins().items()] == [
('internal_state', StatePlugin),
('internal_metrics', MetricsPlugin),
]
def test_autoload_plugins():
config = Config.load_dict({})
assert get_plugins() is None
assert autoload_plugins(config)
assert [(name, p.__class__) for name, p in get_plugins().items()] == [
("internal_state", StatePlugin),
("internal_metrics", MetricsPlugin),
]
def test_manually_set_plugins():
assert get_plugins() == OrderedDict()
p1 = LightbusPlugin()
p2 = LightbusPlugin()
manually_set_plugins(OrderedDict([
('p1', p1),
('p2', p2),
]))
assert get_plugins() == OrderedDict([
('p1', p1),
('p2', p2),
])
def test_is_plugin_loaded():
assert get_plugins() == OrderedDict()
assert is_plugin_loaded(LightbusPlugin) == False
manually_set_plugins(OrderedDict([
('p1', LightbusPlugin()),
]))
assert is_plugin_loaded(LightbusPlugin) == True
async def test_plugin_hook(mocker):
"""Ensure calling plugin_hook() calls the method on the plugin"""
assert get_plugins() == OrderedDict()
plugin = LightbusPlugin()
manually_set_plugins(OrderedDict([
('p1', plugin),
]))
async def dummy_coroutine(*args, **kwargs):
pass
m = mocker.patch.object(plugin, 'before_server_start', return_value=dummy_coroutine())
await plugin_hook('before_server_start', bus_client=None, loop=None)
assert m.called
def test_is_plugin_loaded():
assert get_plugins() is None
assert is_plugin_loaded(LightbusPlugin) == False
manually_set_plugins(OrderedDict([("p1", LightbusPlugin())]))
assert is_plugin_loaded(LightbusPlugin) == True
def test_remove_all_plugins():
assert get_plugins() is None
manually_set_plugins(OrderedDict([("p1", LightbusPlugin())]))
remove_all_plugins()
assert get_plugins() is None
def test_manually_set_plugins():
assert get_plugins() is None
p1 = LightbusPlugin()
p2 = LightbusPlugin()
manually_set_plugins(OrderedDict([("p1", p1), ("p2", p2)]))
assert get_plugins() == OrderedDict([("p1", p1), ("p2", p2)])