import pytest
from function.system_monitor import SystemMonitor
from typing import Dict

class TestSystemMonitor:
    @pytest.fixture
    def system_monitor(self):
        return SystemMonitor()
    
    def test_get_system_resources(self, system_monitor):
        """测试获取系统资源信息"""
        result = system_monitor.get_system_resources()
        
        # 验证返回格式
        assert isinstance(result, dict)
        assert 'status' in result
        assert result['status'] == 'success'
        assert 'resources' in result
        assert 'timestamp' in result
        
        resources = result['resources']
        
        # 验证GPU信息
        assert 'gpu' in resources
        if resources['gpu']:  # 如果有GPU
            gpu = resources['gpu'][0]
            assert 'id' in gpu
            assert 'name' in gpu
            assert 'memory' in gpu
            assert 'utilization' in gpu
            assert 'temperature' in gpu
            
        # 验证CPU信息
        assert 'cpu' in resources
        cpu = resources['cpu']
        assert 'count' in cpu
        assert 'utilization' in cpu
        assert 'memory' in cpu
        assert 'swap' in cpu
        
        # 验证内存信息
        memory = cpu['memory']
        assert 'total' in memory
        assert 'used' in memory
        assert 'free' in memory
        assert 'percent' in memory
        assert memory['total'] > 0
        assert 0 <= memory['percent'] <= 100
        
        # 验证磁盘信息
        assert 'disk' in resources
        assert len(resources['disk']) > 0
        for mount_point, disk_info in resources['disk'].items():
            assert 'total' in disk_info
            assert 'used' in disk_info
            assert 'free' in disk_info
            assert 'percent' in disk_info
            assert disk_info['total'] > 0
            assert 0 <= disk_info['percent'] <= 100
        
        # 验证进程信息
        assert 'processes' in resources
        processes = resources['processes']
        assert 'total' in processes
        assert 'running' in processes
        assert 'sleeping' in processes
        assert processes['total'] > 0
        assert processes['running'] >= 0
        assert processes['sleeping'] >= 0
        
    def test_error_handling(self, system_monitor, monkeypatch):
        """测试错误处理"""
        def mock_gpu_error(*args, **kwargs):
            raise Exception("GPU query failed")
            
        # 模拟GPU查询错误
        monkeypatch.setattr(system_monitor, '_get_gpu_info', mock_gpu_error)
        
        result = system_monitor.get_system_resources()
        assert result['status'] == 'success'  # 即使GPU查询失败，其他资源信息仍应返回
        assert result['resources']['gpu'] == []  # GPU信息应为空列表
        
        # 验证其他资源信息仍然可用
        assert 'cpu' in result['resources']
        assert 'disk' in result['resources']
        assert 'processes' in result['resources']