format

Author: cuericlee

tests/realtime/test_live.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import pytest
-from unittest.mock import AsyncMock, MagicMock
+from unittest.mock import AsyncMock, MagicMock, patch
 from veadk.realtime.live import DoubaoAsyncSession, ProtocolEvents
 from veadk.realtime import protocol
 from google.genai import types
@@ -111,21 +111,22 @@ async def test_receive(mock_session):
 
     for response_data, parse_data, expected in test_cases:
         mock_session._ws.recv = AsyncMock(return_value=response_data)
-        protocol.parse_response = MagicMock(return_value=parse_data)
-        async for msg in mock_session.receive():
-            if response_data["event"] == ProtocolEvents.ASR_INFO:
-                assert msg.server_content.interrupted == expected
-            elif response_data["event"] == ProtocolEvents.ASR_RESPONSE:
-                assert msg.server_content.input_transcription.text == expected
-            elif response_data["event"] == ProtocolEvents.TTS_RESPONSE:
-                assert (
-                    msg.server_content.model_turn.parts[0].inline_data.data == expected
-                )
-            elif response_data["event"] == ProtocolEvents.CHAT_RESPONSE:
-                assert msg.server_content.output_transcription.text == expected
-            elif response_data["event"] == ProtocolEvents.USAGE_RESPONSE:
-                assert msg.usage_metadata.tool_use_prompt_token_count == expected
-            break
+        with patch.object(protocol, "parse_response", return_value=parse_data):
+            async for msg in mock_session.receive():
+                if response_data["event"] == ProtocolEvents.ASR_INFO:
+                    assert msg.server_content.interrupted == expected
+                elif response_data["event"] == ProtocolEvents.ASR_RESPONSE:
+                    assert msg.server_content.input_transcription.text == expected
+                elif response_data["event"] == ProtocolEvents.TTS_RESPONSE:
+                    assert (
+                        msg.server_content.model_turn.parts[0].inline_data.data
+                        == expected
+                    )
+                elif response_data["event"] == ProtocolEvents.CHAT_RESPONSE:
+                    assert msg.server_content.output_transcription.text == expected
+                elif response_data["event"] == ProtocolEvents.USAGE_RESPONSE:
+                    assert msg.usage_metadata.tool_use_prompt_token_count == expected
+                break
 
 
 @pytest.mark.asyncio

tests/test_agent.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import os
-from unittest.mock import Mock, patch
+from unittest.mock import Mock, PropertyMock, patch
 
 from google.adk.agents.llm_agent import LlmAgent
 from google.adk.models.lite_llm import LiteLlm
@@ -76,19 +76,32 @@ def test_agent():
 
 @patch.dict("os.environ", {"MODEL_AGENT_API_KEY": "mock_api_key"})
 def test_agent_default_values():
-    agent = Agent()
-
-    assert agent.name == DEFAULT_AGENT_NAME
-
-    assert agent.model_name == DEFAULT_MODEL_AGENT_NAME
-    assert agent.model_provider == DEFAULT_MODEL_AGENT_PROVIDER
-    assert agent.model_api_base == DEFAULT_MODEL_AGENT_API_BASE
-
-    assert agent.tools == []
-    assert agent.sub_agents == []
-    assert agent.knowledgebase is None
-    assert agent.long_term_memory is None
-    # assert agent.tracers == []
+    with (
+        patch("veadk.agent.settings.model.name", new=DEFAULT_MODEL_AGENT_NAME),
+        patch("veadk.agent.settings.model.provider", new=DEFAULT_MODEL_AGENT_PROVIDER),
+        patch(
+            "veadk.agent.settings.model.api_base",
+            new=DEFAULT_MODEL_AGENT_API_BASE,
+        ),
+        patch(
+            "veadk.configs.model_configs.ModelConfig.api_key",
+            new_callable=PropertyMock,
+            return_value="mock_api_key",
+        ),
+    ):
+        agent = Agent()
+
+        assert agent.name == DEFAULT_AGENT_NAME
+
+        assert agent.model_name == DEFAULT_MODEL_AGENT_NAME
+        assert agent.model_provider == DEFAULT_MODEL_AGENT_PROVIDER
+        assert agent.model_api_base == DEFAULT_MODEL_AGENT_API_BASE
+
+        assert agent.tools == []
+        assert agent.sub_agents == []
+        assert agent.knowledgebase is None
+        assert agent.long_term_memory is None
+        # assert agent.tracers == []
 
 
 @patch.dict("os.environ", {"MODEL_AGENT_API_KEY": "mock_api_key"})

tests/test_feishu_channel_extension.py

@@ -1,3 +1,17 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 from types import SimpleNamespace
 
 import pytest