66
77@pytest .mark .parametrize (
88 "base_url,endpoint,expected" ,
9- [
9+ [
1010 # Additional test cases to verify behavior with different URL structures
1111 (
12- "https://mcp.example.com/weather/sse" ,
13- "/messages/?session_id=616df71373444d76bd566df4377c9629" ,
14- "https://mcp.example.com/weather/messages/?session_id=616df71373444d76bd566df4377c9629"
12+ "https://mcp.example.com/weather/sse" ,
13+ "/messages/?session_id=616df71373444d76bd566df4377c9629" ,
14+ "https://mcp.example.com/weather/messages/?session_id=616df71373444d76bd566df4377c9629" ,
1515 ),
1616 (
17- "https://mcp.example.com/weather/clarksburg/sse" ,
18- "/messages/?session_id=616df71373444d76bd566df4377c9629" ,
19- "https://mcp.example.com/weather/clarksburg/messages/?session_id=616df71373444d76bd566df4377c9629"
17+ "https://mcp.example.com/weather/clarksburg/sse" ,
18+ "/messages/?session_id=616df71373444d76bd566df4377c9629" ,
19+ "https://mcp.example.com/weather/clarksburg/messages/?session_id=616df71373444d76bd566df4377c9629" ,
2020 ),
2121 (
22- "https://mcp.example.com/sse" ,
23- "/messages/?session_id=616df71373444d76bd566df4377c9629" ,
24- "https://mcp.example.com/messages/?session_id=616df71373444d76bd566df4377c9629"
22+ "https://mcp.example.com/sse" ,
23+ "/messages/?session_id=616df71373444d76bd566df4377c9629" ,
24+ "https://mcp.example.com/messages/?session_id=616df71373444d76bd566df4377c9629" ,
2525 ),
2626 ],
2727)
2828def test_custom_url_join (base_url , endpoint , expected ):
2929 """Test the custom_url_join function with messages endpoint and session ID."""
3030 result = custom_url_join (base_url , endpoint )
31- assert result == expected
31+ assert result == expected
0 commit comments