def test_excluded_plugins(self):
self.assertTrue('linux' in DevicePluginManager.get_plugins())
with patch('chroma_agent.plugin_manager.EXCLUDED_PLUGINS', ['linux']):
with patch.object(DevicePluginManager, '_plugins', {}):
self.assertTrue(
'linux' not in DevicePluginManager.get_plugins())
def test_excluded_plugins(self):
self.assertTrue("linux" in DevicePluginManager.get_plugins())
with patch("chroma_agent.plugin_manager.EXCLUDED_PLUGINS", ["linux"]):
with patch.object(DevicePluginManager, "_plugins", {}):
self.assertTrue(
"linux" not in DevicePluginManager.get_plugins())
def main():
"""handle unexpected exceptions"""
parser = argparse.ArgumentParser(
description="Integrated Manager for Lustre software Agent")
parser.add_argument("--publish-zconf", action="store_true")
parser.parse_args()
signal.signal(signal.SIGHUP, signal.SIG_IGN)
daemon_log_setup()
console_log_setup()
daemon_log.info("Starting")
try:
daemon_log.info("Entering main loop")
try:
url = urljoin(os.environ["IML_MANAGER_URL"], "agent/message/")
except KeyError as e:
daemon_log.error(
"No configuration found (must be registered before running the agent service), "
"details: %s" % e)
return
if config.profile_managed is False:
# This is kind of terrible. The design of DevicePluginManager is
# such that it can be called with either class methods or
# instantiated and then called with instance methods. As such,
# we can't pass in a list of excluded plugins to the instance
# constructor. Well, we could, but it would only work some
# of the time and that would be even more awful.
import chroma_agent.plugin_manager
chroma_agent.plugin_manager.EXCLUDED_PLUGINS += ['corosync']
agent_client = AgentClient(url, ActionPluginManager(),
DevicePluginManager(), ServerProperties(),
Crypto(ENV_PATH))
def teardown_callback(*args, **kwargs):
agent_client.stop()
agent_client.join()
[function() for function in agent_daemon_teardown_functions]
signal.signal(signal.SIGINT, teardown_callback)
signal.signal(signal.SIGTERM, teardown_callback)
signal.signal(signal.SIGUSR1, decrease_loglevel)
signal.signal(signal.SIGUSR2, increase_loglevel)
# Call any agent daemon startup methods that were registered.
[function() for function in agent_daemon_startup_functions]
agent_client.start()
# Waking-wait to pick up signals
while not agent_client.stopped.is_set():
agent_client.stopped.wait(timeout=10)
agent_client.join()
except Exception, e:
backtrace = '\n'.join(traceback.format_exception(*(sys.exc_info())))
daemon_log.error("Unhandled exception: %s" % backtrace)
def reregister_server(url, address):
""" Update manager url and register agent address with manager """
if _service_is_running() is True:
console_log.warning(
"chroma-agent service was running before registration, stopping.")
agent_service.stop()
config.set('settings', 'server', {'url': url})
crypto = Crypto(config.path)
agent_client = AgentClient(url + 'reregister/', ActionPluginManager(),
DevicePluginManager(), ServerProperties(),
crypto)
data = {'address': address, 'fqdn': agent_client._fqdn}
try:
result = agent_client.post(data)
except HttpError:
console_log.error("Reregistration failed to %s with request %s" %
(agent_client.url, data))
raise
console_log.info("Starting chroma-agent service")
agent_service.start()
return result
def register_server(url, ca, secret, address=None):
if _service_is_running() is True:
console_log.warning(
"chroma-agent service was running before registration, stopping.")
agent_service.stop()
crypto = Crypto(config.path)
# Call delete in case we are over-writing a previous configuration that wasn't removed properly
crypto.delete()
crypto.install_authority(ca)
agent_client = AgentClient(url + "register/%s/" % secret,
ActionPluginManager(), DevicePluginManager(),
ServerProperties(), crypto)
registration_result = agent_client.register(address)
crypto.install_certificate(registration_result['certificate'])
config.set('settings', 'server', {'url': url})
console_log.info("Enabling chroma-agent service")
agent_service.enable()
console_log.info("Starting chroma-agent service")
agent_service.start()
return registration_result
def reregister_server(url, address):
""" Update manager url and register agent address with manager """
if _service_is_running() is True:
console_log.warning(
"chroma-agent service was running before registration, stopping.")
agent_service.stop()
conf.set_server_url(url)
crypto = Crypto(conf.ENV_PATH)
agent_client = AgentClient(
url + "reregister/",
ActionPluginManager(),
DevicePluginManager(),
ServerProperties(),
crypto,
)
data = {"address": address, "fqdn": agent_client._fqdn}
try:
result = agent_client.post(data)
except HttpError:
console_log.error("Reregistration failed to %s with request %s" %
(agent_client.url, data))
raise
console_log.info("Starting chroma-agent service")
agent_service.start()
return result
def device_plugin(plugin=None):
"""
Invoke a device plugin once to obtain a snapshot of what it
is monitoring
:param plugin: Plugin module name, or None for all plugins
:return: dict of plugin name to data object
"""
all_plugins = DevicePluginManager.get_plugins()
if plugin is None:
plugins = all_plugins
elif plugin == "":
plugins = {}
else:
plugins = {plugin: all_plugins[plugin]}
result = {}
for plugin_name, plugin_class in plugins.items():
result[plugin_name] = plugin_class(None).start_session()
return result
def test_get_device_plugins(self):
"""Test that we get a list of loaded plugin classes."""
self.assertNotEqual(len(DevicePluginManager.get_plugins()), 0)
"details: %s" % e)
return
if config.profile_managed is False:
# This is kind of terrible. The design of DevicePluginManager is
# such that it can be called with either class methods or
# instantiated and then called with instance methods. As such,
# we can't pass in a list of excluded plugins to the instance
# constructor. Well, we could, but it would only work some
# of the time and that would be even more awful.
import chroma_agent.plugin_manager
chroma_agent.plugin_manager.EXCLUDED_PLUGINS += ['corosync']
agent_client = AgentClient(conf['url'] + "message/",
ActionPluginManager(),
DevicePluginManager(), ServerProperties(),
Crypto(config.path))
def teardown_callback(*args, **kwargs):
agent_client.stop()
agent_client.join()
[function() for function in agent_daemon_teardown_functions]
if not args.foreground:
handlers = {
signal.SIGTERM: teardown_callback,
signal.SIGUSR1: decrease_loglevel,
signal.SIGUSR2: increase_loglevel
}
set_signal_handlers(handlers)
else:
return
if config.profile_managed is False:
# This is kind of terrible. The design of DevicePluginManager is
# such that it can be called with either class methods or
# instantiated and then called with instance methods. As such,
# we can't pass in a list of excluded plugins to the instance
# constructor. Well, we could, but it would only work some
# of the time and that would be even more awful.
import chroma_agent.plugin_manager
chroma_agent.plugin_manager.EXCLUDED_PLUGINS += ['corosync']
agent_client = AgentClient(
conf['url'] + "message/",
ActionPluginManager(),
DevicePluginManager(),
ServerProperties(),
Crypto(config.path))
def teardown_callback(*args, **kwargs):
agent_client.stop()
agent_client.join()
[function() for function in agent_daemon_teardown_functions]
if not args.foreground:
handlers = {
signal.SIGTERM: teardown_callback,
signal.SIGUSR1: decrease_loglevel,
signal.SIGUSR2: increase_loglevel
}
set_signal_handlers(handlers)