improve tests

Author: shahar4499

examples/simple_integration.py

@@ -2,7 +2,7 @@
 Simple example of using FastAPI-MCP to add an MCP server to a FastAPI app.
 """
 
-from fastapi import FastAPI, Query
+from fastapi import FastAPI, HTTPException, Query
 from pydantic import BaseModel
 from typing import List, Optional
 
@@ -27,7 +27,7 @@ class Item(BaseModel):
 
 
 # In-memory database
-items_db = {}
+items_db: dict[int, Item] = {}
 
 
 # Define some endpoints
@@ -49,7 +49,7 @@ async def read_item(item_id: int):
     Raises a 404 error if the item does not exist.
     """
     if item_id not in items_db:
-        return {"error": "Item not found"}
+        raise HTTPException(status_code=404, detail="Item not found")
     return items_db[item_id]
 
 
@@ -72,7 +72,7 @@ async def update_item(item_id: int, item: Item):
     Raises a 404 error if the item does not exist.
     """
     if item_id not in items_db:
-        return {"error": "Item not found"}
+        raise HTTPException(status_code=404, detail="Item not found")
 
     item.id = item_id
     items_db[item_id] = item
@@ -87,7 +87,7 @@ async def delete_item(item_id: int):
     Raises a 404 error if the item does not exist.
     """
     if item_id not in items_db:
-        return {"error": "Item not found"}
+        raise HTTPException(status_code=404, detail="Item not found")
 
     del items_db[item_id]
     return {"message": "Item deleted successfully"}

test_mcp_tools.py renamed to tests/manual_test_mcp_tools.py

@@ -4,6 +4,8 @@
 This script connects to the MCP server, initializes a session, and requests a list of available tools.
 """
 
+# TODO: Turn this into a pytest test
+
 import json
 import sys
 import asyncio

tests/test_http_tools.py

@@ -6,7 +6,7 @@
 import pytest
 from fastapi import FastAPI, Query, Path, Body
 from pydantic import BaseModel
-from typing import List, Optional, Dict, Any
+from typing import List, Optional
 
 from fastapi_mcp import add_mcp_server
 from fastapi_mcp.http_tools import (