def test_ack_exception(self):
message = Mock()
session = Mock()
session.acknowledge.side_effect = ValueError
# test
reader = Reader(None, '')
reader.session = session
self.assertRaises(ValueError, reader.ack, message)
def test_reject_exception(self, disposition):
message = Mock()
session = Mock()
session.acknowledge.side_effect = ValueError
# test
reader = Reader(None, '')
reader.session = session
self.assertRaises(ValueError, reader.reject, message)
def test_reject_exception(self, disposition):
message = Mock()
session = Mock()
session.acknowledge.side_effect = ValueError
# test
reader = Reader(None, '')
reader.session = session
self.assertRaises(ValueError, reader.reject, message)
def test_ack_exception(self):
message = Mock()
session = Mock()
session.acknowledge.side_effect = ValueError
# test
reader = Reader(None, '')
reader.session = session
self.assertRaises(ValueError, reader.ack, message)
def test_ack(self):
message = Mock()
# test
reader = Reader(None, '')
reader.session = Mock()
reader.ack(message)
# validation
reader.session.acknowledge.assert_called_once_with(message=message)
def test_ack(self):
message = Mock()
# test
reader = Reader(None, '')
reader.session = Mock()
reader.ack(message)
# validation
reader.session.acknowledge.assert_called_once_with(message=message)
def test_reject_rejected(self, disposition, rejected):
message = Mock()
# test
reader = Reader(None, '')
reader.session = Mock()
reader.reject(message, False)
# validation
disposition.assert_called_once_with(rejected)
reader.session.acknowledge.assert_called_once_with(
message=message, disposition=disposition.return_value)
def test_reject_rejected(self, disposition, rejected):
message = Mock()
# test
reader = Reader(None, '')
reader.session = Mock()
reader.reject(message, False)
# validation
disposition.assert_called_once_with(rejected)
reader.session.acknowledge.assert_called_once_with(
message=message, disposition=disposition.return_value)
def test_close(self):
connection = Mock()
session = Mock(connection=connection)
receiver = Mock()
# test
reader = Reader(None, '')
reader.connection = connection
reader.session = session
reader.receiver = receiver
reader.is_open = Mock(return_value=True)
reader.close()
# validation
receiver.close.assert_called_once_with()
session.close.assert_called_once_with()
self.assertFalse(connection.close.called)
def test_close(self):
connection = Mock()
session = Mock(connection=connection)
receiver = Mock()
# test
reader = Reader(None, '')
reader.connection = connection
reader.session = session
reader.receiver = receiver
reader.is_open = Mock(return_value=True)
reader.close()
# validation
receiver.close.assert_called_once_with()
session.close.assert_called_once_with()
self.assertFalse(connection.close.called)
