Tool_Agent/tools/local/runcomfy_stop_env/tests/test_stop.py

#!/usr/bin/env python3
"""Test RunComfy Stop Service"""
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).parent.parent))

from stop_service import stop_server

def test_stop_server():
    """Test stop server functionality"""
    test_server_id = "test-server-12345"
    
    print(f"Testing stop server: {test_server_id}")
    result = stop_server(test_server_id)
    
    print(f"\nResult:")
    print(f"  server_id: {result['server_id']}")
    print(f"  status: {result['status']}")
    print(f"  message: {result['message']}")
    
    assert "server_id" in result
    assert "status" in result
    assert "message" in result
    
    print("\nTest passed: Response format is correct")

if __name__ == "__main__":
    test_stop_server()