def test_call(self, tracker):
sn = '1'
cancelled = Cancelled(sn)
r = cancelled()
tracker.assert_called_once_with()
tracker.return_value.cancelled.assert_called_once_with(sn)
self.assertEqual(r, tracker.return_value.cancelled.return_value)
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_del_key_error(self, tracker):
sn = '1'
tracker.return_value.remove.side_effect = KeyError()
cancelled = Cancelled(sn)
cancelled.__del__()
def test_del(self, tracker):
sn = '1'
cancelled = Cancelled(sn)
cancelled.__del__()
tracker.return_value.remove.assert_called_once_with(sn)
def test_del(self, tracker):
sn = "1"
cancelled = Cancelled(sn)
cancelled.__del__()
tracker.return_value.remove.assert_called_once_with(sn)
def test_del_key_error(self, tracker):
sn = "1"
tracker.return_value.remove.side_effect = KeyError()
cancelled = Cancelled(sn)
cancelled.__del__()
