forked from orbit-oss/flask
Add Support for FLASK_ENV (#2570)
This introduces environments to Flask
This commit is contained in:
parent
60eecb547d
commit
2433522d29
10 changed files with 151 additions and 71 deletions
|
|
@ -21,7 +21,7 @@ from werkzeug.http import http_date, parse_cache_control_header, \
|
|||
|
||||
import flask
|
||||
from flask._compat import StringIO, text_type
|
||||
from flask.helpers import get_debug_flag
|
||||
from flask.helpers import get_debug_flag, get_env
|
||||
|
||||
|
||||
def has_encoding(name):
|
||||
|
|
@ -886,7 +886,7 @@ class TestSafeJoin(object):
|
|||
class TestHelpers(object):
|
||||
|
||||
@pytest.mark.parametrize('debug, expected_flag, expected_default_flag', [
|
||||
('', None, True),
|
||||
('', False, False),
|
||||
('0', False, False),
|
||||
('False', False, False),
|
||||
('No', False, False),
|
||||
|
|
@ -898,7 +898,18 @@ class TestHelpers(object):
|
|||
assert get_debug_flag() is None
|
||||
else:
|
||||
assert get_debug_flag() == expected_flag
|
||||
assert get_debug_flag(default=True) == expected_default_flag
|
||||
assert get_debug_flag() == expected_default_flag
|
||||
|
||||
@pytest.mark.parametrize('env, ref_env, debug', [
|
||||
('', 'production', False),
|
||||
('production', 'production', False),
|
||||
('development', 'development', True),
|
||||
('other', 'other', False),
|
||||
])
|
||||
def test_get_env(self, monkeypatch, env, ref_env, debug):
|
||||
monkeypatch.setenv('FLASK_ENV', env)
|
||||
assert get_debug_flag() == debug
|
||||
assert get_env() == ref_env
|
||||
|
||||
def test_make_response(self):
|
||||
app = flask.Flask(__name__)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue