mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-15 03:27:24 +08:00
31 lines
1021 B
Python
31 lines
1021 B
Python
import pytest
|
|
|
|
import folder_paths
|
|
|
|
|
|
def test_set_input_directory_creates_missing_directory(tmp_path):
|
|
original_input_directory = folder_paths.get_input_directory()
|
|
custom_input_directory = tmp_path / "custom-input"
|
|
|
|
try:
|
|
folder_paths.set_input_directory(str(custom_input_directory))
|
|
|
|
assert custom_input_directory.is_dir()
|
|
finally:
|
|
folder_paths.set_input_directory(original_input_directory)
|
|
|
|
|
|
def test_set_input_directory_keeps_original_when_creation_fails(tmp_path, monkeypatch):
|
|
original_input_directory = folder_paths.get_input_directory()
|
|
custom_input_directory = tmp_path / "custom-input"
|
|
|
|
def fail_to_create_directory(path, exist_ok=False):
|
|
raise OSError("create failed")
|
|
|
|
monkeypatch.setattr(folder_paths.os, "makedirs", fail_to_create_directory)
|
|
|
|
with pytest.raises(OSError, match="create failed"):
|
|
folder_paths.set_input_directory(str(custom_input_directory))
|
|
|
|
assert folder_paths.get_input_directory() == original_input_directory
|