Clean up syntax a bit

pallets · Apr 1, 2016 · 3353c14 · 3353c14
1 parent b4dd92d
commit 3353c14
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/flask/helpers.py b/flask/helpers.py
@@ -621,9 +621,7 @@ def download_file(filename):
     try:
         if not os.path.isfile(filename):
             raise NotFound()
-    except TypeError:
-        raise BadRequest()
-    except ValueError:
+    except (TypeError, ValueError):
         raise BadRequest()
     options.setdefault('conditional', True)
     return send_file(filename, **options)

diff --git a/tests/test_helpers.py b/tests/test_helpers.py
@@ -511,12 +511,9 @@ def test_send_from_directory_bad_request(self):
         app.root_path = os.path.join(os.path.dirname(__file__),
                                      'test_apps', 'subdomaintestmodule')
         with app.test_request_context():
-            try:
+            with pytest.raises(BadRequest):
                 rv = flask.send_from_directory('static', 'bad\x00')
                 rv.close()
-                assert False
-            except BadRequest:
-                pass
 
 class TestLogging(object):