def test_service_statuss_true(self, monkeypatch):
def mock_command_constructor(obj, command):
assert command == "systemctl status " + \
"tendrl-node-agent.service"
return
monkeypatch.setattr(cmd_utils.Command, '__init__',
mock_command_constructor)
def mock_command_run(obj, exec_path):
stdout = "tendrl-node-agent.service - A python agent local to" + \
" every managed storage node in the sds cluster " + \
" Loaded: loaded (/usr/lib/systemd/system/tendrl" + \
"-node-agent.service;disabled;vendor preset:disabled)" + \
" Active:active(running)since Thu 2017-01-12 19:14:32" + \
" IST; 15h ago Main PID: 5216 (tendrl-node-age) " + \
" CGroup: /system.slice/tendrl-node-agent.service"
stderr = ""
rc = 0
return stdout, stderr, rc
monkeypatch.setattr(cmd_utils.Command, 'run', mock_command_run)
service = ServiceStatus("tendrl-node-agent", '/tmp/')
status = service.status()
assert status
def test_status():
service_status = ServiceStatus("service_status")
ret = service_status.status()
assert ret is False
with patch.object(cmd_utils.Command, 'run') as mock_run:
mock_run.return_value = run(True)
ret = service_status.status()
assert ret is False
with patch.object(cmd_utils.Command, 'run') as mock_run:
mock_run.return_value = run(False)
ret = service_status.status()
assert ret is True
def test_service_status_false(self, monkeypatch):
def mock_command_constructor(obj, command):
assert command == "systemctl status " + \
"tendrl-node-agent.service"
return
monkeypatch.setattr(cmd_utils.Command, '__init__',
mock_command_constructor)
def mock_command_run(obj, exec_path):
stdout = ""
stderr = ""
rc = 1
return stdout, stderr, rc
monkeypatch.setattr(cmd_utils.Command, 'run', mock_command_run)
service = ServiceStatus("tendrl-node-agent", '/tmp/')
status = service.status()
assert not status
def test_service_exists_true(self, monkeypatch):
def mock_command_constructor(obj, command):
assert command == "systemctl show -p LoadState" + \
" tendrl-node-agent.service"
return
monkeypatch.setattr(cmd_utils.Command, '__init__',
mock_command_constructor)
def mock_command_run(obj, exec_path):
stdout = "LoadState=loaded"
stderr = ""
rc = 0
return stdout, stderr, rc
monkeypatch.setattr(cmd_utils.Command, 'run', mock_command_run)
service = ServiceStatus("tendrl-node-agent", '/tmp/')
exists = service.exists()
assert exists
def test_exists():
service_status = ServiceStatus("service_status")
ret = service_status.exists()
assert ret is False
def import_gluster(parameters):
logging_file_name = "gluster-integration_logging.yaml"
logging_config_file_path = "/etc/tendrl/gluster-integration/"
attributes = {}
if NS.config.data['package_source_type'] == 'pip':
_cmd = "nohup tendrl-gluster-integration &"
name = "https://github.com/Tendrl/gluster-integration/archive/master" \
".tar.gz"
attributes["name"] = name
attributes["editable"] = "false"
ansible_module_path = "packaging/language/pip.py"
elif NS.config.data['package_source_type'] == 'rpm':
name = "tendrl-gluster-integration"
_cmd = "systemctl restart %s" % name
ansible_module_path = "packaging/os/yum.py"
attributes["name"] = name
else:
return (False, "Invalid package_source_type: %s" %
NS.config.data['package_source_type'])
logger.log(
"info",
NS.publisher_id,
{
'message':
"Installing tendrl-gluster-integration on "
"%s" % NS.node_context.fqdn
},
job_id=parameters['job_id'],
flow_id=parameters['flow_id'],
)
try:
runner = ansible_module_runner.AnsibleRunner(ansible_module_path,
**attributes)
except ansible_module_runner.AnsibleModuleNotFound:
# Backward compat ansible<=2.2
runner = ansible_module_runner.AnsibleRunner(
"core/" + ansible_module_path, **attributes)
try:
out, err = runner.run()
if type(out) is dict and out.get('rc', 1) != 0:
if 'msg' in out:
err = out['msg']
elif 'stderr' in out:
err = out['stderr']
else:
err = out
err_msg = "Could not install tendrl-gluster-integration " \
"on %s. Error: %s" % (NS.node_context.fqdn, err)
logger.log(
"error",
NS.publisher_id,
{"message": err_msg},
job_id=parameters['job_id'],
flow_id=parameters['flow_id'],
)
return False, err_msg
except ansible_module_runner.AnsibleExecutableGenerationFailed:
err_msg = "Cluster management requires service " \
"tendrl-gluster-integration. Failed to install same" \
"on Node %s" % NS.node_context.fqdn
logger.log(
"error",
NS.publisher_id,
{"message": err_msg},
job_id=parameters['job_id'],
flow_id=parameters['flow_id'],
)
return False, err_msg
logger.log(
"info",
NS.publisher_id,
{
"message":
"Generating tendrl-gluster-integration "
"configuration on %s" % NS.node_context.fqdn
},
job_id=parameters['job_id'],
flow_id=parameters['flow_id'],
)
with open(logging_config_file_path + logging_file_name, 'w+') as f:
f.write(pkg_resources.resource_string(__name__, logging_file_name))
gluster_integration_tag = NS.compiled_definitions.get_parsed_defs(
)['namespace.tendrl']['tags']['tendrl-gluster-integration']
config_data = {
"etcd_port": int(NS.config.data['etcd_port']),
"etcd_connection": str(NS.config.data['etcd_connection']),
"log_cfg_path": (logging_config_file_path + logging_file_name),
"log_level": "DEBUG",
"logging_socket_path": "/var/run/tendrl/message.sock",
"sync_interval": int(NS.config.data['sync_interval']),
"tags": [gluster_integration_tag]
}
etcd_ca_cert_file = NS.config.data.get("etcd_ca_cert_file")
etcd_cert_file = NS.config.data.get("etcd_cert_file")
etcd_key_file = NS.config.data.get("etcd_key_file")
if etcd_ca_cert_file and str(
etcd_ca_cert_file) != "" and etcd_cert_file and str(
etcd_cert_file) != "" and etcd_key_file and str(
etcd_key_file) != "":
config_data.update({
"etcd_ca_cert_file":
NS.config.data['etcd_ca_cert_file'],
"etcd_cert_file":
NS.config.data['etcd_cert_file'],
"etcd_key_file":
NS.config.data['etcd_key_file']
})
_gluster_integration_conf_file_path = \
"/etc/tendrl/gluster-integration/gluster-integration.conf.yaml"
with open(_gluster_integration_conf_file_path, 'w') as outfile:
yaml.dump(config_data, outfile, default_flow_style=False)
logger.log(
"info",
NS.publisher_id,
{
"message":
"Running tendrl-gluster-integration on "
"%s" % NS.node_context.fqdn
},
job_id=parameters['job_id'],
flow_id=parameters['flow_id'],
)
os.chmod(_gluster_integration_conf_file_path, 0o640)
if NS.config.data['package_source_type'] == 'rpm':
command = cmd_utils.Command(
"systemctl enable tendrl-gluster-integration")
err, out, rc = command.run()
if err:
err_msg = "Could not enable tendrl-gluster-integration " \
"service. Error: %s" % err
logger.log(
"error",
NS.publisher_id,
{"message": err_msg},
job_id=parameters['job_id'],
flow_id=parameters['flow_id'],
)
return False, err_msg
cmd = cmd_utils.Command(_cmd)
err, out, rc = cmd.run()
if err:
err_msg = "Could not start tendrl-gluster-integration " \
"service. Error: %s" % err
logger.log(
"error",
NS.publisher_id,
{"message": err_msg},
job_id=parameters['job_id'],
flow_id=parameters['flow_id'],
)
return False, err_msg
# Check if tendrl-gluster-integration active on the node
response = ServiceStatus("tendrl-gluster-integration").status()
if not response:
return (
False,
"Cluster management requires service tendrl-gluster-integration."
" The same is not running on node: %s" % NS.node_context.fqdn)
return True, ""
def get_service_info(service_name):
service = ServiceStatus(service_name)
return {"exists": service.exists(), "running": service.status()}