31 lines
631 B
Python
31 lines
631 B
Python
from flask import template_rendered
|
|
from contextlib import contextmanager
|
|
|
|
|
|
@contextmanager
|
|
def captured_templates(app):
|
|
recorded = []
|
|
|
|
def record(sender, template, context, **extra):
|
|
recorded.append((template, context))
|
|
|
|
template_rendered.connect(record, app)
|
|
try:
|
|
yield recorded
|
|
finally:
|
|
template_rendered.disconnect(record, app)
|
|
|
|
|
|
class FakeLogger:
|
|
def __init__(self):
|
|
self.messages = []
|
|
|
|
def info(self, msg):
|
|
self.messages.append(msg)
|
|
|
|
def warning(self, msg):
|
|
self.messages.append(msg)
|
|
|
|
def error(self, msg):
|
|
self.messages.append(msg)
|