def test_reject_discarded(self): url = 'test-url' tag = '1234' queue = Mock() message = Mock(delivery_info={DELIVERY_TAG: tag}) # test reader = Reader(queue, url=url) reader.channel = Mock() reader.reject(message, False) # validation reader.channel.basic_reject.assert_called_once_with(tag, False)
def test_ack(self): url = 'test-url' tag = '1234' queue = Mock() message = Mock(delivery_info={DELIVERY_TAG: tag}) # test reader = Reader(queue, url=url) reader.channel = Mock() reader.ack(message) # validation reader.channel.basic_ack.assert_called_once_with(tag)
def test_reject_exception(self): url = 'test-url' tag = '1234' queue = Mock() message = Mock(delivery_info={DELIVERY_TAG: tag}) # test reader = Reader(queue, url=url) reader.channel = Mock() reader.channel.basic_reject.side_effect = ValueError # validation self.assertRaises(ValueError, reader.reject, message)
def test_reject_discarded(self): url = 'test-url' tag = '1234' queue = Mock() message = Mock(delivery_info={DELIVERY_TAG: tag}) # test reader = Reader(queue, url=url) reader.channel = Mock() reader.reject(message, False) # validation reader.channel.basic_reject.assert_called_once_with(tag, False)
def test_reject_exception(self): url = 'test-url' tag = '1234' queue = Mock() message = Mock(delivery_info={DELIVERY_TAG: tag}) # test reader = Reader(queue, url=url) reader.channel = Mock() reader.channel.basic_reject.side_effect = ValueError # validation self.assertRaises(ValueError, reader.reject, message)
def test_ack(self): url = 'test-url' tag = '1234' queue = Mock() message = Mock(delivery_info={DELIVERY_TAG: tag}) # test reader = Reader(queue, url=url) reader.channel = Mock() reader.ack(message) # validation reader.channel.basic_ack.assert_called_once_with(tag)
def test_close(self): connection = Mock() channel = Mock() receiver = Mock() # test reader = Reader(None, '') reader.connection = connection reader.channel = channel reader.receiver = receiver reader.is_open = Mock(return_value=True) reader.close() # validation receiver.close.assert_called_once_with() channel.close.assert_called_once_with() self.assertFalse(connection.close.called)
def test_close(self): connection = Mock() channel = Mock() receiver = Mock() # test reader = Reader(None, '') reader.connection = connection reader.channel = channel reader.receiver = receiver reader.is_open = Mock(return_value=True) reader.close() # validation receiver.close.assert_called_once_with() channel.close.assert_called_once_with() self.assertFalse(connection.close.called)