fix endline in the signal.py

pallets · Jun 17, 2015 · 1fbeb33 · 1fbeb33
1 parent d53d5c7
commit 1fbeb33
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/flask/signals.py b/flask/signals.py
@@ -53,4 +53,4 @@ def _fail(self, *args, **kwargs):
 appcontext_tearing_down = _signals.signal('appcontext-tearing-down')
 appcontext_pushed = _signals.signal('appcontext-pushed')
 appcontext_popped = _signals.signal('appcontext-popped')
-message_flashed = _signals.signal('message-flashed')
+message_flashed = _signals.signal('message-flashed')
diff --git a/tests/test_signals.py b/tests/test_signals.py
@@ -59,16 +59,14 @@ def record(sender, template, context):
         context['whiskey'] = 43
         recorded.append((template, context))
 
-    flask.before_render_template.connect(record, app)
-    try:
+    with flask.before_render_template.connected_to(record):
         rv = app.test_client().get('/')
         assert len(recorded) == 1
         template, context = recorded[0]
         assert template.name == 'simple_template.html'
         assert context['whiskey'] == 43
         assert rv.data == b'<h1>43</h1>'
-    finally:
-        flask.before_render_template.disconnect(record, app)
+
 
 def test_request_signals():
     app = flask.Flask(__name__)