test: add logs

Author: radugheo

testcases/ground-to-cloud/test.py

@@ -1,76 +1,84 @@
-import asyncio
-import os
-import sys
-
-from dotenv import load_dotenv
-from mcp.client.session import ClientSession
-from mcp.client.streamable_http import streamablehttp_client
-from retry import retry
-
-
-def get_required_env_var(name: str) -> str:
-    """Get required environment variable or raise an error if not set."""
-    value = os.getenv(name)
-    if not value:
-        raise ValueError(f"Required environment variable {name} is not set")
-    return value
-
-@retry(tries=3, delay=2, backoff=2)
-async def call_add_tool():
-    # Load configuration from environment variables
-    base_url = get_required_env_var("BASE_URL")
-    folder_key = get_required_env_var("UIPATH_FOLDER_KEY")
-    token = get_required_env_var("UIPATH_ACCESS_TOKEN")
-    mcp_server_name = get_required_env_var("MCP_SERVER_NAME")
-    
-    # Construct the MCP server URL
-    mcp_url = f"{base_url}/agenthub_/mcp/{folder_key}/{mcp_server_name}"
-    
-    try:
-        # Use streamable HTTP client to connect to the MCP server
-        async with streamablehttp_client(mcp_url, headers={ 'Authorization': f'Bearer {token}' }) as (read_stream, write_stream, _):
-            async with ClientSession(read_stream, write_stream) as session:
-                # Initialize the session
-                await session.initialize()
-                
-                # List available tools
-                tools_result = await session.list_tools()
-                available_tools = [tool.name for tool in tools_result.tools]
-                expected_available_tools = [
-                    "add", "subtract", "multiply", "divide", "power", "square_root", "nth_root", 
-                    "sin", "cos", "tan", "log10", "natural_log", "log_base", "mean", "median", "standard_deviation", 
-                    "complex_add", "complex_multiply", "convert_temperature", "solve_quadratic", "get_constants" 
-                ]
-
-                print (f"Available tools: {available_tools}")
-
-                if set(available_tools) != set(expected_available_tools):
-                    raise AssertionError(f"Tool sets don't match. Expected: {set(expected_available_tools)}, Got: {set(available_tools)}")
-
-                # Call the add tool directly
-                call_tool_result = await session.call_tool(name="add", arguments={"a": 7, "b": 5})
-
-                expected_result = "12.0"
-                actual_result = call_tool_result.content[0].text if call_tool_result.content else None
-
-                if actual_result != expected_result:
-                    raise AssertionError(f"Expected {expected_result}, got {actual_result}")
-
-                print("Test completed successfully")
-    except Exception as e:
-        print(f"Unexpected error connecting to MCP server: {e}")
-        raise AssertionError(f"Connection error, {e}") from e
-
-async def main():
-    """Main async function to run the test."""
-    try:
-        load_dotenv()
-
-        await call_add_tool()
-    except Exception as e:
-        print(f"Test failed with error: {e}")
-        sys.exit(1)
-
-
-if __name__ == '__main__':
-    asyncio.run(main())
+import asyncio
+import os
+import sys
+
+from dotenv import load_dotenv
+from mcp.client.session import ClientSession
+from mcp.client.streamable_http import streamablehttp_client
+from retry import retry
+
+
+def get_required_env_var(name: str) -> str:
+    """Get required environment variable or raise an error if not set."""
+    value = os.getenv(name)
+    if not value:
+        raise ValueError(f"Required environment variable {name} is not set")
+    return value
+
+@retry(tries=3, delay=2, backoff=2)
+async def call_add_tool():
+    # Load configuration from environment variables
+    base_url = get_required_env_var("BASE_URL")
+    folder_key = get_required_env_var("UIPATH_FOLDER_KEY")
+    token = get_required_env_var("UIPATH_ACCESS_TOKEN")
+    mcp_server_name = get_required_env_var("MCP_SERVER_NAME")
+
+    # Construct the MCP server URL
+    mcp_url = f"{base_url}/agenthub_/mcp/{folder_key}/{mcp_server_name}"
+
+    try:
+        # Use streamable HTTP client to connect to the MCP server
+        async with streamablehttp_client(mcp_url, headers={ 'Authorization': f'Bearer {token}' }) as (read_stream, write_stream, _):
+            async with ClientSession(read_stream, write_stream) as session:
+                # Initialize the session
+                await session.initialize()
+
+                # List available tools
+                try:
+                    tools_result = await session.list_tools()
+                    available_tools = [tool.name for tool in tools_result.tools]
+                    expected_available_tools = [
+                        "add", "subtract", "multiply", "divide", "power", "square_root", "nth_root",
+                        "sin", "cos", "tan", "log10", "natural_log", "log_base", "mean", "median", "standard_deviation",
+                        "complex_add", "complex_multiply", "convert_temperature", "solve_quadratic", "get_constants"
+                    ]
+
+                    print (f"Available tools: {available_tools}")
+                except Exception as e:
+                    print(f"[Tools] Error listing tools: {e}")
+                    raise AssertionError(f"Error listing tools: {e}") from e
+
+                if set(available_tools) != set(expected_available_tools):
+                    raise AssertionError(f"Tool sets don't match. Expected: {set(expected_available_tools)}, Got: {set(available_tools)}")
+
+                # Call the add tool directly
+                try:
+                    call_tool_result = await session.call_tool(name="add", arguments={"a": 7, "b": 5})
+
+                    expected_result = "12.0"
+                    actual_result = call_tool_result.content[0].text if call_tool_result.content else None
+                except Exception as e:
+                    print(f"[Add Tool] Error calling add tool: {e}")
+                    raise AssertionError(f"Error calling add tool: {e}") from e
+
+                if actual_result != expected_result:
+                    raise AssertionError(f"Expected {expected_result}, got {actual_result}")
+
+                print("Test completed successfully")
+    except Exception as e:
+        print(f"Unexpected error connecting to MCP server: {e}")
+        raise AssertionError(f"Connection error, {e}") from e
+
+async def main():
+    """Main async function to run the test."""
+    try:
+        load_dotenv()
+
+        await call_add_tool()
+    except Exception as e:
+        print(f"Test failed with error: {e}")
+        sys.exit(1)
+
+
+if __name__ == '__main__':
+    asyncio.run(main())