File size: 3,070 Bytes
fea1bd1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 |
# -*- coding: utf-8 -*-
"""
Testes para o ponto de entrada principal (main.py).
Estes são testes de integração de alto nível (smoke tests) que garantem
que a aplicação pode ser inicializada e que o fluxo principal pode ser
executado sem erros, usando mocks para os componentes externos.
"""
import pytest
import asyncio
from unittest.mock import patch, MagicMock, AsyncMock
# Mock do hardware_manager para evitar dependência de GPU real
@pytest.fixture
def mock_hardware():
mock = MagicMock()
mock.is_initialized = True
mock.get_live_report.return_value = {
0: MagicMock(name="Mock GPU 0", temperature_c=50, utilization_percent=10, used_memory_mb=100, total_memory_mb=8000, free_memory_mb=7900)
}
mock.assign_model_to_gpu.return_value = 0
return mock
# Mock do session_manager
@pytest.fixture
def mock_session():
mock = MagicMock()
mock.create_session.return_value = "mock-session-id"
return mock
# Mock do command_dispatcher
@pytest.fixture
def mock_dispatcher():
mock = MagicMock()
# O dispatch_async retorna um Awaitable (Future)
future = asyncio.Future()
future.set_result({"status": "ok"})
mock.dispatch_async.return_value = future
return mock
@pytest.mark.asyncio
@patch('superezio_enterprise.main.hardware_manager')
@patch('superezio_enterprise.main.session_manager')
@patch('superezio_enterprise.main.command_dispatcher')
@patch('superezio_enterprise.main.setup_enterprise_logging')
async def test_main_execution_flow(
mock_setup_logging, mock_dispatcher_main, mock_session_main, mock_hardware_main,
mock_hardware, mock_session, mock_dispatcher
):
"""Testa o fluxo de execução da função main, garantindo as chamadas esperadas."""
# Associa os mocks dos fixtures aos patches do main
mock_hardware_main.return_value = mock_hardware
mock_session_main.return_value = mock_session
mock_dispatcher_main.return_value = mock_dispatcher
# Importa o main aqui para que ele use os mocks
from superezio_enterprise import main
# Reduz o tempo de sleep para acelerar o teste
with patch('asyncio.sleep', new_callable=AsyncMock):
await main.main()
# Verifica se as funções chave foram chamadas
mock_setup_logging.assert_called_once()
mock_session.create_session.assert_called_once_with(user_id="gpu_test_user")
# Verifica se a alocação de modelos foi tentada
assert mock_hardware.assign_model_to_gpu.call_count >= 1
mock_hardware.assign_model_to_gpu.assert_any_call(
"large-language-model-v1", required_vram_mb=8000
)
# Verifica se o monitoramento foi chamado
assert mock_hardware.get_live_report.call_count > 0
# Verifica se a liberação foi chamada
mock_hardware.release_model.assert_any_call("large-language-model-v1")
# Verifica se o health check foi chamado no final
mock_dispatcher.dispatch_async.assert_called_once_with("system:health_check")
# Verifica se o shutdown foi chamado para limpar os recursos
mock_hardware.shutdown.assert_called_once()
|