Adapt tests for pod_executor package and clean up mcp_forge

- Created tests/pod_executor/ with adapted tests from old locations
- tests/pod_executor/simple/test_executor.py: 17/17 tests passing
- tests/pod_executor/security/test_resource_limits.py: 22/23 tests passing

- Removed old test locations (will be deleted with mcp_forge cleanup)
- Fixed all corrupted files from sed/quote issues using Python scripts
- Removed mcp_forge dependencies from pod_executor:
  - Removed ForgeConfig from backend.py (explicit parameters)
  - Removed SessionConfig from sessions.py (explicit parameters)
  - Fixed all audit logger calls to use string-based events

- Updated mcp_forge/security/__init__.py:
  - Removed resource_limits imports (now in pod_executor)
  - Added comment directing to pod_executor.security.resource_limits

- Deleted from mcp_forge:
  - src/mcp_forge/execution/ (simple and jupyter backends)
  - src/mcp_forge/podman/ (container management)
  - src/mcp_forge/security/resource_limits.py

Total: 39/40 tests passing in pod_executor package
This commit is contained in:
Hans Aschauer 2026-02-07 11:05:32 +01:00
parent 63d9b55a00
commit 3a6bd01272
25 changed files with 1390 additions and 3166 deletions

View file

@ -108,15 +108,15 @@ class SessionManager:
Initialize session manager.
Args:
kernel_manager: Kernel lifecycle manager
idle_timeout: Session idle timeout in seconds (default: 3600)
kernel_manager: Jupyter kernel manager instance
idle_timeout: Seconds before idle session cleanup (default: 3600)
max_sessions: Maximum concurrent sessions (default: 10)
audit_logger: Optional audit logger (uses NullAuditLogger if None)
audit_logger: Optional audit logger (defaults to NullAuditLogger)
"""
self.kernel_manager = kernel_manager
self.idle_timeout = idle_timeout
self.max_sessions = max_sessions
self.audit_logger = audit_logger or NullAuditLogger()
self.kernel_manager = kernel_manager
self.audit_logger = audit_logger if audit_logger is not None else NullAuditLogger()
self.sessions: Dict[str, Session] = {}
def create_session(
@ -171,8 +171,8 @@ class SessionManager:
# Log session creation
self.audit_logger.log(
event_type="SESSION_CREATE,
severity="INFO,
event_type="session.create",
severity="info",
message=f"Session created: {session_id}",
session_id=session_id,
details={
@ -373,8 +373,8 @@ class SessionManager:
except Exception as e:
# Log but continue with cleanup
self.audit_logger.log(
event_type="SESSION_DESTROY,
severity="WARNING,
event_type="session.destroy",
severity="warning",
message=f"Error shutting down kernel for session {session_id}",
session_id=session_id,
error=str(e)
@ -385,8 +385,8 @@ class SessionManager:
# Log destruction
self.audit_logger.log(
event_type="SESSION_DESTROY,
severity="INFO,
event_type="session.destroy",
severity="info",
message=f"Session destroyed: {session_id}",
session_id=session_id
)