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_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 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,
)
