async def test_service_sampler(self, *, mon): mon = Monitor() i = 0 mon.events_runtime = [] mon.sleep = AsyncMock(name='sleep') def on_cleanup(): nonlocal i mon.events_runtime.append(i + 0.34) i += 1 if i > 10: mon._stopped.set() mon._cleanup = Mock(name='_cleanup') mon._cleanup.side_effect = on_cleanup await mon._sampler(mon)
async def test_service_sampler(self, *, mon): mon = Monitor() i = 0 mon.events_runtime = [] mon.sleep = AsyncMock(name='sleep') def on_sample(prev_events, prev_messages): nonlocal i mon.events_runtime.append(i + 0.34) i += 1 if i > 10: mon._stopped.set() return prev_events, prev_messages mon._sample = Mock(name='_sample') mon._sample.side_effect = on_sample await mon._sampler(mon)