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__()