def MockSyncPlugin(name, app): from zgres.plugin import get_plugin_manager from zgres import sync pm = get_plugin_manager([], sync) p = mock.Mock(spec=['state', 'conn_info', 'masters', 'databases']) class Proxy: pass proxy = Proxy() hooks = setup_proxy(proxy, pm, p) for name in hooks: # return None for all hooks by default getattr(p, name).return_value = None return proxy, p
def pmC(pluginC): from zgres.plugin import get_plugin_manager from zgres import deadman return get_plugin_manager([pluginC], deadman).hook
def pmB(pluginB): from zgres.plugin import get_plugin_manager from zgres import deadman return get_plugin_manager([('B', pluginB)], deadman).hook
def pmA(pluginA): from zgres.plugin import get_plugin_manager from zgres import deadman return get_plugin_manager([('A', pluginA)], deadman).hook