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