def restart():
"""Wrapper for 'restart_manager' function due to interdependence with cluster module and permission access. """
result = AffectedItemsWazuhResult(**_restart_default_result_kwargs)
try:
manager_restart()
result.affected_items.append(node_id)
except WazuhError as e:
result.add_failed_item(id_=node_id, error=e)
result.total_affected_items = len(result.affected_items)
return result
def test_manager_restart():
"""Verify that manager_restart send to the manager the restart request."""
with patch('wazuh.core.cluster.utils.open', side_effect=None):
with patch('fcntl.lockf', side_effect=None):
with pytest.raises(WazuhInternalError, match='.* 1901 .*'):
utils.manager_restart()
with patch('wazuh.core.cluster.utils.exists', return_value=True):
with pytest.raises(WazuhInternalError, match='.* 1902 .*'):
utils.manager_restart()
with patch('socket.socket.connect', side_effect=None):
with pytest.raises(WazuhInternalError, match='.* 1014 .*'):
utils.manager_restart()
with patch('socket.socket.send', side_effect=None):
status = utils.manager_restart()
assert WazuhResult({'message': 'Restart request sent'}) == status
