unit test fixes and logo

Author: mrachidi

README.md

@@ -1,3 +1,8 @@
+<p align="center">
+  <img src="logo.png" alt="Vibe tester Logo" width="270"/>
+</p>
+
+
 # MCP Vulnerability Checker Server
 
 A modular Model Context Protocol (MCP) server providing comprehensive security vulnerability intelligence tools including CVE lookup, EPSS scoring, CVSS calculation, exploit detection, and Python package vulnerability checking.

logo.png

@@ -46,5 +46,19 @@ ignore = []
 line-length = 88
 target-version = "py310"
 
+[tool.pytest.ini_options]
+asyncio_mode = "auto"
+asyncio_default_fixture_loop_scope = "function"
+testpaths = ["tests"]
+python_files = ["test_*.py"]
+python_classes = ["Test*"]
+python_functions = ["test_*"]
+addopts = ["-v", "--tb=short", "--strict-markers"]
+markers = [
+    "asyncio: marks tests as async (defers to pytest-asyncio)",
+    "integration: marks tests as integration tests",
+    "unit: marks tests as unit tests"
+]
+
 [tool.uv]
 dev-dependencies = ["pyright>=1.1.378", "pytest>=8.3.3", "ruff>=0.6.9", "pytest-asyncio>=0.23.5"]

pyproject.toml

@@ -1,7 +1,11 @@
 #!/usr/bin/env python3
-"""Comprehensive test script for all vulnerability intelligence tools"""
+"""
+Comprehensive test suite for all vulnerability intelligence tools
+Tests the complete MCP server toolkit with real data
+"""
 
 import asyncio
+import pytest
 import sys
 from pathlib import Path
 
@@ -15,9 +19,15 @@
 from mcp_simple_tool.tools.exploit_availability import get_exploit_availability
 from mcp_simple_tool.tools.vulnerability_timeline import get_vulnerability_timeline
 from mcp_simple_tool.tools.vex_status import get_vex_status
+from mcp_simple_tool.tools.package_vulnerability import check_package_vulnerabilities
 
+
+@pytest.mark.asyncio
 async def test_all_tools():
-    """Test all vulnerability intelligence tools with Log4Shell CVE"""
+    """
+    Comprehensive test of all vulnerability intelligence tools.
+    Tests with CVE-2021-44228 (Log4Shell) - a well-documented critical vulnerability.
+    """
 
     cve_id = "CVE-2021-44228"  # Log4Shell - well-known vulnerability
     print(f"🔍 **Comprehensive Vulnerability Intelligence Report for {cve_id}**\n")

tests/test_all_tools.py

@@ -2,6 +2,7 @@
 """Test script for CVSS calculator tool"""
 
 import asyncio
+import pytest
 import sys
 from pathlib import Path
 
@@ -10,8 +11,9 @@
 
 from mcp_simple_tool.tools.cvss_calculator import calculate_cvss_score
 
+@pytest.mark.asyncio
 async def test_cvss():
-    """Test the CVSS calculator with different vector strings"""
+    """Test the CVSS calculator tool with different vector strings"""
     # Test cases with different severity levels
     test_vectors = [
         ("CVSS:3.1/AV:N/AC:L/PR:N/UI:N/S:U/C:H/I:H/A:H", "Critical - Network accessible, no privileges"),

tests/test_cvss.py

@@ -2,6 +2,7 @@
 """Test script for EPSS lookup tool"""
 
 import asyncio
+import pytest
 import sys
 from pathlib import Path
 
@@ -10,6 +11,7 @@
 
 from mcp_simple_tool.tools.epss_lookup import get_epss_score
 
+@pytest.mark.asyncio
 async def test_epss():
     """Test the EPSS tool with Log4Shell CVE"""
     print("Testing EPSS tool with CVE-2021-44228 (Log4Shell)...")

tests/test_epss.py

@@ -2,6 +2,7 @@
 """Test script for exploit availability tool"""
 
 import asyncio
+import pytest
 import sys
 from pathlib import Path
 
@@ -10,8 +11,9 @@
 
 from mcp_simple_tool.tools.exploit_availability import get_exploit_availability
 
+@pytest.mark.asyncio
 async def test_exploit():
-    """Test the exploit availability tool with a well-known CVE"""
+    """Test the exploit availability checker tool"""
     print("Testing exploit availability checker with CVE-2021-44228 (Log4Shell)...")
     result = await get_exploit_availability('CVE-2021-44228')
     print("\n" + "="*60)

tests/test_exploit.py

@@ -4,6 +4,7 @@
 """
 
 import asyncio
+import pytest
 import sys
 import os
 
@@ -23,6 +24,7 @@
 )
 
 
+@pytest.mark.asyncio
 async def test_imports():
     """Test that all imports work correctly."""
 

tests/test_modular_server.py

@@ -4,15 +4,18 @@
 """
 
 import asyncio
+import pytest
 import sys
 import os
+from pathlib import Path
 
 # Add parent directory to path so we can import mcp_simple_tool
-sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+sys.path.append(str(Path(__file__).parent.parent))
 
 from mcp_simple_tool.tools.package_vulnerability import check_package_vulnerabilities
 
 
+@pytest.mark.asyncio
 async def test_package_vulnerability():
     """Test the package vulnerability checking functionality."""
 

tests/test_package_vulnerability.py

@@ -2,6 +2,7 @@
 """Test script for vulnerability search tool"""
 
 import asyncio
+import pytest
 import sys
 from pathlib import Path
 
@@ -10,6 +11,7 @@
 
 from mcp_simple_tool.tools.vulnerability_search import search_vulnerabilities
 
+@pytest.mark.asyncio
 async def test_search():
     """Test the vulnerability search tool with different parameters"""