orbit-oss/flask
4
4
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Removed temp subscription contextmanager in blinker tests to support upcoming api improvements better

Browse source
This commit is contained in:
Armin Ronacher 2010-07-20 15:09:51 +01:00
parent 80268a408b
commit b49afa21ad
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/flask_tests.py
View file

@ -1052,12 +1052,15 @@ class TestSignals(unittest.TestCase):
def record(sender, template, context): def record(sender, template, context):
recorded.append((template, context)) recorded.append((template, context))
with flask.template_rendered.temporarily_connected_to(record, app): flask.template_rendered.connect(record, app)
try:
rv = app.test_client().get('/') rv = app.test_client().get('/')
assert len(recorded) == 1 assert len(recorded) == 1
template, context = recorded[0] template, context = recorded[0]
assert template.name == 'simple_template.html' assert template.name == 'simple_template.html'
assert context['whiskey'] == 42 assert context['whiskey'] == 42
finally:
flask.template_rendered.disconnect(record, app)
def test_request_signals(self): def test_request_signals(self):
app = flask.Flask(__name__) app = flask.Flask(__name__)
@ -1110,10 +1113,13 @@ class TestSignals(unittest.TestCase):
def record(sender, exception): def record(sender, exception):
recorded.append(exception) recorded.append(exception)
with flask.got_request_exception.temporarily_connected_to(record): flask.got_request_exception.connect(record, app)
try:
assert app.test_client().get('/').status_code == 500 assert app.test_client().get('/').status_code == 500
assert len(recorded) == 1 assert len(recorded) == 1
assert isinstance(recorded[0], ZeroDivisionError) assert isinstance(recorded[0], ZeroDivisionError)
finally:
flask.got_request_exception.disconnect(record, app)
def suite(): def suite():