def test_report_no_replyto(self): request = Mock(sn=1, data=2, replyto=None) producer = Mock() progress = Progress(request, producer) # test progress.report() # validation self.assertFalse(producer.send.called)
def test_report_exception(self, log): request = Mock(sn=1, data=2, replyto=3) producer = Mock() producer.send.side_effect = ValueError() progress = Progress(request, producer) # test progress.report() # validation self.assertTrue(log.exception.called)
def __call__(self): """ Dispatch received request. """ request = self.request cancelled = Cancelled(request.sn) latency = self.plugin.latency if latency: sleep(latency) if not self.plugin.url or cancelled(): self.discard() return producer = self._producer(self.plugin) progress = Progress(request, producer) context = Context(request.sn, progress, cancelled) Context.set(context) producer.open() try: self.producer = producer self.send_started(request) result = self.plugin.dispatch(request) self.send_reply(request, result) self.commit() finally: producer.close() Context.set()
def test_report(self): request = Mock(sn=1, data=2, replyto=3) producer = Mock() progress = Progress(request, producer) progress.total = 10 progress.completed = 4 progress.details = {'A': 1} # test progress.report() # validation producer.send.assert_called_once_with(request.replyto, status='progress', completed=progress.completed, details=progress.details, total=progress.total, data=request.data, sn=1)
def test_report(self): request = Mock(sn=1, data=2, replyto=3) producer = Mock() progress = Progress(request, producer) progress.total = 10 progress.completed = 4 progress.details = {"A": 1} # test progress.report() # validation producer.send.assert_called_once_with( request.replyto, status="progress", completed=progress.completed, details=progress.details, total=progress.total, data=request.data, sn=1, )