intercept/tests/test_system.py

"""Tests for the System Health monitoring blueprint."""

from __future__ import annotations

from unittest.mock import MagicMock, patch


def _login(client):
    """Mark the Flask test session as authenticated."""
    with client.session_transaction() as sess:
        sess['logged_in'] = True
        sess['username'] = 'test'
        sess['role'] = 'admin'


def test_metrics_returns_expected_keys(client):
    """GET /system/metrics returns top-level metric keys."""
    _login(client)
    resp = client.get('/system/metrics')
    assert resp.status_code == 200
    data = resp.get_json()
    assert 'system' in data
    assert 'processes' in data
    assert 'cpu' in data
    assert 'memory' in data
    assert 'disk' in data
    assert data['system']['hostname']
    assert 'version' in data['system']
    assert 'uptime_seconds' in data['system']
    assert 'uptime_human' in data['system']


def test_metrics_enhanced_keys(client):
    """GET /system/metrics returns enhanced metric keys."""
    _login(client)
    resp = client.get('/system/metrics')
    assert resp.status_code == 200
    data = resp.get_json()
    # New enhanced keys
    assert 'network' in data
    assert 'disk_io' in data
    assert 'boot_time' in data
    assert 'battery' in data
    assert 'fans' in data
    assert 'power' in data

    # CPU should have per_core and freq
    if data['cpu'] is not None:
        assert 'per_core' in data['cpu']
        assert 'freq' in data['cpu']

    # Network should have interfaces and connections
    if data['network'] is not None:
        assert 'interfaces' in data['network']
        assert 'connections' in data['network']
        assert 'io' in data['network']


def test_metrics_without_psutil(client):
    """Metrics degrade gracefully when psutil is unavailable."""
    _login(client)
    import routes.system as mod

    orig = mod._HAS_PSUTIL
    mod._HAS_PSUTIL = False
    try:
        resp = client.get('/system/metrics')
        assert resp.status_code == 200
        data = resp.get_json()
        # These fields should be None without psutil
        assert data['cpu'] is None
        assert data['memory'] is None
        assert data['disk'] is None
        assert data['network'] is None
        assert data['disk_io'] is None
        assert data['battery'] is None
        assert data['boot_time'] is None
        assert data['power'] is None
    finally:
        mod._HAS_PSUTIL = orig


def test_sdr_devices_returns_list(client):
    """GET /system/sdr_devices returns a devices list."""
    _login(client)
    mock_device = MagicMock()
    mock_device.sdr_type = MagicMock()
    mock_device.sdr_type.value = 'rtlsdr'
    mock_device.index = 0
    mock_device.name = 'Generic RTL2832U'
    mock_device.serial = '00000001'
    mock_device.driver = 'rtlsdr'

    with patch('utils.sdr.detection.detect_all_devices', return_value=[mock_device]):
        resp = client.get('/system/sdr_devices')
    assert resp.status_code == 200
    data = resp.get_json()
    assert 'devices' in data
    assert len(data['devices']) == 1
    assert data['devices'][0]['type'] == 'rtlsdr'
    assert data['devices'][0]['name'] == 'Generic RTL2832U'


def test_sdr_devices_handles_detection_failure(client):
    """SDR detection failure returns empty list with error."""
    _login(client)
    with patch('utils.sdr.detection.detect_all_devices', side_effect=RuntimeError('no devices')):
        resp = client.get('/system/sdr_devices')
    assert resp.status_code == 200
    data = resp.get_json()
    assert data['devices'] == []
    assert 'error' in data


def test_stream_returns_sse_content_type(client):
    """GET /system/stream returns text/event-stream."""
    _login(client)
    resp = client.get('/system/stream')
    assert resp.status_code == 200
    assert 'text/event-stream' in resp.content_type


def test_location_returns_shape(client):
    """GET /system/location returns lat/lon/source shape."""
    _login(client)
    with patch('routes.system.contextlib.suppress'):
        resp = client.get('/system/location')
    assert resp.status_code == 200
    data = resp.get_json()
    assert 'lat' in data
    assert 'lon' in data
    assert 'source' in data


def test_location_falls_back_to_config(client):
    """Location endpoint returns config defaults when GPS unavailable."""
    _login(client)
    with patch('routes.system.DEFAULT_LATITUDE', 40.7128, create=True), \
         patch('routes.system.DEFAULT_LONGITUDE', -74.006, create=True):
        # Mock the import inside _get_observer_location
        resp = client.get('/system/location')
    assert resp.status_code == 200
    data = resp.get_json()
    assert 'source' in data


def test_weather_requires_location(client):
    """Weather endpoint returns error when no location available."""
    _login(client)
    # Without lat/lon params and no GPS state or config
    resp = client.get('/system/weather')
    assert resp.status_code == 200
    data = resp.get_json()
    # Either returns weather or error (depending on config)
    assert 'error' in data or 'temp_c' in data


def test_weather_with_mocked_response(client):
    """Weather endpoint returns parsed weather data with mocked HTTP."""
    _login(client)
    mock_resp = MagicMock()
    mock_resp.status_code = 200
    mock_resp.json.return_value = {
        'current_condition': [{
            'temp_C': '22',
            'temp_F': '72',
            'weatherDesc': [{'value': 'Clear'}],
            'humidity': '45',
            'windspeedMiles': '8',
            'winddir16Point': 'NW',
            'FeelsLikeC': '20',
            'visibility': '10',
            'pressure': '1013',
        }]
    }
    mock_resp.raise_for_status = MagicMock()

    import routes.system as mod
    # Clear cache
    mod._weather_cache.clear()
    mod._weather_cache_time = 0.0

    with patch('routes.system._requests') as mock_requests:
        mock_requests.get.return_value = mock_resp
        resp = client.get('/system/weather?lat=40.7&lon=-74.0')

    assert resp.status_code == 200
    data = resp.get_json()
    assert data['temp_c'] == '22'
    assert data['condition'] == 'Clear'
    assert data['humidity'] == '45'
    assert data['wind_mph'] == '8'