Add test.

tests-unit/main_test.py

@@ -0,0 +1,46 @@
+import pytest
+from unittest.mock import patch, mock_open
+import os
+
+from main import load_extra_path_config
+
+@pytest.fixture
+def mock_add_path():
+    with patch('folder_paths.add_model_folder_path') as mock:
+        yield mock
+
+@pytest.mark.parametrize("os_type,base_path,expected_paths", [
+    ("windows", "~\\AppData\\Local\\ComfyUI", {
+        'checkpoints': os.path.expanduser('~\\AppData\\Local\\ComfyUI\\models\\checkpoints'),
+        'clip': os.path.expanduser('~\\AppData\\Local\\ComfyUI\\models\\clip')
+    }),
+    ("linux", "~/comfyui", {
+        'checkpoints': os.path.expanduser('~/comfyui/models/checkpoints'),
+        'clip': os.path.expanduser('~/comfyui/models/clip')
+    }),
+    ("mac", "~/Library/Application Support/ComfyUI", {
+        'checkpoints': os.path.expanduser('~/Library/Application Support/ComfyUI/models/checkpoints'),
+        'clip': os.path.expanduser('~/Library/Application Support/ComfyUI/models/clip')
+    }),
+])
+def test_load_config_with_base_path(mock_add_path, os_type, base_path, expected_paths, monkeypatch):
+    if os_type == "windows":
+        monkeypatch.setattr(os, 'name', 'nt')
+        monkeypatch.setattr(os.path, 'sep', '\\')
+    else:
+        monkeypatch.setattr(os, 'name', 'posix')
+        monkeypatch.setattr(os.path, 'sep', '/')
+
+    yaml_content = f"""
+    comfyui:
+        base_path: "{base_path}"
+        checkpoints: models{os.path.sep}checkpoints{os.path.sep}
+        clip: models{os.path.sep}clip{os.path.sep}
+    """
+    mock_open_file = mock_open(read_data=yaml_content)
+
+    with patch('builtins.open', mock_open_file):
+        load_extra_path_config('dummy_path.yaml')
+
+    for model_type, expected_path in expected_paths.items():
+        mock_add_path.assert_any_call(model_type, expected_path)