def test_next_not_found(self, _find):
_impl = Mock()
plugin = Mock()
plugin.Reader.return_value = _impl
_find.return_value = plugin
# test
reader = Reader(Node(''))
reader.get = Mock(return_value=None)
reader.authenticator = Mock()
_message, _document = reader.next(10)
# validation
reader.get.assert_called_once_with(10)
self.assertEqual(_message, None)
self.assertEqual(_document, None)
def test_next_not_found(self, _find):
_impl = Mock()
plugin = Mock()
plugin.Reader.return_value = _impl
_find.return_value = plugin
# test
reader = Reader(Node(''))
reader.get = Mock(return_value=None)
reader.authenticator = Mock()
_message, _document = reader.next(10)
# validation
reader.get.assert_called_once_with(10)
self.assertEqual(_message, None)
self.assertEqual(_document, None)
def test_next_auth_rejected(self, _find, auth, validate):
_impl = Mock()
plugin = Mock()
plugin.Reader.return_value = _impl
_find.return_value = plugin
message = Mock(body='test-content')
auth.validate.side_effect = ModelError
# test
reader = Reader(Node(''))
reader.get = Mock(return_value=message)
reader.authenticator = Mock()
self.assertRaises(ModelError, reader.next, 10)
# validation
reader.get.assert_called_once_with(10)
auth.validate.assert_called_once_with(reader.authenticator, message.body)
message.ack.assert_called_once_with()
self.assertFalse(validate.called)
def test_next(self, _find, auth, validate):
_impl = Mock()
plugin = Mock()
plugin.Reader.return_value = _impl
_find.return_value = plugin
message = Mock(body='test-content')
document = Mock()
auth.validate.return_value = document
# test
reader = Reader(Node(''))
reader.get = Mock(return_value=message)
reader.authenticator = Mock()
_message, _document = reader.next(10)
# validation
reader.get.assert_called_once_with(10)
auth.validate.assert_called_once_with(reader.authenticator, message.body)
validate.assert_called_once_with(document)
self.assertEqual(_message, reader.get.return_value)
self.assertEqual(_document, document)
