1-
21import asyncio
32import pytest
43from unittest .mock import AsyncMock , MagicMock , patch
3635 'messageId' : '111' ,
3736}
3837
38+
3939@pytest .fixture
4040def mock_event_queue ():
4141 return AsyncMock (spec = EventQueue )
4242
43+
4344@pytest .fixture
44- def event_consumer (
45- mock_event_queue : EventQueue
46- ):
45+ def event_consumer (mock_event_queue : EventQueue ):
4746 return EventConsumer (queue = mock_event_queue )
4847
48+
4949@pytest .mark .asyncio
5050async def test_consume_one_task_event (
5151 event_consumer : MagicMock ,
@@ -57,6 +57,7 @@ async def test_consume_one_task_event(
5757 assert result == task_event
5858 mock_event_queue .task_done .assert_called_once ()
5959
60+
6061@pytest .mark .asyncio
6162async def test_consume_one_message_event (
6263 event_consumer : MagicMock ,
@@ -68,6 +69,7 @@ async def test_consume_one_message_event(
6869 assert result == message_event
6970 mock_event_queue .task_done .assert_called_once ()
7071
72+
7173@pytest .mark .asyncio
7274async def test_consume_one_a2a_error_event (
7375 event_consumer : MagicMock ,
@@ -79,6 +81,7 @@ async def test_consume_one_a2a_error_event(
7981 assert result == error_event
8082 mock_event_queue .task_done .assert_called_once ()
8183
84+
8285@pytest .mark .asyncio
8386async def test_consume_one_jsonrpc_error_event (
8487 event_consumer : MagicMock ,
@@ -90,6 +93,7 @@ async def test_consume_one_jsonrpc_error_event(
9093 assert result == error_event
9194 mock_event_queue .task_done .assert_called_once ()
9295
96+
9397@pytest .mark .asyncio
9498async def test_consume_one_queue_empty (
9599 event_consumer : MagicMock ,
@@ -103,6 +107,7 @@ async def test_consume_one_queue_empty(
103107 pass
104108 mock_event_queue .task_done .assert_not_called ()
105109
110+
106111@pytest .mark .asyncio
107112async def test_consume_all_multiple_events (
108113 event_consumer : MagicMock ,
@@ -125,12 +130,14 @@ async def test_consume_all_multiple_events(
125130 ),
126131 ]
127132 cursor = 0
133+
128134 async def mock_dequeue () -> Any :
129135 nonlocal cursor
130136 if cursor < len (events ):
131137 event = events [cursor ]
132138 cursor += 1
133139 return event
140+
134141 mock_event_queue .dequeue_event = mock_dequeue
135142 consumed_events : list [Any ] = []
136143 async for event in event_consumer .consume_all ():
@@ -141,6 +148,7 @@ async def mock_dequeue() -> Any:
141148 assert consumed_events [2 ] == events [2 ]
142149 assert mock_event_queue .task_done .call_count == 3
143150
151+
144152@pytest .mark .asyncio
145153async def test_consume_until_message (
146154 event_consumer : MagicMock ,
@@ -164,12 +172,14 @@ async def test_consume_until_message(
164172 ),
165173 ]
166174 cursor = 0
175+
167176 async def mock_dequeue () -> Any :
168177 nonlocal cursor
169178 if cursor < len (events ):
170179 event = events [cursor ]
171180 cursor += 1
172181 return event
182+
173183 mock_event_queue .dequeue_event = mock_dequeue
174184 consumed_events : list [Any ] = []
175185 async for event in event_consumer .consume_all ():
@@ -180,6 +190,7 @@ async def mock_dequeue() -> Any:
180190 assert consumed_events [2 ] == events [2 ]
181191 assert mock_event_queue .task_done .call_count == 3
182192
193+
183194@pytest .mark .asyncio
184195async def test_consume_message_events (
185196 event_consumer : MagicMock ,
@@ -190,12 +201,14 @@ async def test_consume_message_events(
190201 Message (** MESSAGE_PAYLOAD , final = True ),
191202 ]
192203 cursor = 0
204+
193205 async def mock_dequeue () -> Any :
194206 nonlocal cursor
195207 if cursor < len (events ):
196208 event = events [cursor ]
197209 cursor += 1
198210 return event
211+
199212 mock_event_queue .dequeue_event = mock_dequeue
200213 consumed_events : list [Any ] = []
201214 async for event in event_consumer .consume_all ():
0 commit comments