File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed
tests/server/agent_execution Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -276,7 +276,6 @@ async def test_build_populate_false_with_reference_task_ids(self) -> None:
276276 self .assertEqual (request_context .related_tasks , [])
277277 self .mock_task_store .get .assert_not_called ()
278278
279-
280279 async def test_build_with_custom_id_generators (self ) -> None :
281280 mock_task_id_generator = AsyncMock (spec = IDGenerator )
282281 mock_context_id_generator = AsyncMock (spec = IDGenerator )
@@ -299,7 +298,7 @@ async def test_build_with_custom_id_generators(self) -> None:
299298 task = None ,
300299 context = server_call_context ,
301300 )
302-
301+
303302 mock_task_id_generator .generate .assert_called_once ()
304303 mock_context_id_generator .generate .assert_called_once ()
305304 self .assertEqual (request_context .task_id , 'custom_task_id' )
@@ -328,7 +327,7 @@ async def test_build_with_provided_ids_and_custom_id_generators(self) -> None:
328327 task = None ,
329328 context = server_call_context ,
330329 )
331-
330+
332331 mock_task_id_generator .generate .assert_not_called ()
333332 mock_context_id_generator .generate .assert_not_called ()
334333 self .assertEqual (request_context .task_id , provided_task_id )
You can’t perform that action at this time.
0 commit comments