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_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]) 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))