Python LightbusPlugin示例

示例#1

文件： test_unit_plugin_base.py 项目： holycrepe/lightbus

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

示例#2

文件： test_unit_plugin_base.py 项目： holycrepe/lightbus

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

示例#3

文件： test_unit_plugin_base.py 项目： holycrepe/lightbus

def test_remove_all_plugins():
    assert get_plugins() == OrderedDict()
    manually_set_plugins(OrderedDict([
        ('p1', LightbusPlugin()),
    ]))
    remove_all_plugins()
    assert get_plugins() == OrderedDict()

示例#4

文件： test_unit_plugin_base.py 项目： imlzg/lightbus

async def test_execute_hook(mocker, plugin_registry: PluginRegistry):
    """Ensure calling execute_hook() calls the method on the plugin"""
    assert not plugin_registry._plugins
    plugin = LightbusPlugin()
    plugin_registry.set_plugins([plugin])

    async def dummy_coroutine(*args, **kwargs):
        pass

    m = mocker.patch.object(plugin, "before_worker_start", return_value=dummy_coroutine())

    await plugin_registry.execute_hook("before_worker_start", client=None, loop=None)
    assert m.called

示例#5

文件： test_unit_plugin_base.py 项目： holycrepe/lightbus

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

示例#6

文件： test_unit_plugin_hooks.py 项目： rizplate/lightbus

 def do_add_base_plugin():
     manually_set_plugins(plugins={"base": LightbusPlugin()})

示例#7

 def do_add_base_plugin():
     dummy_bus.client.plugin_registry.set_plugins([LightbusPlugin()])

示例#8

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

示例#9

def test_remove_all_plugins():
    assert get_plugins() is None
    manually_set_plugins(OrderedDict([("p1", LightbusPlugin())]))
    remove_all_plugins()
    assert get_plugins() is None

示例#10

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

示例#11

def test_is_plugin_loaded(plugin_registry: PluginRegistry):
    assert plugin_registry.is_plugin_loaded(LightbusPlugin) == False
    plugin_registry.set_plugins([LightbusPlugin()])
    assert plugin_registry.is_plugin_loaded(LightbusPlugin) == True

示例#12

def test_manually_set_plugins(plugin_registry: PluginRegistry):
    assert not plugin_registry._plugins
    p1 = LightbusPlugin()
    p2 = LightbusPlugin()
    plugin_registry.set_plugins([p1, p2])
    assert plugin_registry._plugins == [p1, p2]