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