def _post_teardown(self): if getattr(self, "_ctx", None) is not None: self._ctx.pop() del self._ctx if getattr(self, "app", None) is not None: if getattr(self, "_orig_response_class", None) is not None: self.app.response_class = self._orig_response_class del self.app if hasattr(self, "client"): del self.client if hasattr(self, "templates"): del self.templates if hasattr(self, "flashed_messages"): del self.flashed_messages if _is_signals: template_rendered.disconnect(self._add_template) if _is_message_flashed: message_flashed.disconnect(self._add_flash_message) if hasattr(self, "_original_template_render"): templating._render = self._original_template_render if self.run_gc_after_test: gc.collect()
def _post_teardown(self): if getattr(self, 'app', None) is not None: if getattr(self, '_orig_response_class', None) is not None: self.app.response_class = self._orig_response_class del self.app if hasattr(self, 'client'): del self.client if hasattr(self, 'templates'): del self.templates if hasattr(self, 'flashed_messages'): del self.flashed_messages if _is_signals: template_rendered.disconnect(self._add_template) if _is_message_flashed: message_flashed.disconnect(self._add_flash_message) if hasattr(self, '_original_template_render'): templating._render = self._original_template_render if self.run_gc_after_test: gc.collect()
def __exit__(self, *nargs): if getattr(self, 'app', None) is not None: del self.app del self.templates[:] del self.flashed_messages[:] template_rendered.disconnect(self._add_template) message_flashed.disconnect(self._add_flash_message)
def record_flashes(app): recorded = [] def record(sender, message, category, **extra): recorded.append(message) message_flashed.connect(record, app) yield recorded message_flashed.disconnect(record, app)
def capture_flashes(self): if hasattr(self, '_captured_flashes'): # nested call to this context manager -- just pass through yield self._captured_flashes return self._captured_flashes = [] def record(sender, message, category, **extra): self._captured_flashes.append((message, category)) message_flashed.connect(record, self.app) try: yield self._captured_flashes finally: message_flashed.disconnect(record, self.app) del self._captured_flashes
def tearDown(self): message_flashed.disconnect(self._add_flash_message)
def teardown(): message_flashed.disconnect(record, app)