encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-03-30 10:57:20 -07:00
Code Issues Packages Projects Releases Wiki Activity

Update test_readonly_context.py

Browse Source
This commit is contained in:
seanzhou1023
2025-07-11 15:32:37 -07:00
committed by GitHub
parent 080edded76
commit 7de173ed90
1 changed files with 0 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unittests/agents/test_readonly_context.py
-18
View File
@@ -2,26 +2,14 @@ from types import MappingProxyType
from unittest.mock import MagicMock
from google.adk.agents.readonly_context import ReadonlyContext
from google.adk.auth.credential_service.base_credential_service import BaseCredentialService
import pytest
class DummyCredentialService(BaseCredentialService):
async def load_credential(self,auth_config, tool_context):
pass
async def save_credential(self,auth_config, tool_context):
pass
@pytest.fixture
def mock_invocation_context():
mock_context = MagicMock()
mock_context.invocation_id = "test-invocation-id"
mock_context.agent.name = "test-agent-name"
mock_context.session.state = {"key1": "value1", "key2": "value2"}
mock_context.credential_service = DummyCredentialService()
return mock_context
@@ -42,9 +30,3 @@ def test_state_content(mock_invocation_context):
assert isinstance(state, MappingProxyType)
assert state["key1"] == "value1"
assert state["key2"] == "value2"
def test_credential_service(mock_invocation_context):
readonly_context = ReadonlyContext(mock_invocation_context)
assert readonly_context.credential_service is not None
assert isinstance(readonly_context.credential_service, BaseCredentialService)