From 180e19e92365068a69abb1110dd045b58f1c518e Mon Sep 17 00:00:00 2001 From: Arjan Singh Date: Thu, 28 Aug 2025 18:06:41 -0700 Subject: [PATCH] test: mock torch in cache middleware tests --- tests-unit/server_test/test_cache_control.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests-unit/server_test/test_cache_control.py b/tests-unit/server_test/test_cache_control.py index 011525f78..183490b4d 100644 --- a/tests-unit/server_test/test_cache_control.py +++ b/tests-unit/server_test/test_cache_control.py @@ -12,6 +12,8 @@ with ( patch("app.frontend_management.FrontendManager"), patch("utils.install_util.get_missing_requirements_message"), patch("utils.install_util.requirements_path"), + patch("comfy.model_management.get_torch_device"), + patch("comfy.model_management.get_total_memory", return_value=1024 * 1024 * 1024), ): from server import cache_control, ONE_HOUR, ONE_DAY, IMG_EXTENSIONS