def test__consumer_fails(self):
queue = collections.deque([1, 2, 3])
mock_consume = mock.MagicMock(side_effect=Exception())
mock_is_published = mock.MagicMock()
mock_is_published.isSet = mock.MagicMock(return_value=True)
broker._consumer(mock_consume, queue, mock_is_published)
self.assertEqual(0, len(queue))
def test__consumer_fails(self):
queue = collections.deque([1, 2, 3])
mock_consume = mock.MagicMock(side_effect=Exception())
mock_is_published = mock.MagicMock()
mock_is_published.isSet = mock.MagicMock(return_value=True)
broker._consumer(mock_consume, queue, mock_is_published)
self.assertEqual(0, len(queue))
def test__consumer(self):
queue = collections.deque([1, 2, 3])
mock_consume = mock.MagicMock()
mock_is_published = mock.MagicMock()
mock_is_published.isSet = mock.MagicMock(return_value=True)
broker._consumer(mock_consume, queue, mock_is_published)
self.assertEqual(3, mock_consume.call_count)
self.assertEqual(0, len(queue))
def test__consumer(self):
queue = collections.deque([1, 2, 3])
mock_consume = mock.MagicMock()
mock_is_published = mock.MagicMock()
mock_is_published.isSet = mock.MagicMock(return_value=True)
broker._consumer(mock_consume, queue, mock_is_published)
self.assertEqual(3, mock_consume.call_count)
self.assertEqual(0, len(queue))
def test__consumer_indexerror(self, mock_log):
consume = mock.Mock()
consume.side_effect = IndexError()
queue = collections.deque([1, 2, 3])
broker._consumer(consume, queue)
self.assertTrue(mock_log.warning.called)
self.assertFalse(queue)
expected = [mock.call({}, 1), mock.call({}, 2), mock.call({}, 3)]
self.assertEqual(expected, consume.mock_calls)
def test__consumer_indexerror(self, mock_log):
consume = mock.Mock()
consume.side_effect = IndexError()
queue = collections.deque([1, 2, 3])
broker._consumer(consume, queue)
self.assertTrue(mock_log.warning.called)
self.assertFalse(queue)
expected = [mock.call({}, 1), mock.call({}, 2), mock.call({}, 3)]
self.assertEqual(expected, consume.mock_calls)
def test__consumer_cache(self):
cache_keys_history = []
def consume(cache, item):
cache[item] = True
cache_keys_history.append(list(cache))
queue = collections.deque([1, 2, 3])
broker._consumer(consume, queue)
self.assertEqual([[1], [1, 2], [1, 2, 3]], cache_keys_history)
def test__consumer_cache(self):
cache_keys_history = []
def consume(cache, item):
cache[item] = True
cache_keys_history.append(list(cache))
queue = collections.deque([1, 2, 3])
broker._consumer(consume, queue)
self.assertEqual([[1], [1, 2], [1, 2, 3]], cache_keys_history)
def test__consumer_cache(self):
cache_keys_history = []
def consume(cache, item):
cache[item] = True
cache_keys_history.append(list(cache))
queue = collections.deque([1, 2, 3])
mock_is_published = mock.MagicMock()
mock_is_published.isSet = mock.MagicMock(return_value=True)
broker._consumer(consume, queue, mock_is_published)
self.assertEqual([[1], [1, 2], [1, 2, 3]], cache_keys_history)
def test__consumer_cache(self):
cache_keys_history = []
def consume(cache, item):
cache[item] = True
cache_keys_history.append(list(cache))
queue = collections.deque([1, 2, 3])
mock_is_published = mock.MagicMock()
mock_is_published.isSet = mock.MagicMock(return_value=True)
broker._consumer(consume, queue, mock_is_published)
self.assertEqual([[1], [1, 2], [1, 2, 3]], cache_keys_history)
def test__consumer_fails(self):
queue = collections.deque([1, 2, 3])
mock_consume = mock.MagicMock(side_effect=Exception())
broker._consumer(mock_consume, queue)
self.assertEqual(0, len(queue))
def test__consumer(self):
queue = collections.deque([1, 2, 3])
mock_consume = mock.MagicMock()
broker._consumer(mock_consume, queue)
self.assertEqual(3, mock_consume.call_count)
self.assertEqual(0, len(queue))
def test__consumer_fails(self):
queue = collections.deque([1, 2, 3])
mock_consume = mock.MagicMock(side_effect=Exception())
broker._consumer(mock_consume, queue)
self.assertEqual(0, len(queue))
def test__consumer(self):
queue = collections.deque([1, 2, 3])
mock_consume = mock.MagicMock()
broker._consumer(mock_consume, queue)
self.assertEqual(3, mock_consume.call_count)
self.assertEqual(0, len(queue))
