def test_restart_agents_by_group_ok(mock_get_agent_group, mock_ossec_queue,
test_data):
"""Test restart_agents_by_group method when all agents are restarted."""
with patch('sqlite3.connect') as mock_db:
mock_db.return_value = test_data.global_db
result = Agent.restart_agents_by_group('dmz')
# check result fields
assert set(result.keys()) == {'msg', 'affected_agents'}
assert result['msg'] == 'All selected agents were restarted'
assert set(result['affected_agents']) == {'001', '002'}
def test_restart_agents_by_group_ko(mock_get_agent_group, mock_ossec_queue,
test_data):
"""Test restart_agents_by_group method when some agents are not restarted."""
with patch('sqlite3.connect') as mock_db:
mock_db.return_value = test_data.global_db
result = Agent.restart_agents_by_group('dmz')
# check result fields
assert set(result.keys()) == {'failed_ids', 'msg', 'affected_agents'}
assert result['msg'] == 'Some agents were not restarted'
assert set(result['affected_agents']) == {'001', '002'}
assert isinstance(result['failed_ids'], list)
for failed_id in result['failed_ids']:
assert set(failed_id.keys()) == {'id', 'error'}
assert isinstance(failed_id['id'], str)
assert set(failed_id['error']) == {'message', 'code'}
