tests: rename comfyui plugin root folder name

to prevent import name conflict in test scripts

Signed-off-by: Huang, Zeyu <11222265+fhfuih@users.noreply.github.com>

Author: fhfuih

apps/ComfyUI-vLLM-Omni/__init__.py

@@ -1,4 +1,4 @@
-"""Top-level package for vllm_omni."""
+"""Top-level package for comfyui_vllm_omni."""
 
 __all__ = [
     "NODE_CLASS_MAPPINGS",
@@ -10,7 +10,7 @@
 __email__ = "11222265+fhfuih@users.noreply.github.com"
 __version__ = "0.0.1"
 
-from .vllm_omni.nodes import (
+from .comfyui_vllm_omni.nodes import (
     VLLMOmniARSampling,
     VLLMOmniComprehension,
     VLLMOmniDiffusionSampling,

…/ComfyUI-vLLM-Omni/vllm_omni/__init__.py …-vLLM-Omni/comfyui_vllm_omni/__init__.pyapps/ComfyUI-vLLM-Omni/vllm_omni/__init__.py renamed to apps/ComfyUI-vLLM-Omni/comfyui_vllm_omni/__init__.py apps/ComfyUI-vLLM-Omni/vllm_omni/__init__.py renamed to apps/ComfyUI-vLLM-Omni/comfyui_vllm_omni/__init__.py

@@ -1,6 +1,83 @@
+"""
+Conftest for ComfyUI-vLLM-Omni tests.
+
+This module sets up the test environment by:
+1. Adding the ComfyUI plugin to Python path
+2. Mocking comfy_api.input module (AudioInput, VideoInput) since comfyui is not installed
+3. Mocking comfy_extras.nodes_audio module
+"""
+
 import os
 import sys
+from typing import BinaryIO, TypedDict
+from unittest.mock import MagicMock
+
+
+def pytest_configure(config):
+    """
+    Called after command line options have been parsed and before test collection.
+    This is the right place to set up sys.path and mock modules.
+    """
+    _setup_comfyui_test_environment()
+
+
+def _setup_comfyui_test_environment():
+    """Set up the test environment for ComfyUI plugin testing."""
+    # Add ComfyUI plugin path to allow importing comfyui_vllm_omni
+    _COMFYUI_PLUGIN_PATH = os.path.abspath(
+        os.path.join(os.path.dirname(__file__), "..", "..", "apps", "ComfyUI-vLLM-Omni")
+    )
+    if _COMFYUI_PLUGIN_PATH not in sys.path:
+        sys.path.insert(0, _COMFYUI_PLUGIN_PATH)
+
+    # Now we can import torch (which is safe as it doesn't conflict)
+    import torch
+
+    # === Mock comfy_api.input module ===
+    class AudioInput(TypedDict):
+        """Mock AudioInput TypedDict from comfy_api.input"""
+        waveform: torch.Tensor  # Shape: (B, C, T)
+        sample_rate: int
+
+    class VideoInput:
+        """Mock VideoInput class from comfy_api.input"""
+        def __init__(self, data: bytes = b"mock_video_data"):
+            self._data = data
+
+        def save_to(self, file: str | BinaryIO):
+            """Save video data to file or file-like object."""
+            if isinstance(file, str):
+                print("Called VideoInput.save_to with file path. Saving to a path is no-op in tests.")
+            else:
+                file.write(self._data)
+
+    # Create mock comfy_api module hierarchy
+    mock_comfy_api = MagicMock()
+    mock_comfy_api_input = MagicMock()
+    mock_comfy_api_input.AudioInput = AudioInput
+    mock_comfy_api_input.VideoInput = VideoInput
+    mock_comfy_api.input = mock_comfy_api_input
+
+    # Create mock comfy_extras module
+    mock_comfy_extras = MagicMock()
+    mock_nodes_audio = MagicMock()
+
+    def mock_load(_: str | BinaryIO):
+        """Mock nodes_audio.load that returns a waveform tensor and sample rate."""
+        # Return mock audio data: (channels, samples) tensor and sample rate
+        waveform = torch.zeros((1, 24000), dtype=torch.float32)
+        sample_rate = 24000
+        return waveform, sample_rate
+
+    mock_nodes_audio.load = mock_load
+    mock_comfy_extras.nodes_audio = mock_nodes_audio
+
+    # Install mock modules BEFORE importing any comfyui_vllm_omni code
+    sys.modules["comfy_api"] = mock_comfy_api
+    sys.modules["comfy_api.input"] = mock_comfy_api_input
+    sys.modules["comfy_extras"] = mock_comfy_extras
+    sys.modules["comfy_extras.nodes_audio"] = mock_nodes_audio
+
 
-# Add the project root directory to Python path
-# This allows the tests to import the project
-sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..", "apps", "ComfyUI-vLLM-Omni")))
+# Run setup immediately when conftest is imported (before pytest_configure for early loading)
+_setup_comfyui_test_environment()