From 1745749cd2745c94c3f852e9c02dfde19d8d9c20 Mon Sep 17 00:00:00 2001
From: Dawid Rycerz <dawid@rycerz.xyz>
Date: Fri, 28 Mar 2025 20:53:41 +0100
Subject: Fix ruff errors and warnings in hello_world service

---
 servers/hello_world/tests/test_integration.py | 74 +++++++++++++++++++++++++++
 1 file changed, 74 insertions(+)
 create mode 100644 servers/hello_world/tests/test_integration.py

(limited to 'servers/hello_world/tests/test_integration.py')

diff --git a/servers/hello_world/tests/test_integration.py b/servers/hello_world/tests/test_integration.py
new file mode 100644
index 0000000..5948dae
--- /dev/null
+++ b/servers/hello_world/tests/test_integration.py
@@ -0,0 +1,74 @@
+import pytest
+import asyncio
+from unittest.mock import AsyncMock, MagicMock, patch
+from io import StringIO
+import json
+from mcp_server_hello_world.server import HelloWorldServer, create_server
+
+class TestIntegration:
+    """Integration tests for the hello_world server."""
+    
+    @pytest.mark.asyncio
+    async def test_hello_world_server(self):
+        """Test the HelloWorldServer class."""
+        # Create a HelloWorldServer instance
+        server = HelloWorldServer()
+        
+        # Test the welcome message
+        assert server.welcome_message == "Welcome to the Hello World MCP Server! This is a simple static resource."
+        
+        # Test the get_greeting method
+        name = "Test User"
+        expected_greeting = f"Hello, {name}! Welcome to the Hello World MCP Server."
+        assert server.get_greeting(name) == expected_greeting
+    
+    @pytest.mark.asyncio
+    async def test_server_initialization(self):
+        """Test server initialization with stdio transport."""
+        # Import the main function here to avoid issues with the mocking
+        from mcp_server_hello_world.server import main
+        
+        # Mock the run_stdio_async method
+        with patch('mcp.server.fastmcp.FastMCP.run_stdio_async') as mock_run_stdio:
+            
+            # Set up the mocks
+            mock_run_stdio.return_value = AsyncMock()
+            
+            # Start the server with stdio transport
+            task = asyncio.create_task(main("stdio", "localhost", 8080))
+            
+            # Wait for the server to start
+            await asyncio.sleep(0.1)
+            
+            # Check that run_stdio_async was called
+            mock_run_stdio.assert_called_once()
+            
+            # Cancel the task
+            task.cancel()
+            try:
+                await task
+            except asyncio.CancelledError:
+                pass
+    
+    @pytest.mark.asyncio
+    async def test_server_functionality(self):
+        """Test the server's functionality."""
+        # Create a server
+        mcp = create_server()
+        
+        # Test the resource
+        resource = await mcp.read_resource("hello://welcome")
+        assert len(resource) == 1
+        assert resource[0].content == "Welcome to the Hello World MCP Server! This is a simple static resource."
+        
+        # Test the tool
+        greeting = await mcp.call_tool("hello", {"name": "Test User"})
+        assert len(greeting) == 1
+        assert greeting[0].text == "Hello, Test User! Welcome to the Hello World MCP Server."
+        
+        # Test the prompt
+        prompt = await mcp.get_prompt("greeting", {"name": "Test User"})
+        assert len(prompt.messages) > 0
+        assert prompt.messages[0].role == "user"
+        assert prompt.messages[0].content.type == "text"
+        assert "Hello there! You've chosen to use the greeting prompt with the name: Test User." in prompt.messages[0].content.text
-- 
cgit v1.2.3