Skip to content

Commit

Permalink
asyncio: Refactor test__run_once_logging() to not rely on the exact n…
Browse files Browse the repository at this point in the history
…umber of

calls to time.monotonic(). Use a "fast select" and a "slow select" instead.
  • Loading branch information
vstinner committed Jun 17, 2014
1 parent 8d3e02e commit d6f02fc
Showing 1 changed file with 11 additions and 18 deletions.
29 changes: 11 additions & 18 deletions Lib/test/test_asyncio/test_base_events.py
Original file line number Diff line number Diff line change
Expand Up @@ -240,30 +240,23 @@ def test_set_debug(self):
self.loop.set_debug(False)
self.assertFalse(self.loop.get_debug())

@mock.patch('asyncio.base_events.time')
@mock.patch('asyncio.base_events.logger')
def test__run_once_logging(self, m_logger, m_time):
# Log to INFO level if timeout > 1.0 sec.
idx = -1
data = [10.0, 10.0, 12.0, 13.0]

def monotonic():
nonlocal data, idx
idx += 1
return data[idx]
def test__run_once_logging(self, m_logger):
def slow_select(timeout):
time.sleep(1.0)
return []

m_time.monotonic = monotonic

self.loop._scheduled.append(
asyncio.TimerHandle(11.0, lambda: True, (), self.loop))
# Log to INFO level if timeout > 1.0 sec.
self.loop._selector.select = slow_select
self.loop._process_events = mock.Mock()
self.loop._run_once()
self.assertEqual(logging.INFO, m_logger.log.call_args[0][0])

idx = -1
data = [10.0, 10.0, 10.3, 13.0]
self.loop._scheduled = [asyncio.TimerHandle(11.0, lambda: True, (),
self.loop)]
def fast_select(timeout):
time.sleep(0.001)
return []

self.loop._selector.select = fast_select
self.loop._run_once()
self.assertEqual(logging.DEBUG, m_logger.log.call_args[0][0])

Expand Down

0 comments on commit d6f02fc

Please sign in to comment.