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