mirror of https://github.com/mitsuhiko/flask.git
Armin Ronacher
14 years ago
1 changed files with 46 additions and 0 deletions
@ -0,0 +1,46 @@ |
|||||||
|
# -*- coding: utf-8 -*- |
||||||
|
""" |
||||||
|
flask.testsuite.subclassing |
||||||
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
||||||
|
|
||||||
|
Test that certain behavior of flask can be customized by |
||||||
|
subclasses. |
||||||
|
|
||||||
|
:copyright: (c) 2011 by Armin Ronacher. |
||||||
|
:license: BSD, see LICENSE for more details. |
||||||
|
""" |
||||||
|
import flask |
||||||
|
import unittest |
||||||
|
from StringIO import StringIO |
||||||
|
from logging import StreamHandler |
||||||
|
from flask.testsuite import FlaskTestCase |
||||||
|
|
||||||
|
|
||||||
|
class FlaskSubclassingTestCase(FlaskTestCase): |
||||||
|
|
||||||
|
def test_supressed_exception_logging(self): |
||||||
|
class SupressedFlask(flask.Flask): |
||||||
|
def log_exception(self, exc_info): |
||||||
|
pass |
||||||
|
|
||||||
|
out = StringIO() |
||||||
|
app = SupressedFlask(__name__) |
||||||
|
app.logger_name = 'flask_tests/test_supressed_exception_logging' |
||||||
|
app.logger.addHandler(StreamHandler(out)) |
||||||
|
|
||||||
|
@app.route('/') |
||||||
|
def index(): |
||||||
|
1/0 |
||||||
|
|
||||||
|
rv = app.test_client().get('/') |
||||||
|
self.assert_equal(rv.status_code, 500) |
||||||
|
self.assert_('Internal Server Error' in rv.data) |
||||||
|
|
||||||
|
err = out.getvalue() |
||||||
|
self.assert_equal(err, '') |
||||||
|
|
||||||
|
|
||||||
|
def suite(): |
||||||
|
suite = unittest.TestSuite() |
||||||
|
suite.addTest(unittest.makeSuite(FlaskSubclassingTestCase)) |
||||||
|
return suite |
Loading…
Reference in new issue