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)])