test2

Author: ionmincu

testcases/ground-to-cloud/run.sh

@@ -18,26 +18,21 @@ uv sync
 echo "Authenticating with UiPath..."
 uv run uipath auth --client-id="$CLIENT_ID" --client-secret="$CLIENT_SECRET" --base-url="$BASE_URL"
 
-# Remove empty tenantId line if exists
-sed -i '/^UIPATH_TENANT_ID=[[:space:]]*$/d' .env
-
 # Generate dynamic values
 PR_NUMBER=${GITHUB_PR_NUMBER:-"local"}
-#UNIQUE_ID=$(python -c "import uuid; print(str(uuid.uuid4()))" 2>/dev/null || python3 -c "import uuid; print(str(uuid.uuid4()))")
 UNIQUE_ID=$(cat /proc/sys/kernel/random/uuid)
 MCP_SERVER_NAME="mathmcp-${PR_NUMBER}"
 
-echo "Updating uipath.json with dynamic values..."
-echo "PR Number: $PR_NUMBER"
-echo "MCP Server Name: $MCP_SERVER_NAME"
-echo "Unique ID: $UNIQUE_ID"
+echo "Updating uipath.json with dynamic values... PR Number: $PR_NUMBER, MCP Server Name: $MCP_SERVER_NAME, Unique ID: $UNIQUE_ID"
+
+# Remove empty tenantId line if exists
+sed -i '/^UIPATH_TENANT_ID=[[:space:]]*$/d' .env
 
 # Replace placeholders in uipath.json using sed
+sed -i "s/PRNUMBER/$PR_NUMBER/g" mcp.json
 sed -i "s/PRNUMBER/$PR_NUMBER/g" uipath.json
 sed -i "s/163f06b8-31e6-4639-aa31-ae4a88968a92/$UNIQUE_ID/g" uipath.json
 
-sed -i "s/PRNUMBER/$PR_NUMBER/g" mcp.json
-
 echo "Packing agent..."
 uv run uipath pack
 

testcases/ground-to-cloud/test.py

@@ -17,7 +17,6 @@ def get_required_env_var(name: str) -> str:
 
 @retry(tries=3, delay=2, backoff=2)
 async def call_add_tool():
-    load_dotenv()
     # Load configuration from environment variables
     base_url = get_required_env_var("BASE_URL")
     folder_key = get_required_env_var("UIPATH_FOLDER_KEY")
@@ -27,37 +26,42 @@ async def call_add_tool():
     # Construct the MCP server URL
     mcp_url = f"{base_url}/agenthub_/mcp/{folder_key}/{mcp_server_name}"
 
-    # 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}")
+    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" ]
 
-            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)}")
+                print (f"Available tools: {available_tools}")
 
-            # Call the add tool directly
-            call_tool_result = await session.call_tool(name="add", arguments={"a": 7, "b": 5})
+                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)}")
 
-            expected_result = "12.0"
-            actual_result = call_tool_result.content[0].text if call_tool_result.content else None
+                # Call the add tool directly
+                call_tool_result = await session.call_tool(name="add", arguments={"a": 7, "b": 5})
 
-            if actual_result != expected_result:
-                raise AssertionError(f"Expected {expected_result}, got {actual_result}")
+                expected_result = "12.0"
+                actual_result = call_tool_result.content[0].text if call_tool_result.content else None
 
-            print("Test completed successfully")
+                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}")