mirror of
https://github.com/EvolutionAPI/adk-python.git
synced 2025-07-14 01:41:25 -06:00
57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
from unittest.mock import AsyncMock
|
|
from unittest.mock import MagicMock
|
|
from unittest.mock import patch
|
|
|
|
from google.adk.agents.live_request_queue import LiveRequest
|
|
from google.adk.agents.live_request_queue import LiveRequestQueue
|
|
from google.genai import types
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_close_queue():
|
|
queue = LiveRequestQueue()
|
|
|
|
with patch.object(queue._queue, "put_nowait") as mock_put_nowait:
|
|
queue.close()
|
|
mock_put_nowait.assert_called_once_with(LiveRequest(close=True))
|
|
|
|
|
|
def test_send_content():
|
|
queue = LiveRequestQueue()
|
|
content = MagicMock(spec=types.Content)
|
|
|
|
with patch.object(queue._queue, "put_nowait") as mock_put_nowait:
|
|
queue.send_content(content)
|
|
mock_put_nowait.assert_called_once_with(LiveRequest(content=content))
|
|
|
|
|
|
def test_send_realtime():
|
|
queue = LiveRequestQueue()
|
|
blob = MagicMock(spec=types.Blob)
|
|
|
|
with patch.object(queue._queue, "put_nowait") as mock_put_nowait:
|
|
queue.send_realtime(blob)
|
|
mock_put_nowait.assert_called_once_with(LiveRequest(blob=blob))
|
|
|
|
|
|
def test_send():
|
|
queue = LiveRequestQueue()
|
|
req = LiveRequest(content=MagicMock(spec=types.Content))
|
|
|
|
with patch.object(queue._queue, "put_nowait") as mock_put_nowait:
|
|
queue.send(req)
|
|
mock_put_nowait.assert_called_once_with(req)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get():
|
|
queue = LiveRequestQueue()
|
|
res = MagicMock(spec=types.Content)
|
|
|
|
with patch.object(queue._queue, "get", return_value=res) as mock_get:
|
|
result = await queue.get()
|
|
|
|
assert result == res
|
|
mock_get.assert_called_once()
|