mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-04-30 23:02:50 +00:00
7780565f1b
- Remove unused pytest import from test_stdio_config.py - Strip file extension from base_command for Windows compatibility (e.g., python.exe -> python) using os.path.splitext Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
"""Tests for MCPServerStdio allowed_commands config integration."""
|
|
|
|
from crewai.mcp.config import MCPServerStdio
|
|
from crewai.mcp.transports.stdio import DEFAULT_ALLOWED_COMMANDS
|
|
|
|
|
|
class TestMCPServerStdioConfig:
|
|
"""Tests for the allowed_commands field on MCPServerStdio."""
|
|
|
|
def test_default_allowed_commands(self):
|
|
"""MCPServerStdio should default to DEFAULT_ALLOWED_COMMANDS."""
|
|
config = MCPServerStdio(command="python", args=["server.py"])
|
|
assert config.allowed_commands == DEFAULT_ALLOWED_COMMANDS
|
|
|
|
def test_custom_allowed_commands(self):
|
|
"""Users can override allowed_commands in config."""
|
|
custom = frozenset({"my-runtime"})
|
|
config = MCPServerStdio(
|
|
command="my-runtime", args=[], allowed_commands=custom
|
|
)
|
|
assert config.allowed_commands == custom
|
|
|
|
def test_none_allowed_commands(self):
|
|
"""Users can disable the allowlist via config."""
|
|
config = MCPServerStdio(
|
|
command="anything", args=[], allowed_commands=None
|
|
)
|
|
assert config.allowed_commands is None
|