test(adk): fix test_3lo_flow_simulation by properly mocking credential_service

Author: anubhav756

packages/toolbox-adk/tests/integration/test_integration.py

@@ -16,7 +16,7 @@
 import os
 from typing import Any, Optional
 from inspect import signature, Parameter
-from unittest.mock import MagicMock
+from unittest.mock import MagicMock, AsyncMock
 
 import pytest
 from pydantic import ValidationError
@@ -192,12 +192,16 @@ async def test_3lo_flow_simulation(self):
             mock_ctx_first = MagicMock()
             # Simulate "No Auth Response Found"
             mock_ctx_first.get_auth_response.return_value = None
+            mock_cred_service_first = AsyncMock()
+            mock_cred_service_first.load_credential.return_value = None
+            mock_ctx_first._invocation_context = MagicMock()
+            mock_ctx_first._invocation_context.credential_service = mock_cred_service_first
 
             print("Running tool first time (expecting auth request)...")
             result_first = await tool.run_async({"num_rows": "1"}, mock_ctx_first)
 
             # The wrapper should catch the missing creds and request them.
-            assert isinstance(result_first, dict) and "error" in result_first, "Tool should return error sig for auth requirement"
+            assert result_first is None, "Tool should return None sig for auth requirement"
             mock_ctx_first.request_credential.assert_called_once()
 
             # Inspect the requested config