test: update test_update_pipeline_yaml_success_empty_response to use mocked DeepsetClient

Author: mathislucka

test/test_main.py

@@ -89,16 +89,30 @@ async def test_update_pipeline_yaml_success_json_response() -> None:
         assert result == {"status": "success", "message": "Updated"}
 
 
+@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_success_empty_response(mock_update: mock.Mock) -> None:
+async def test_update_pipeline_yaml_success_empty_response() -> None:
     """Tests successful update with empty response body."""
-    mock_update.return_value = {"status": "success", "message": "Pipeline YAML updated successfully (empty response body)"}
-
-    result = update_pipeline_yaml(TEST_PIPELINE_NAME, VALID_YAML_CONTENT)
-
-    mock_update.assert_called_once_with(TEST_PIPELINE_NAME, VALID_YAML_CONTENT)
-    assert result == {"status": "success", "message": "Pipeline YAML updated successfully (empty response body)"}
+    # 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 = {"status": "success", "message": "API returned empty response body"}
+        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 result == {"status": "success", "message": "API returned empty response body"}
 
 
 @mock.patch.dict(os.environ, {"DEEPSET_WORKSPACE": TEST_WORKSPACE, "DEEPSET_API_KEY": TEST_API_KEY})