test: update test_update_pipeline_yaml_request_exception to use mocked DeepsetClient

Author: mathislucka

test/test_main.py

@@ -173,17 +173,33 @@ async def test_update_pipeline_yaml_api_error_500_text() -> None:
         assert result == error_response
 
 
+@pytest.mark.asyncio
 @mock.patch.dict(os.environ, {"DEEPSET_WORKSPACE": TEST_WORKSPACE, "DEEPSET_API_KEY": TEST_API_KEY})
-@mock.patch("deepset_mcp.client.DeepsetClient.update_pipeline_yaml")
-def test_update_pipeline_yaml_request_exception(mock_update: mock.Mock) -> None:
+async def test_update_pipeline_yaml_request_exception() -> None:
     """Tests network request failure."""
-    mock_update.return_value = {"error": "Request failed: Connection timed out"}
-
-    result = update_pipeline_yaml(TEST_PIPELINE_NAME, VALID_YAML_CONTENT)
-
-    mock_update.assert_called_once_with(TEST_PIPELINE_NAME, VALID_YAML_CONTENT)
-    assert "error" in result
-    assert "Request failed: Connection timed out" in result["error"]
+    error_response = {"error": "Request failed: Connection timed out"}
+    
+    # Since update_pipeline_yaml is async, we need to test it differently
+    with mock.patch("deepset_mcp.main.DeepsetClient") as mock_client_class:
+        # Setup the mock client instance
+        mock_client_instance = mock.MagicMock()
+        mock_client_instance.__aenter__.return_value = mock_client_instance
+        mock_client_instance.__aexit__.return_value = None
+        mock_client_instance.workspace = TEST_WORKSPACE
+        mock_client_instance.request.return_value = error_response
+        mock_client_class.return_value = mock_client_instance
+        
+        # Call the function
+        result = await update_pipeline_yaml(TEST_PIPELINE_NAME, VALID_YAML_CONTENT)
+        
+        # Verify the call was made correctly
+        mock_client_instance.request.assert_called_once_with(
+            f"/workspaces/{TEST_WORKSPACE}/pipelines/{TEST_PIPELINE_NAME}/yaml", 
+            method="PUT", 
+            data={"query_yaml": VALID_YAML_CONTENT}
+        )
+        assert "error" in result
+        assert "Request failed: Connection timed out" in result["error"]
 
 
 @mock.patch.dict(os.environ, {"DEEPSET_WORKSPACE": TEST_WORKSPACE, "DEEPSET_API_KEY": TEST_API_KEY})