Adding unit tests

Author: amri369

tests/mcp/helpers.py

@@ -123,26 +123,28 @@ async def get_prompt(
         return GetPromptResult(description=f"Fake prompt: {name}", messages=[message])
 
     async def list_resources(self, run_context=None, agent=None) -> ListResourcesResult:
-        """Return empty list of resources for fake server"""
+        """Return empty list of resources for the fake server"""
         return ListResourcesResult(resources=self.resources)
 
     async def list_resource_templates(self, run_context=None, agent=None) \
             -> ListResourceTemplatesResult:
-        """Return empty list of resources templates for fake server"""
+        """Return empty list of resources templates for the fake server"""
         return ListResourceTemplatesResult(resourceTemplates=self.resources_templates)
 
     async def read_resource(self, uri: AnyUrl) -> ReadResourceResult:
-        """Return a fake resource read for fake server"""
+        """Return a fake resource read for the fake server"""
         for resource in self.resources:
             if resource.uri == uri:
                 return ReadResourceResult(**resource.model_dump(), contents=[])
 
         raise KeyError
 
     def add_resource(self, uri: AnyUrl, name: str, description: str | None = None):
+        """Add a resource to the fake server"""
         self.resources.append(Resource(uri=uri, description=description, name=name))
 
     def add_resource_template(self, uri: str, name: str, description: str | None = None):
+        """Add a resource template to the fake server"""
         self.resources_templates.append(
             ResourceTemplate(uriTemplate=uri, description=description, name=name)
         )

tests/mcp/test_resources_server.py

@@ -34,3 +34,12 @@ async def test_read_resource():
     server.add_resource(AnyUrl("docs://api/reference"), name="Docs Search")
 
     await server.read_resource(AnyUrl("docs://api/reference"))
+
+@pytest.mark.asyncio
+async def test_read_resource_not_found():
+    """Test getting a resource that doesn't exist"""
+    server = FakeMCPServer()
+
+    uri = AnyUrl("docs://api/reference")
+    with pytest.raises(ValueError, match=f"Resource {uri} not found"):
+        await server.read_resource(uri)