def test_read_nothing(self): url = 'test-url' node = Node('test-queue') consumer = ConsumerThread(node, url) consumer._reader = Mock() consumer._reader.next.return_value = (None, None) consumer.dispatch = Mock() # test consumer._read() # validate self.assertFalse(consumer.dispatch.called)
def test_read_nothing(self): url = 'test-url' node = Node('test-queue') consumer = ConsumerThread(node, url) consumer._reader = Mock() consumer._reader.next.return_value = (None, None) consumer.dispatch = Mock() # test consumer._read() # validate self.assertFalse(consumer.dispatch.called)
def test_read_validation_failed(self): url = 'test-url' node = Node('test-queue') failed = ValidationFailed(details='test') consumer = ConsumerThread(node, url) consumer._reader = Mock() consumer._reader.next.side_effect = failed consumer._rejected = Mock() # test consumer._read() # validate consumer._rejected.assert_called_once_with( failed.code, failed.description, failed.document, failed.details)
def test_read_exception(self, sleep): url = 'test-url' node = Node('test-queue') consumer = ConsumerThread(node, url) consumer._reader = Mock() consumer._reader.next.side_effect = IndexError consumer._open = Mock() consumer._close = Mock() # test consumer._read() # validation consumer._close.assert_called_once_with() consumer._open.assert_called_once_with() sleep.assert_called_once_with(60)
def test_read(self): url = 'test-url' node = Node('test-queue') message = Mock() document = Mock() consumer = ConsumerThread(node, url) consumer._reader = Mock() consumer._reader.next.return_value = (message, document) consumer.dispatch = Mock() # test consumer._read() # validate consumer.dispatch.assert_called_once_with(document) message.ack.assert_called_once_with()
def test_read_exception(self, sleep): url = 'test-url' node = Node('test-queue') consumer = ConsumerThread(node, url) consumer._reader = Mock() consumer._reader.next.side_effect = IndexError consumer._open = Mock() consumer._close = Mock() # test consumer._read() # validation consumer._close.assert_called_once_with() consumer._open.assert_called_once_with() sleep.assert_called_once_with(60)
def test_read(self): url = 'test-url' node = Node('test-queue') message = Mock() document = Mock() consumer = ConsumerThread(node, url) consumer._reader = Mock() consumer._reader.next.return_value = (message, document) consumer.dispatch = Mock() # test consumer._read() # validate consumer.dispatch.assert_called_once_with(document) message.ack.assert_called_once_with()
def test_read_validation_failed(self): url = 'test-url' node = Node('test-queue') failed = ValidationFailed(details='test') consumer = ConsumerThread(node, url) consumer._reader = Mock() consumer._reader.next.side_effect = failed consumer._rejected = Mock() # test consumer._read() # validate consumer._rejected.assert_called_once_with(failed.code, failed.description, failed.document, failed.details)
def test_read_invalid_document(self): url = 'test-url' node = Node('test-queue') code = 12 description = 'just up and failed' document = Mock() details = 'crashed' ir = InvalidDocument(code, description, document, details) consumer = ConsumerThread(node, url) consumer._reader = Mock() consumer._reader.next.side_effect = ir consumer._rejected = Mock() # test consumer._read() # validate consumer._rejected.assert_called_once_with( ir.code, ir.description, ir.document, ir.details)
def test_read_invalid_document(self): url = 'test-url' node = Node('test-queue') code = 12 description = 'just up and failed' document = Mock() details = 'crashed' ir = InvalidDocument(code, description, document, details) consumer = ConsumerThread(node, url) consumer._reader = Mock() consumer._reader.next.side_effect = ir consumer._rejected = Mock() # test consumer._read() # validate consumer._rejected.assert_called_once_with(ir.code, ir.description, ir.document, ir.details)
def test_run(self, reader): url = 'test-url' node = Node('test-queue') consumer = ConsumerThread(node, url) consumer._open = Mock() consumer._close = Mock() consumer._read = Mock(side_effect=StopIteration) # test try: consumer.run() except StopIteration: pass # validation reader.assert_called_once_with(node, url) consumer._open.assert_called_once_with() consumer._read.assert_called_once_with() consumer._close.assert_called_once_with()
def test_run(self, reader): url = 'test-url' node = Node('test-queue') consumer = ConsumerThread(node, url) consumer._open = Mock() consumer._close = Mock() consumer._read = Mock(side_effect=StopIteration) # test try: consumer.run() except StopIteration: pass # validation reader.assert_called_once_with(node, url) consumer._open.assert_called_once_with() consumer._read.assert_called_once_with() consumer._close.assert_called_once_with()