File tree Expand file tree Collapse file tree 1 file changed +6
-5
lines changed
Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Original file line number Diff line number Diff line change 22
33import asyncio
44from datetime import timedelta
5+ from functools import update_wrapper
56from typing import Any
67from unittest .mock import AsyncMock
78
@@ -16,15 +17,15 @@ def listener() -> AsyncMock:
1617 async def async_listener (payload : Any ) -> None :
1718 pass
1819
19- return AsyncMock (target = async_listener )
20+ al = AsyncMock ()
21+ update_wrapper (al , async_listener )
22+ return al
2023
2124
2225async def test_emit_system_info_event (listener : AsyncMock ) -> None :
23- system_info_interval = timedelta (milliseconds = 50 )
24- test_tolerance_coefficient = 10
25- async with LocalEventManager (system_info_interval = system_info_interval ) as event_manager :
26+ async with LocalEventManager (system_info_interval = timedelta (milliseconds = 50 )) as event_manager :
2627 event_manager .on (event = Event .SYSTEM_INFO , listener = listener )
27- await asyncio .sleep (system_info_interval . total_seconds () * test_tolerance_coefficient )
28+ await asyncio .sleep (0.2 )
2829
2930 assert listener .call_count >= 1
3031 assert isinstance (listener .call_args [0 ][0 ], EventSystemInfoData )
You can’t perform that action at this time.
0 commit comments