def test_call_new(self, m_queue_type):
event = mock.sentinel.event
group = mock.sentinel.group
queue_depth = mock.sentinel.queue_depth
m_queue = mock.Mock()
m_queue_type.return_value = m_queue
m_handler = mock.Mock()
m_th = mock.Mock()
m_tg = mock.Mock()
m_tg.add_thread.return_value = m_th
m_group_by = mock.Mock(return_value=group)
async_handler = h_async.Async(m_handler,
m_tg,
m_group_by,
queue_depth=queue_depth)
async_handler(event)
m_handler.assert_not_called()
m_queue_type.assert_called_once_with(queue_depth)
self.assertEqual({group: m_queue}, async_handler._queues)
m_tg.add_thread.assert_called_once_with(async_handler._run, group,
m_queue)
m_th.link.assert_called_once_with(async_handler._done, group)
m_queue.put.assert_called_once_with(event)
def test_done(self):
group = mock.sentinel.group
m_queue = mock.Mock()
async_handler = h_async.Async(mock.Mock(), mock.Mock(), mock.Mock())
async_handler._queues[group] = m_queue
async_handler._done(mock.Mock(), group)
self.assertFalse(async_handler._queues)
def test_done_terminated(self, m_critical):
group = mock.sentinel.group
m_queue = mock.Mock()
m_queue.empty.return_value = False
async_handler = h_async.Async(mock.Mock(), mock.Mock(), mock.Mock())
async_handler._queues[group] = m_queue
async_handler._done(mock.Mock(), group)
m_critical.assert_called_once()
def test_run_empty(self, m_count):
events = [mock.sentinel.event1, mock.sentinel.event2]
group = mock.sentinel.group
m_queue = mock.Mock()
m_queue.get.side_effect = events + [six_queue.Empty()]
m_handler = mock.Mock()
m_count.return_value = list(range(5))
async_handler = h_async.Async(m_handler, mock.Mock(), mock.Mock())
async_handler._run(group, m_queue)
m_handler.assert_has_calls([mock.call(event) for event in events])
def test_run(self, m_count):
event = mock.sentinel.event
group = mock.sentinel.group
m_queue = mock.Mock()
m_queue.get.return_value = event
m_handler = mock.Mock()
m_count.return_value = [1]
async_handler = h_async.Async(m_handler, mock.Mock(), mock.Mock(),
queue_depth=1)
with mock.patch('time.time', return_value=0):
async_handler._run(group, m_queue)
m_handler.assert_called_once_with(event)
def test_call(self):
event = mock.sentinel.event
group = mock.sentinel.group
m_queue = mock.Mock()
m_handler = mock.Mock()
m_group_by = mock.Mock(return_value=group)
async_handler = h_async.Async(m_handler, mock.Mock(), m_group_by)
async_handler._queues[group] = m_queue
async_handler(event)
m_handler.assert_not_called()
self.assertEqual({group: m_queue}, async_handler._queues)
m_queue.put.assert_called_once_with(event)
def test_run_stale(self, m_count):
events = [mock.sentinel.event1, mock.sentinel.event2]
group = mock.sentinel.group
m_queue = mock.Mock()
m_queue.empty.side_effect = [False, True, True]
m_queue.get.side_effect = events + [queue.Empty()]
m_handler = mock.Mock()
m_count.return_value = list(range(5))
async_handler = h_async.Async(m_handler, mock.Mock(), mock.Mock())
with mock.patch('time.sleep'):
async_handler._run(group, m_queue)
m_handler.assert_called_once_with(mock.sentinel.event2)
def test_run_empty(self, m_count):
events = [mock.sentinel.event1, mock.sentinel.event2]
group = mock.sentinel.group
m_queue = mock.Mock()
m_queue.empty.return_value = True
m_queue.get.side_effect = events + [queue.Empty()]
m_handler = mock.Mock()
m_count.return_value = list(range(5))
async_handler = h_async.Async(m_handler, mock.Mock(), mock.Mock())
with mock.patch('time.sleep'):
async_handler._run(group, m_queue)
m_handler.assert_has_calls([mock.call(event) for event in events])
self.assertEqual(len(events), m_handler.call_count)
def _wrap_dispatcher(self, dispatcher):
return h_log.LogExceptions(h_async.Async(dispatcher, self._tg,
h_k8s.object_uid))
