From 0c509600a40b44e23d5dadab96a3fd279a1b53dc Mon Sep 17 00:00:00 2001 From: silveroxides Date: Tue, 7 Jul 2026 13:29:39 +0200 Subject: [PATCH] Updated the test case in tests-unit/comfy_test/folder_path_test.py to prevent any module-level state mutation as suggested by CodeRabbit review. --- tests-unit/comfy_test/folder_path_test.py | 19 ++++++++++++------- 1 file changed, 12 insertions(+), 7 deletions(-) diff --git a/tests-unit/comfy_test/folder_path_test.py b/tests-unit/comfy_test/folder_path_test.py index 6ee931902..eaef7c127 100644 --- a/tests-unit/comfy_test/folder_path_test.py +++ b/tests-unit/comfy_test/folder_path_test.py @@ -166,15 +166,20 @@ def test_base_path_change_clears_old(set_base_dir): def test_models_directory_cli_and_getters(temp_dir): - with patch.object(sys, 'argv', ["main.py", "--models-directory", temp_dir]): - reload(comfy.cli_args) - reload(folder_paths) + try: + with patch.object(sys, 'argv', ["main.py", "--models-directory", temp_dir]): + reload(comfy.cli_args) + reload(folder_paths) - assert folder_paths.get_models_directory() == os.path.abspath(temp_dir) - assert folder_paths.models_dir == os.path.abspath(temp_dir) + assert folder_paths.get_models_directory() == os.path.abspath(temp_dir) + assert folder_paths.models_dir == os.path.abspath(temp_dir) - with pytest.raises(Exception): - comfy.cli_args.is_valid_directory(os.path.join(temp_dir, "non_existent_folder_path")) + with pytest.raises(Exception): + comfy.cli_args.is_valid_directory(os.path.join(temp_dir, "non_existent_folder_path")) + finally: + with patch.object(sys, 'argv', ["main.py"]): + reload(comfy.cli_args) + reload(folder_paths) def test_models_dir_getter_setter():