diff --git a/tests/test_api_metrics.py b/tests/test_api_metrics.py index 7e60b2af..60d9f01e 100644 --- a/tests/test_api_metrics.py +++ b/tests/test_api_metrics.py @@ -21,6 +21,27 @@ def before_each() -> None: @pytest.fixture(scope = 'function', autouse = True) def mock_detect_execution_devices(mocker : MockerFixture) -> None: + mocker.patch('facefusion.system.state_manager.get_temp_path', return_value = '/tmp') + mocker.patch('facefusion.system.detect_disk_metrics', return_value = + [ + { + 'total': + { + 'value': 500, + 'unit': 'GB' + }, + 'free': + { + 'value': 200, + 'unit': 'GB' + }, + 'utilization': + { + 'value': 60, + 'unit': '%' + } + } + ]) mocker.patch('facefusion.system.detect_execution_devices', return_value = [ { @@ -99,6 +120,9 @@ def test_get_metrics(test_client : TestClient) -> None: assert metrics_body.get('execution_devices')[0].get('driver_version') == '555.42' assert metrics_body.get('execution_devices')[0].get('product').get('name') == 'RTX 4090' assert metrics_body.get('execution_devices')[0].get('video_memory').get('total').get('value') == 24 + assert metrics_body.get('disks')[0].get('total').get('value') == 500 + assert metrics_body.get('disks')[0].get('free').get('unit') == 'GB' + assert metrics_body.get('disks')[0].get('utilization').get('value') == 60 def test_websocket_metrics(test_client : TestClient) -> None: @@ -117,3 +141,6 @@ def test_websocket_metrics(test_client : TestClient) -> None: assert metrics_set.get('execution_devices')[0].get('driver_version') == '555.42' assert metrics_set.get('execution_devices')[0].get('product').get('name') == 'RTX 4090' assert metrics_set.get('execution_devices')[0].get('video_memory').get('total').get('value') == 24 + assert metrics_set.get('disks')[0].get('total').get('value') == 500 + assert metrics_set.get('disks')[0].get('free').get('unit') == 'GB' + assert metrics_set.get('disks')[0].get('utilization').get('value') == 60