update tests

Author: fayedraza1234

auth_platform/docker-compose.yml

@@ -19,11 +19,13 @@ services:
       - mcp-server
 
   mcp-server:
-    build: ../mcp_server
+    build:
+      context: ../mcp_server
+      target: development
     ports:
       - "8001:8001"
     volumes:
-      - ../mcp_server:/app
+      - ../mcp_server:/workspace/mcp_server
     environment:
       - MCP_PORT=8001
       - LOG_LEVEL=INFO
@@ -34,3 +36,16 @@ services:
       - BAML_ENABLED=false
       - ALERT_CONSOLIDATION_WINDOW_MINUTES=5
       - MAX_EVENTS_PER_ALERT=10
+
+  # Test service for running unit tests
+  mcp-server-tests:
+    build:
+      context: ../mcp_server
+      target: test
+    volumes:
+      - ../mcp_server:/workspace/mcp_server
+    environment:
+      - PYTHONPATH=/workspace
+    profiles:
+      - test
+    command: ["python", "-m", "pytest", "mcp_server/tests/unit/", "-v"]

mcp_server/Dockerfile

@@ -1,18 +1,67 @@
-# Use Python 3.11 slim image as base
-FROM python:3.11-slim
+# Multi-stage Dockerfile for MCP Server
+FROM python:3.11-slim as base
 
-# Set working directory to parent directory
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+# Set working directory
 WORKDIR /workspace
 
+# Copy pyproject.toml first for better caching
+COPY pyproject.toml ./mcp_server/
+
+# Development stage - includes test dependencies
+FROM base as development
+
+# Install the package with all dependencies (including test and dev dependencies)
+RUN cd mcp_server && pip install --no-cache-dir -e .[test,dev]
+
 # Copy the entire mcp_server directory
 COPY . ./mcp_server/
 
-# Install dependencies
-RUN pip install --no-cache-dir fastapi==0.104.1 uvicorn==0.24.0 httpx==0.25.1 sqlalchemy==2.0.23 pydantic==2.8.2 pydantic-settings==2.4.0 python-dotenv==1.0.0
+# Set PYTHONPATH to ensure imports work correctly
+ENV PYTHONPATH=/workspace
 
 # Expose port 8001
 EXPOSE 8001
 
-# Set PYTHONPATH and run the application
+# Health check
+HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:8001/health || exit 1
+
+# Default command to run the application
+CMD ["uvicorn", "mcp_server.main:app", "--host", "0.0.0.0", "--port", "8001"]
+
+# Production stage - minimal dependencies
+FROM base as production
+
+# Install only production dependencies
+RUN cd mcp_server && pip install --no-cache-dir -e .
+
+# Copy the entire mcp_server directory
+COPY . ./mcp_server/
+
+# Set PYTHONPATH to ensure imports work correctly
 ENV PYTHONPATH=/workspace
+
+# Create a non-root user for security
+RUN useradd --create-home --shell /bin/bash app
+USER app
+
+# Expose port 8001
+EXPOSE 8001
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:8001/health || exit 1
+
+# Default command to run the application
 CMD ["uvicorn", "mcp_server.main:app", "--host", "0.0.0.0", "--port", "8001"]
+
+# Test stage - for running tests in CI
+FROM development as test
+
+# Run tests by default
+CMD ["python", "-m", "pytest", "mcp_server/tests/", "-v"]