def restart_all_nodes():
"""
Restart all cluster nodes.
:return: Confirmation message.
"""
restart()
return "Restarting cluster"
def test_restart_ko_socket(self, mock1):
with self.assertRaises(WazuhException) as cm:
restart()
self.assertEqual(cm.exception.code, 1901)
def test_restart_ok(self, mock1):
self.assertEqual(restart(), 'Manager is going to restart now')
def test_restart_ok(test_manager):
"""
Tests restarting a manager
"""
assert restart() == 'Restart request sent'
def test_restart_ko_socket(test_manager):
"""
Tests restarting a manager when the socket is not created
"""
with pytest.raises(WazuhException, match='.* 1901 .*'):
restart()
