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)
