ComfyUI/tests/isolation/test_init.py

"""Unit tests for PyIsolate isolation system initialization."""



def test_log_prefix():
    """Verify LOG_PREFIX constant is correctly defined."""
    from comfy.isolation import LOG_PREFIX
    assert LOG_PREFIX == "]["
    assert isinstance(LOG_PREFIX, str)


def test_module_initialization():
    """Verify module initializes without errors."""
    import comfy.isolation
    assert hasattr(comfy.isolation, 'LOG_PREFIX')
    assert hasattr(comfy.isolation, 'initialize_proxies')


class TestInitializeProxies:
    def test_initialize_proxies_runs_without_error(self):
        from comfy.isolation import initialize_proxies
        initialize_proxies()

    def test_initialize_proxies_registers_folder_paths_proxy(self):
        from comfy.isolation import initialize_proxies
        from comfy.isolation.proxies.folder_paths_proxy import FolderPathsProxy
        initialize_proxies()
        proxy = FolderPathsProxy()
        assert proxy is not None
        assert hasattr(proxy, "get_temp_directory")

    def test_initialize_proxies_registers_model_management_proxy(self):
        from comfy.isolation import initialize_proxies
        from comfy.isolation.proxies.model_management_proxy import ModelManagementProxy
        initialize_proxies()
        proxy = ModelManagementProxy()
        assert proxy is not None
        assert hasattr(proxy, "get_torch_device")

    def test_initialize_proxies_can_be_called_multiple_times(self):
        from comfy.isolation import initialize_proxies
        initialize_proxies()
        initialize_proxies()
        initialize_proxies()

    def test_dev_proxies_accessible_when_dev_mode(self, monkeypatch):
        """Verify dev mode does not break core proxy initialization."""
        monkeypatch.setenv("PYISOLATE_DEV", "1")
        from comfy.isolation import initialize_proxies
        from comfy.isolation.proxies.folder_paths_proxy import FolderPathsProxy
        from comfy.isolation.proxies.utils_proxy import UtilsProxy
        initialize_proxies()
        folder_proxy = FolderPathsProxy()
        utils_proxy = UtilsProxy()
        assert folder_proxy is not None
        assert utils_proxy is not None