encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Extract reusable private methods

Browse Source 
Co-authored-by: Xiang (Sean) Zhou <seanzhougoogle@google.com>
PiperOrigin-RevId: 871128393
This commit is contained in:
Xiang (Sean) Zhou
2026-02-16 22:52:47 -08:00
committed by Copybara-Service
parent 4e2d6159ae
commit 706f9fe74d
4 changed files with 384 additions and 334 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unittests/auth/test_toolset_auth.py
+6 -18
View File
@@ -31,6 +31,7 @@ from google.adk.auth.auth_credential import OAuth2Auth
from google.adk.auth.auth_preprocessor import TOOLSET_AUTH_CREDENTIAL_ID_PREFIX
from google.adk.auth.auth_tool import AuthConfig
from google.adk.auth.auth_tool import AuthToolArguments
from google.adk.flows.llm_flows.base_llm_flow import _resolve_toolset_auth
from google.adk.flows.llm_flows.base_llm_flow import BaseLlmFlow
from google.adk.flows.llm_flows.base_llm_flow import TOOLSET_AUTH_CREDENTIAL_ID_PREFIX as FLOW_PREFIX
from google.adk.flows.llm_flows.functions import build_auth_request_event
@@ -119,14 +120,6 @@ class TestResolveToolsetAuth:
agent.tools = []
return agent
@pytest.fixture
def flow(self):
"""Create a BaseLlmFlow instance for testing."""
# BaseLlmFlow is abstract, but we can still test _resolve_toolset_auth
flow = Mock(spec=BaseLlmFlow)
flow._resolve_toolset_auth = BaseLlmFlow._resolve_toolset_auth
return flow
@pytest.mark.asyncio
async def test_no_tools_returns_no_events(
self, mock_invocation_context, mock_agent
@@ -134,9 +127,8 @@ class TestResolveToolsetAuth:
"""Test that no events are yielded when agent has no tools."""
mock_agent.tools = []
flow = BaseLlmFlow.__new__(BaseLlmFlow)
events = []
async for event in flow._resolve_toolset_auth(
async for event in _resolve_toolset_auth(
mock_invocation_context, mock_agent
):
events.append(event)
@@ -152,9 +144,8 @@ class TestResolveToolsetAuth:
toolset = MockToolset(auth_config=None)
mock_agent.tools = [toolset]
flow = BaseLlmFlow.__new__(BaseLlmFlow)
events = []
async for event in flow._resolve_toolset_auth(
async for event in _resolve_toolset_auth(
mock_invocation_context, mock_agent
):
events.append(event)
@@ -184,9 +175,8 @@ class TestResolveToolsetAuth:
mock_manager.get_auth_credential = AsyncMock(return_value=mock_credential)
MockCredentialManager.return_value = mock_manager
flow = BaseLlmFlow.__new__(BaseLlmFlow)
events = []
async for event in flow._resolve_toolset_auth(
async for event in _resolve_toolset_auth(
mock_invocation_context, mock_agent
):
events.append(event)
@@ -213,9 +203,8 @@ class TestResolveToolsetAuth:
mock_manager.get_auth_credential = AsyncMock(return_value=None)
MockCredentialManager.return_value = mock_manager
flow = BaseLlmFlow.__new__(BaseLlmFlow)
events = []
async for event in flow._resolve_toolset_auth(
async for event in _resolve_toolset_auth(
mock_invocation_context, mock_agent
):
events.append(event)
@@ -258,9 +247,8 @@ class TestResolveToolsetAuth:
mock_manager.get_auth_credential = AsyncMock(return_value=None)
MockCredentialManager.return_value = mock_manager
flow = BaseLlmFlow.__new__(BaseLlmFlow)
events = []
async for event in flow._resolve_toolset_auth(
async for event in _resolve_toolset_auth(
mock_invocation_context, mock_agent
):
events.append(event)