mirror of https://github.com/pallets/flask.git
Added tests for make_response and get_debug_flag to improve coverage of helpers.py
This commit is contained in:
parent
a0e38a8cf1
commit
6f49089a62
|
|
@ -22,6 +22,7 @@ from werkzeug.exceptions import BadRequest, NotFound
|
|||
from werkzeug.http import parse_cache_control_header, parse_options_header
|
||||
from werkzeug.http import http_date
|
||||
from flask._compat import StringIO, text_type
|
||||
from flask.helpers import get_debug_flag, make_response
|
||||
|
||||
|
||||
def has_encoding(name):
|
||||
|
|
@ -941,3 +942,52 @@ class TestSafeJoin(object):
|
|||
for args in failing:
|
||||
with pytest.raises(NotFound):
|
||||
print(flask.safe_join(*args))
|
||||
|
||||
class TestHelpers(object):
|
||||
|
||||
def test_get_debug_flag(self):
|
||||
original_debug_value = os.environ.get('FLASK_DEBUG') or ''
|
||||
os.environ['FLASK_DEBUG'] = ''
|
||||
assert get_debug_flag() == None
|
||||
assert get_debug_flag(default=True) == True
|
||||
|
||||
os.environ['FLASK_DEBUG'] = '0'
|
||||
assert get_debug_flag() == False
|
||||
assert get_debug_flag(default=True) == False
|
||||
|
||||
os.environ['FLASK_DEBUG'] = 'False'
|
||||
assert get_debug_flag() == False
|
||||
assert get_debug_flag(default=True) == False
|
||||
|
||||
os.environ['FLASK_DEBUG'] = 'No'
|
||||
assert get_debug_flag() == False
|
||||
assert get_debug_flag(default=True) == False
|
||||
|
||||
os.environ['FLASK_DEBUG'] = 'True'
|
||||
assert get_debug_flag() == True
|
||||
assert get_debug_flag(default=True) == True
|
||||
|
||||
os.environ['FLASK_DEBUG'] = original_debug_value
|
||||
|
||||
def test_make_response_no_args(self):
|
||||
app = flask.Flask(__name__)
|
||||
app.testing = True
|
||||
@app.route('/')
|
||||
def index():
|
||||
return flask.helpers.make_response()
|
||||
c = app.test_client()
|
||||
rv = c.get()
|
||||
assert rv
|
||||
|
||||
def test_make_response_with_args(self):
|
||||
app = flask.Flask(__name__)
|
||||
app.testing = True
|
||||
@app.route('/')
|
||||
def index():
|
||||
response = flask.helpers.make_response(flask.render_template_string('Hello World'))
|
||||
response.headers['X-Parachutes'] = 'parachutes are cool'
|
||||
return response
|
||||
c = app.test_client()
|
||||
rv = c.get()
|
||||
assert rv
|
||||
assert rv.headers['X-Parachutes'] == 'parachutes are cool'
|
||||
|
|
|
|||
Loading…
Reference in New Issue