Django Shell MCP Server

from __future__ import annotations import logging import pytest from django.apps import apps from mcp_django.shell.core import DjangoShell from mcp_django.shell.core import ErrorResult from mcp_django.shell.core import StatementResult @pytest.fixture def shell(): shell = DjangoShell() yield shell shell.clear_history() class TestCodeExecution: def test_execute_simple_statement(self, shell): result = shell._execute("x = 5") assert isinstance(result, StatementResult) def test_execute_print_captures_stdout(self, shell): result = shell._execute('print("Hello, World!")') assert isinstance(result, StatementResult) assert result.stdout == "Hello, World!\n" def test_execute_multiline_with_print(self, shell): code = """\ x = 5 y = 10 print(f"Sum: {x + y}") """ result = shell._execute(code) assert isinstance(result, StatementResult) assert result.stdout == "Sum: 15\n" def test_execute_invalid_code_returns_error(self, shell): result = shell._execute("1 / 0") assert isinstance(result, ErrorResult) def test_execute_empty_string_returns_ok(self, shell): result = shell._execute("") assert isinstance(result, StatementResult) def test_execute_whitespace_only_returns_ok(self, shell): result = shell._execute(" \n \t ") assert isinstance(result, StatementResult) @pytest.mark.asyncio async def test_async_execute_returns_result(self): shell = DjangoShell() result = await shell.execute("x = 5") assert isinstance(result, StatementResult) class TestShellState: def test_init_django_setup_completes(self): shell = DjangoShell() assert apps.ready assert shell.history == [] def test_execution_uses_fresh_globals(self, shell): """Verify each execution uses fresh globals (stateless).""" # First execution result = shell._execute("x = 42") assert isinstance(result, StatementResult) # Second execution should NOT have access to 'x' (fresh globals) result = shell._execute("print(x)") assert isinstance(result, ErrorResult) assert isinstance(result.exception, NameError) def test_clear_history_clears_history_only(self, shell): """Verify clear_history clears the execution history.""" shell._execute("x = 42") assert len(shell.history) == 1 shell.clear_history() assert len(shell.history) == 0 class TestLoggingCoverage: @pytest.fixture(autouse=True) def debug_loglevel(self, caplog): caplog.set_level(logging.DEBUG) yield def test_statement_result_with_stdout_and_stderr(self, shell, caplog): code = """ import sys sys.stdout.write("Output message\\n") sys.stderr.write("Error message\\n") x = 42 """ result = shell._execute(code.strip()) assert isinstance(result, StatementResult) assert result.stdout == "Output message\n" assert result.stderr == "Error message\n" assert "StatementResult.stdout: Output message" in caplog.text assert "StatementResult.stderr: Error message" in caplog.text def test_error_result_with_stdout_and_stderr(self, shell, caplog): code = """ import sys sys.stdout.write("Before error\\n") sys.stderr.write("Warning before error\\n") 1 / 0 """ result = shell._execute(code.strip()) assert isinstance(result, ErrorResult) assert result.stdout == "Before error\n" assert result.stderr == "Warning before error\n" assert "ErrorResult.stdout: Before error" in caplog.text assert "ErrorResult.stderr: Warning before error" in caplog.text