[tests] Add API test suite

tests-api/mocks/__init__.py

@@ -0,0 +1 @@
+# Make tests-api/mocks directory a proper package

tests-api/mocks/custom_node_manager.py

@@ -0,0 +1,26 @@
+"""
+Mock CustomNodeManager for testing purposes
+"""
+
+class CustomNodeManager:
+    """
+    Mock implementation of the CustomNodeManager class
+    """
+    instance = None
+    
+    def __init__(self):
+        self.custom_nodes = {}
+        self.node_paths = []
+        self.refresh_timeout = None
+        
+    def get_node_path(self, node_class):
+        """
+        Mock implementation to get the path for a node class
+        """
+        return self.custom_nodes.get(node_class, None)
+        
+    def update_node_paths(self):
+        """
+        Mock implementation to update node paths
+        """
+        pass

tests-api/mocks/patch.py

@@ -0,0 +1,116 @@
+"""
+Patch module to mock imports for testing
+"""
+import sys
+import importlib.util
+import os
+from pathlib import Path
+
+# Import mock modules
+from mocks.prompt_server import PromptServer
+from mocks.custom_node_manager import CustomNodeManager
+
+# Current directory 
+current_dir = Path(__file__).parent.parent  # tests-api directory
+
+# Define mocks
+class MockModule:
+    """Base class for mock modules"""
+    pass
+
+# Create server mock module with PromptServer
+server_mock = MockModule()
+server_mock.PromptServer = PromptServer
+prompt_server_instance = PromptServer()
+server_mock.PromptServer.instance = prompt_server_instance
+server_mock.PromptServer.inst = prompt_server_instance
+
+# Create app mock module with custom_node_manager submodule
+app_mock = MockModule()
+app_custom_node_manager = MockModule()
+app_custom_node_manager.CustomNodeManager = CustomNodeManager
+app_custom_node_manager.CustomNodeManager.instance = CustomNodeManager()
+
+# Create utils mock module with json_util submodule
+utils_mock = MockModule()
+utils_json_util = MockModule()
+
+# Create utils.validation and utils.schema_utils submodules
+utils_validation = MockModule()
+utils_schema_utils = MockModule()
+
+# Import actual modules (make sure path is set up correctly)
+sys.path.insert(0, str(current_dir))
+
+try:
+    # Import the validation module
+    from utils.validation import load_openapi_spec
+    utils_validation.load_openapi_spec = load_openapi_spec
+    
+    # Import all schema_utils functions
+    from utils.schema_utils import (
+        get_all_paths, 
+        get_grouped_paths,
+        get_methods_for_path,
+        find_paths_with_security,
+        get_content_types_for_response,
+        get_required_parameters
+    )
+    
+    utils_schema_utils.get_all_paths = get_all_paths
+    utils_schema_utils.get_grouped_paths = get_grouped_paths
+    utils_schema_utils.get_methods_for_path = get_methods_for_path
+    utils_schema_utils.find_paths_with_security = find_paths_with_security
+    utils_schema_utils.get_content_types_for_response = get_content_types_for_response
+    utils_schema_utils.get_required_parameters = get_required_parameters
+    
+except ImportError as e:
+    print(f"Error importing test utilities: {e}")
+    # Define dummy functions if imports fail
+    def dummy_load_openapi_spec():
+        """Dummy function for testing"""
+        return {"paths": {}}
+    utils_validation.load_openapi_spec = dummy_load_openapi_spec
+    
+    def dummy_get_all_paths(spec):
+        return list(spec.get("paths", {}).keys())
+    utils_schema_utils.get_all_paths = dummy_get_all_paths
+    
+    def dummy_get_grouped_paths(spec):
+        return {}
+    utils_schema_utils.get_grouped_paths = dummy_get_grouped_paths
+    
+    def dummy_get_methods_for_path(spec, path):
+        return []
+    utils_schema_utils.get_methods_for_path = dummy_get_methods_for_path
+    
+    def dummy_find_paths_with_security(spec, security_scheme=None):
+        return []
+    utils_schema_utils.find_paths_with_security = dummy_find_paths_with_security
+    
+    def dummy_get_content_types_for_response(spec, path, method, status_code="200"):
+        return []
+    utils_schema_utils.get_content_types_for_response = dummy_get_content_types_for_response
+    
+    def dummy_get_required_parameters(spec, path, method):
+        return []
+    utils_schema_utils.get_required_parameters = dummy_get_required_parameters
+
+# Add merge_json_recursive from our mock utils
+from mocks.utils import merge_json_recursive
+utils_json_util.merge_json_recursive = merge_json_recursive
+
+# Apply the mocks to sys.modules
+def apply_mocks():
+    """Apply all mocks to sys.modules"""
+    sys.modules['server'] = server_mock
+    sys.modules['app'] = app_mock
+    sys.modules['app.custom_node_manager'] = app_custom_node_manager
+    sys.modules['utils'] = utils_mock
+    sys.modules['utils.json_util'] = utils_json_util
+    sys.modules['utils.validation'] = utils_validation
+    sys.modules['utils.schema_utils'] = utils_schema_utils
+    
+    # Make sure our actual utils module is importable
+    if current_dir not in sys.path:
+        sys.path.insert(0, str(current_dir))

tests-api/mocks/prompt_server.py

@@ -0,0 +1,71 @@
+"""
+Mock PromptServer for testing purposes
+"""
+
+class MockRoutes:
+    """
+    Mock routing class with method decorators
+    """
+    def __init__(self):
+        self.routes = {}
+        
+    def get(self, path):
+        """Decorator for GET routes"""
+        def decorator(f):
+            self.routes[('GET', path)] = f
+            return f
+        return decorator
+        
+    def post(self, path):
+        """Decorator for POST routes"""
+        def decorator(f):
+            self.routes[('POST', path)] = f
+            return f
+        return decorator
+        
+    def put(self, path):
+        """Decorator for PUT routes"""
+        def decorator(f):
+            self.routes[('PUT', path)] = f
+            return f
+        return decorator
+        
+    def delete(self, path):
+        """Decorator for DELETE routes"""
+        def decorator(f):
+            self.routes[('DELETE', path)] = f
+            return f
+        return decorator
+
+
+class PromptServer:
+    """
+    Mock implementation of the PromptServer class
+    """
+    instance = None
+    inst = None
+    
+    def __init__(self):
+        self.routes = MockRoutes()
+        self.registered_paths = set()
+        self.base_url = "http://127.0.0.1:8188"  # Assuming server is running on default port
+        self.queue_lock = None
+        
+    def add_route(self, method, path, handler, *args, **kwargs):
+        """
+        Add a mock route to the server
+        """
+        self.routes.routes[(method.upper(), path)] = handler
+        self.registered_paths.add(path)
+        
+    async def send_msg(self, message, data=None):
+        """
+        Mock send_msg method (does nothing in the mock)
+        """
+        pass
+        
+    def send_sync(self, message, data=None):
+        """
+        Mock send_sync method (does nothing in the mock)
+        """
+        pass

tests-api/mocks/utils.py

@@ -0,0 +1,20 @@
+"""
+Mock utils module for testing purposes
+"""
+
+def merge_json_recursive(a, b):
+    """
+    Mock implementation of merge_json_recursive
+    """
+    if isinstance(a, dict) and isinstance(b, dict):
+        result = a.copy()
+        for key, value in b.items():
+            if key in result and isinstance(result[key], (dict, list)) and isinstance(value, (dict, list)):
+                result[key] = merge_json_recursive(result[key], value)
+            else:
+                result[key] = value
+        return result
+    elif isinstance(a, list) and isinstance(b, list):
+        return a + b
+    else:
+        return b