"""Tests for SurrealDB MCP Server startup and configuration."""
import os
import pytest
import sys
from unittest.mock import patch, MagicMock
from surreal_mcp.server import mcp, main
class TestServerConfiguration:
"""Test server configuration and environment variables."""
def test_mcp_server_initialized(self):
"""Test that MCP server is properly initialized."""
assert mcp is not None
assert mcp.name == "SurrealDB MCP Server"
def test_tools_registered(self):
"""Test that all tools are registered."""
expected_tools = [
"query", "select", "create", "update", "delete",
"merge", "patch", "upsert", "insert", "relate"
]
# For FastMCP, tools are registered synchronously even if decorated async
tools = mcp._tool_manager._tools
registered_tools = list(tools.keys())
for tool in expected_tools:
assert tool in registered_tools, f"Tool '{tool}' not registered"
assert len(registered_tools) == len(expected_tools)
def test_tool_signatures(self):
"""Test that tools have correct signatures."""
tools = mcp._tool_manager._tools
# Test query tool
query_tool = tools["query"]
assert "query_string" in query_tool.fn.__code__.co_varnames
# Test select tool
select_tool = tools["select"]
params = select_tool.fn.__code__.co_varnames
assert "table" in params
assert "id" in params
# Test create tool
create_tool = tools["create"]
params = create_tool.fn.__code__.co_varnames
assert "table" in params
assert "data" in params
# Test relate tool
relate_tool = tools["relate"]
params = relate_tool.fn.__code__.co_varnames
assert "from_thing" in params
assert "relation_name" in params
assert "to_thing" in params
assert "data" in params
class TestEnvironmentValidation:
"""Test environment variable validation."""
def test_missing_env_vars(self):
"""Test server exits when required env vars are missing."""
# Save original env vars
original_env = {}
required_vars = [
"SURREAL_URL", "SURREAL_USER", "SURREAL_PASSWORD",
"SURREAL_NAMESPACE", "SURREAL_DATABASE"
]
for var in required_vars:
original_env[var] = os.environ.get(var)
# Test each missing variable
for missing_var in required_vars:
# Clear all required vars
for var in required_vars:
if var in os.environ:
del os.environ[var]
# Set all except the one we're testing
for var in required_vars:
if var != missing_var:
os.environ[var] = "test_value"
# Import should fail due to missing env var
with pytest.raises(SystemExit):
# We need to reload the module to trigger the env var check
if 'surreal_mcp.server' in sys.modules:
del sys.modules['surreal_mcp.server']
import surreal_mcp.server
# Restore original env vars
for var, value in original_env.items():
if value is not None:
os.environ[var] = value
elif var in os.environ:
del os.environ[var]
def test_env_vars_present(self):
"""Test that all required env vars are set in test environment."""
required_vars = [
"SURREAL_URL", "SURREAL_USER", "SURREAL_PASSWORD",
"SURREAL_NAMESPACE", "SURREAL_DATABASE"
]
for var in required_vars:
assert var in os.environ, f"Missing required env var: {var}"
class TestMainFunction:
"""Test the main entry point function."""
@patch('surreal_mcp.server.mcp.run')
@patch('surreal_mcp.server.close_database_pool')
def test_main_normal_execution(self, mock_close_pool, mock_mcp_run):
"""Test normal execution of main function."""
# Run main
main()
# Verify MCP server was started
mock_mcp_run.assert_called_once()
@patch('surreal_mcp.server.mcp.run')
@patch('surreal_mcp.server.close_database_pool')
def test_main_keyboard_interrupt(self, mock_close_pool, mock_mcp_run):
"""Test handling of keyboard interrupt."""
# Make mcp.run raise KeyboardInterrupt
mock_mcp_run.side_effect = KeyboardInterrupt()
# Run main - should handle the interrupt gracefully
main()
# Verify cleanup was called
mock_close_pool.assert_called_once()
@patch('surreal_mcp.server.mcp.run')
@patch('surreal_mcp.server.close_database_pool')
def test_main_exception_handling(self, mock_close_pool, mock_mcp_run):
"""Test handling of general exceptions."""
# Make mcp.run raise an exception
test_error = Exception("Test error")
mock_mcp_run.side_effect = test_error
# Run main - should raise the exception
with pytest.raises(Exception) as exc_info:
main()
assert str(exc_info.value) == "Test error"
# Verify cleanup was called
mock_close_pool.assert_called_once()
class TestToolDocstrings:
"""Test that all tools have comprehensive docstrings."""
def test_all_tools_have_docstrings(self):
"""Verify all tools have docstrings."""
tools = mcp._tool_manager._tools
for tool_name, tool in tools.items():
assert tool.fn.__doc__ is not None, f"Tool '{tool_name}' missing docstring"
# Check docstring has minimum content
doc = tool.fn.__doc__
assert len(doc) > 100, f"Tool '{tool_name}' has too short docstring"
assert "Args:" in doc, f"Tool '{tool_name}' missing Args section"
assert "Returns:" in doc, f"Tool '{tool_name}' missing Returns section"
assert "Example" in doc, f"Tool '{tool_name}' missing Example section"
