Browse Source

Kill class in test_basic

pull/1165/head
Markus Unterwaditzer 10 years ago
parent
commit
a4931ff3a7
  1. 304
      tests/test_basic.py

304
tests/test_basic.py

@ -16,20 +16,17 @@ import uuid
import time
import flask
import pickle
import unittest
from datetime import datetime
from threading import Thread
from tests import emits_module_deprecation_warning
from flask._compat import text_type
from werkzeug.exceptions import BadRequest, NotFound, Forbidden
from werkzeug.http import parse_date
from werkzeug.routing import BuildError
class TestBasicFunctionality(object):
def test_options_work(self):
def test_options_work():
app = flask.Flask(__name__)
@app.route('/', methods=['GET', 'POST'])
def index():
return 'Hello World'
@ -37,19 +34,24 @@ class TestBasicFunctionality(object):
assert sorted(rv.allow) == ['GET', 'HEAD', 'OPTIONS', 'POST']
assert rv.data == b''
def test_options_on_multiple_rules(self):
def test_options_on_multiple_rules():
app = flask.Flask(__name__)
@app.route('/', methods=['GET', 'POST'])
def index():
return 'Hello World'
@app.route('/', methods=['PUT'])
def index_put():
return 'Aha!'
rv = app.test_client().open('/', method='OPTIONS')
assert sorted(rv.allow) == ['GET', 'HEAD', 'OPTIONS', 'POST', 'PUT']
def test_options_handling_disabled(self):
def test_options_handling_disabled():
app = flask.Flask(__name__)
def index():
return 'Hello World!'
index.provide_automatic_options = False
@ -58,6 +60,7 @@ class TestBasicFunctionality(object):
assert rv.status_code == 405
app = flask.Flask(__name__)
def index2():
return 'Hello World!'
index2.provide_automatic_options = True
@ -65,11 +68,14 @@ class TestBasicFunctionality(object):
rv = app.test_client().open('/', method='OPTIONS')
assert sorted(rv.allow) == ['OPTIONS']
def test_request_dispatching(self):
def test_request_dispatching():
app = flask.Flask(__name__)
@app.route('/')
def index():
return flask.request.method
@app.route('/more', methods=['GET', 'POST'])
def more():
return flask.request.method
@ -88,17 +94,21 @@ class TestBasicFunctionality(object):
assert rv.status_code == 405
assert sorted(rv.allow) == ['GET', 'HEAD', 'OPTIONS', 'POST']
def test_disallow_string_for_allowed_methods(self):
def test_disallow_string_for_allowed_methods():
app = flask.Flask(__name__)
with pytest.raises(TypeError):
@app.route('/', methods='GET POST')
def index():
return "Hey"
def test_url_mapping(self):
def test_url_mapping():
app = flask.Flask(__name__)
def index():
return flask.request.method
def more():
return flask.request.method
@ -119,15 +129,18 @@ class TestBasicFunctionality(object):
assert rv.status_code == 405
assert sorted(rv.allow) == ['GET', 'HEAD', 'OPTIONS', 'POST']
def test_werkzeug_routing(self):
def test_werkzeug_routing():
from werkzeug.routing import Submount, Rule
app = flask.Flask(__name__)
app.url_map.add(Submount('/foo', [
Rule('/bar', endpoint='bar'),
Rule('/', endpoint='index')
]))
def bar():
return 'bar'
def index():
return 'index'
app.view_functions['bar'] = bar
@ -137,7 +150,8 @@ class TestBasicFunctionality(object):
assert c.get('/foo/').data == b'index'
assert c.get('/foo/bar').data == b'bar'
def test_endpoint_decorator(self):
def test_endpoint_decorator():
from werkzeug.routing import Submount, Rule
app = flask.Flask(__name__)
app.url_map.add(Submount('/foo', [
@ -157,13 +171,16 @@ class TestBasicFunctionality(object):
assert c.get('/foo/').data == b'index'
assert c.get('/foo/bar').data == b'bar'
def test_session(self):
def test_session():
app = flask.Flask(__name__)
app.secret_key = 'testkey'
@app.route('/set', methods=['POST'])
def set():
flask.session['value'] = flask.request.form['value']
return 'value set'
@app.route('/get')
def get():
return flask.session['value']
@ -172,12 +189,14 @@ class TestBasicFunctionality(object):
assert c.post('/set', data={'value': '42'}).data == b'value set'
assert c.get('/get').data == b'42'
def test_session_using_server_name(self):
def test_session_using_server_name():
app = flask.Flask(__name__)
app.config.update(
SECRET_KEY='foo',
SERVER_NAME='example.com'
)
@app.route('/')
def index():
flask.session['testing'] = 42
@ -186,12 +205,14 @@ class TestBasicFunctionality(object):
assert 'domain=.example.com' in rv.headers['set-cookie'].lower()
assert 'httponly' in rv.headers['set-cookie'].lower()
def test_session_using_server_name_and_port(self):
def test_session_using_server_name_and_port():
app = flask.Flask(__name__)
app.config.update(
SECRET_KEY='foo',
SERVER_NAME='example.com:8080'
)
@app.route('/')
def index():
flask.session['testing'] = 42
@ -200,13 +221,15 @@ class TestBasicFunctionality(object):
assert 'domain=.example.com' in rv.headers['set-cookie'].lower()
assert 'httponly' in rv.headers['set-cookie'].lower()
def test_session_using_server_name_port_and_path(self):
def test_session_using_server_name_port_and_path():
app = flask.Flask(__name__)
app.config.update(
SECRET_KEY='foo',
SERVER_NAME='example.com:8080',
APPLICATION_ROOT='/foo'
)
@app.route('/')
def index():
flask.session['testing'] = 42
@ -216,11 +239,14 @@ class TestBasicFunctionality(object):
assert 'path=/foo' in rv.headers['set-cookie'].lower()
assert 'httponly' in rv.headers['set-cookie'].lower()
def test_session_using_application_root(self):
def test_session_using_application_root():
class PrefixPathMiddleware(object):
def __init__(self, app, prefix):
self.app = app
self.prefix = prefix
def __call__(self, environ, start_response):
environ['SCRIPT_NAME'] = self.prefix
return self.app(environ, start_response)
@ -231,6 +257,7 @@ class TestBasicFunctionality(object):
SECRET_KEY='foo',
APPLICATION_ROOT='/bar'
)
@app.route('/')
def index():
flask.session['testing'] = 42
@ -238,7 +265,8 @@ class TestBasicFunctionality(object):
rv = app.test_client().get('/', 'http://example.com:8080/')
assert 'path=/bar' in rv.headers['set-cookie'].lower()
def test_session_using_session_settings(self):
def test_session_using_session_settings():
app = flask.Flask(__name__)
app.config.update(
SECRET_KEY='foo',
@ -249,6 +277,7 @@ class TestBasicFunctionality(object):
SESSION_COOKIE_SECURE=True,
SESSION_COOKIE_PATH='/'
)
@app.route('/')
def index():
flask.session['testing'] = 42
@ -260,8 +289,10 @@ class TestBasicFunctionality(object):
assert 'secure' in cookie
assert 'httponly' not in cookie
def test_missing_session(self):
def test_missing_session():
app = flask.Flask(__name__)
def expect_exception(f, *args, **kwargs):
try:
f(*args, **kwargs)
@ -274,10 +305,12 @@ class TestBasicFunctionality(object):
expect_exception(flask.session.__setitem__, 'foo', 42)
expect_exception(flask.session.pop, 'foo')
def test_session_expiration(self):
def test_session_expiration():
permanent = True
app = flask.Flask(__name__)
app.secret_key = 'testkey'
@app.route('/')
def index():
flask.session['test'] = 42
@ -307,7 +340,8 @@ class TestBasicFunctionality(object):
match = re.search(r'\bexpires=([^;]+)', rv.headers['set-cookie'])
assert match is None
def test_session_stored_last(self):
def test_session_stored_last():
app = flask.Flask(__name__)
app.secret_key = 'development-key'
app.testing = True
@ -316,6 +350,7 @@ class TestBasicFunctionality(object):
def modify_session(response):
flask.session['foo'] = 42
return response
@app.route('/')
def dump_session_contents():
return repr(flask.session.get('foo'))
@ -324,7 +359,8 @@ class TestBasicFunctionality(object):
assert c.get('/').data == b'None'
assert c.get('/').data == b'42'
def test_session_special_types(self):
def test_session_special_types():
app = flask.Flask(__name__)
app.secret_key = 'development-key'
app.testing = True
@ -355,7 +391,8 @@ class TestBasicFunctionality(object):
assert type(rv['b']) == bytes
assert rv['t'] == (1, 2, 3)
def test_session_cookie_setting(self):
def test_session_cookie_setting():
app = flask.Flask(__name__)
app.testing = True
app.secret_key = 'dev key'
@ -398,7 +435,8 @@ class TestBasicFunctionality(object):
app.config['SESSION_REFRESH_EACH_REQUEST'] = False
run_test(expect_header=False)
def test_flashes(self):
def test_flashes():
app = flask.Flask(__name__)
app.secret_key = 'testkey'
@ -410,7 +448,8 @@ class TestBasicFunctionality(object):
assert flask.session.modified
assert list(flask.get_flashed_messages()) == ['Zap', 'Zip']
def test_extended_flashing(self):
def test_extended_flashing():
# Be sure app.testing=True below, else tests can fail silently.
#
# Specifically, if app.testing is not set to True, the AssertionErrors
@ -451,13 +490,15 @@ class TestBasicFunctionality(object):
@app.route('/test_filter/')
def test_filter():
messages = flask.get_flashed_messages(category_filter=['message'], with_categories=True)
messages = flask.get_flashed_messages(
category_filter=['message'], with_categories=True)
assert list(messages) == [('message', u'Hello World')]
return ''
@app.route('/test_filters/')
def test_filters():
messages = flask.get_flashed_messages(category_filter=['message', 'warning'], with_categories=True)
messages = flask.get_flashed_messages(
category_filter=['message', 'warning'], with_categories=True)
assert list(messages) == [
('message', u'Hello World'),
('warning', flask.Markup(u'<em>Testing</em>'))
@ -466,7 +507,8 @@ class TestBasicFunctionality(object):
@app.route('/test_filters_without_returning_categories/')
def test_filters2():
messages = flask.get_flashed_messages(category_filter=['message', 'warning'])
messages = flask.get_flashed_messages(
category_filter=['message', 'warning'])
assert len(messages) == 2
assert messages[0] == u'Hello World'
assert messages[1] == flask.Markup(u'<em>Testing</em>')
@ -494,17 +536,21 @@ class TestBasicFunctionality(object):
c.get('/')
c.get('/test_filters_without_returning_categories/')
def test_request_processing(self):
def test_request_processing():
app = flask.Flask(__name__)
evts = []
@app.before_request
def before_request():
evts.append('before')
@app.after_request
def after_request(response):
response.data += b'|after'
evts.append('after')
return response
@app.route('/')
def index():
assert 'before' in evts
@ -515,9 +561,11 @@ class TestBasicFunctionality(object):
assert 'after' in evts
assert rv == b'request|after'
def test_after_request_processing(self):
def test_after_request_processing():
app = flask.Flask(__name__)
app.testing = True
@app.route('/')
def index():
@flask.after_this_request
@ -530,13 +578,16 @@ class TestBasicFunctionality(object):
assert resp.status_code == 200
assert resp.headers['X-Foo'] == 'a header'
def test_teardown_request_handler(self):
def test_teardown_request_handler():
called = []
app = flask.Flask(__name__)
@app.teardown_request
def teardown_request(exc):
called.append(True)
return "Ignored"
@app.route('/')
def root():
return "Response"
@ -545,14 +596,17 @@ class TestBasicFunctionality(object):
assert b'Response' in rv.data
assert len(called) == 1
def test_teardown_request_handler_debug_mode(self):
def test_teardown_request_handler_debug_mode():
called = []
app = flask.Flask(__name__)
app.testing = True
@app.teardown_request
def teardown_request(exc):
called.append(True)
return "Ignored"
@app.route('/')
def root():
return "Response"
@ -561,10 +615,12 @@ class TestBasicFunctionality(object):
assert b'Response' in rv.data
assert len(called) == 1
def test_teardown_request_handler_error(self):
def test_teardown_request_handler_error():
called = []
app = flask.Flask(__name__)
app.config['LOGGER_HANDLER_POLICY'] = 'never'
@app.teardown_request
def teardown_request1(exc):
assert type(exc) == ZeroDivisionError
@ -576,6 +632,7 @@ class TestBasicFunctionality(object):
raise TypeError()
except:
pass
@app.teardown_request
def teardown_request2(exc):
assert type(exc) == ZeroDivisionError
@ -587,6 +644,7 @@ class TestBasicFunctionality(object):
raise TypeError()
except:
pass
@app.route('/')
def fails():
1 // 0
@ -595,29 +653,37 @@ class TestBasicFunctionality(object):
assert b'Internal Server Error' in rv.data
assert len(called) == 2
def test_before_after_request_order(self):
def test_before_after_request_order():
called = []
app = flask.Flask(__name__)
@app.before_request
def before1():
called.append(1)
@app.before_request
def before2():
called.append(2)
@app.after_request
def after1(response):
called.append(4)
return response
@app.after_request
def after2(response):
called.append(3)
return response
@app.teardown_request
def finish1(exc):
called.append(6)
@app.teardown_request
def finish2(exc):
called.append(5)
@app.route('/')
def index():
return '42'
@ -625,24 +691,31 @@ class TestBasicFunctionality(object):
assert rv.data == b'42'
assert called == [1, 2, 3, 4, 5, 6]
def test_error_handling(self):
def test_error_handling():
app = flask.Flask(__name__)
app.config['LOGGER_HANDLER_POLICY'] = 'never'
@app.errorhandler(404)
def not_found(e):
return 'not found', 404
@app.errorhandler(500)
def internal_server_error(e):
return 'internal server error', 500
@app.errorhandler(Forbidden)
def forbidden(e):
return 'forbidden', 403
@app.route('/')
def index():
flask.abort(404)
@app.route('/error')
def error():
1 // 0
@app.route('/forbidden')
def error2():
flask.abort(403)
@ -657,11 +730,14 @@ class TestBasicFunctionality(object):
assert rv.status_code == 403
assert b'forbidden' == rv.data
def test_before_request_and_routing_errors(self):
def test_before_request_and_routing_errors():
app = flask.Flask(__name__)
@app.before_request
def attach_something():
flask.g.something = 'value'
@app.errorhandler(404)
def return_something(error):
return flask.g.something, 404
@ -669,15 +745,18 @@ class TestBasicFunctionality(object):
assert rv.status_code == 404
assert rv.data == b'value'
def test_user_error_handling(self):
def test_user_error_handling():
class MyException(Exception):
pass
app = flask.Flask(__name__)
@app.errorhandler(MyException)
def handle_my_exception(e):
assert isinstance(e, MyException)
return '42'
@app.route('/')
def index():
raise MyException()
@ -685,15 +764,18 @@ class TestBasicFunctionality(object):
c = app.test_client()
assert c.get('/').data == b'42'
def test_http_error_subclass_handling(self):
def test_http_error_subclass_handling():
class ForbiddenSubclass(Forbidden):
pass
app = flask.Flask(__name__)
@app.errorhandler(ForbiddenSubclass)
def handle_forbidden_subclass(e):
assert isinstance(e, ForbiddenSubclass)
return 'banana'
@app.errorhandler(403)
def handle_forbidden_subclass(e):
assert not isinstance(e, ForbiddenSubclass)
@ -703,9 +785,11 @@ class TestBasicFunctionality(object):
@app.route('/1')
def index1():
raise ForbiddenSubclass()
@app.route('/2')
def index2():
flask.abort(403)
@app.route('/3')
def index3():
raise Forbidden()
@ -715,9 +799,11 @@ class TestBasicFunctionality(object):
assert c.get('/2').data == b'apple'
assert c.get('/3').data == b'apple'
def test_trapping_of_bad_request_key_errors(self):
def test_trapping_of_bad_request_key_errors():
app = flask.Flask(__name__)
app.testing = True
@app.route('/fail')
def fail():
flask.request.form['missing_key']
@ -733,26 +819,26 @@ class TestBasicFunctionality(object):
else:
assert False, 'Expected exception'
def test_trapping_of_all_http_exceptions(self):
def test_trapping_of_all_http_exceptions():
app = flask.Flask(__name__)
app.testing = True
app.config['TRAP_HTTP_EXCEPTIONS'] = True
@app.route('/fail')
def fail():
flask.abort(404)
c = app.test_client()
try:
with pytest.raises(NotFound):
c.get('/fail')
except NotFound as e:
pass
else:
assert False, 'Expected exception'
def test_enctype_debug_helper(self):
def test_enctype_debug_helper():
from flask.debughelpers import DebugFilesKeyError
app = flask.Flask(__name__)
app.debug = True
@app.route('/fail', methods=['POST'])
def index():
return flask.request.files['foo'].filename
@ -769,29 +855,36 @@ class TestBasicFunctionality(object):
else:
assert False, 'Expected exception'
def test_response_creation(self):
def test_response_creation():
app = flask.Flask(__name__)
@app.route('/unicode')
def from_unicode():
return u'Hällo Wörld'
@app.route('/string')
def from_string():
return u'Hällo Wörld'.encode('utf-8')
@app.route('/args')
def from_tuple():
return 'Meh', 400, {
'X-Foo': 'Testing',
'Content-Type': 'text/plain; charset=utf-8'
}
@app.route('/two_args')
def from_two_args_tuple():
return 'Hello', {
'X-Foo': 'Test',
'Content-Type': 'text/plain; charset=utf-8'
}
@app.route('/args_status')
def from_status_tuple():
return 'Hi, status!', 400
@app.route('/args_header')
def from_response_instance_status_tuple():
return flask.Response('Hello world', 404), {
@ -822,7 +915,8 @@ class TestBasicFunctionality(object):
assert rv4.headers['X-Bar'] == 'Foo'
assert rv4.status_code == 404
def test_make_response(self):
def test_make_response():
app = flask.Flask(__name__)
with app.test_request_context():
rv = flask.make_response()
@ -840,7 +934,8 @@ class TestBasicFunctionality(object):
assert rv.data == b'W00t'
assert rv.mimetype == 'text/html'
def test_make_response_with_response_instance(self):
def test_make_response_with_response_instance():
app = flask.Flask(__name__)
with app.test_request_context():
rv = flask.make_response(
@ -862,8 +957,10 @@ class TestBasicFunctionality(object):
assert rv.headers['Content-Type'] == 'text/html'
assert rv.headers['X-Foo'] == 'bar'
def test_url_generation(self):
def test_url_generation():
app = flask.Flask(__name__)
@app.route('/hello/<name>', methods=['POST'])
def hello():
pass
@ -872,7 +969,8 @@ class TestBasicFunctionality(object):
assert flask.url_for('hello', name='test x', _external=True) == \
'http://localhost/hello/test%20x'
def test_build_error_handler(self):
def test_build_error_handler():
app = flask.Flask(__name__)
# Test base case, a URL which results in a BuildError.
@ -888,7 +986,8 @@ class TestBasicFunctionality(object):
try:
raise RuntimeError('Test case where BuildError is not current.')
except RuntimeError:
pytest.raises(BuildError, app.handle_url_build_error, error, 'spam', {})
pytest.raises(
BuildError, app.handle_url_build_error, error, 'spam', {})
# Test a custom handler.
def handler(error, endpoint, values):
@ -898,23 +997,29 @@ class TestBasicFunctionality(object):
with app.test_request_context():
assert flask.url_for('spam') == '/test_handler/'
def test_custom_converters(self):
def test_custom_converters():
from werkzeug.routing import BaseConverter
class ListConverter(BaseConverter):
def to_python(self, value):
return value.split(',')
def to_url(self, value):
base_to_url = super(ListConverter, self).to_url
return ','.join(base_to_url(x) for x in value)
app = flask.Flask(__name__)
app.url_map.converters['list'] = ListConverter
@app.route('/<list:args>')
def index(args):
return '|'.join(args)
c = app.test_client()
assert c.get('/1,2,3').data == b'1|2|3'
def test_static_files(self):
def test_static_files():
app = flask.Flask(__name__)
app.testing = True
rv = app.test_client().get('/static/index.html')
@ -925,9 +1030,11 @@ class TestBasicFunctionality(object):
'/static/index.html'
rv.close()
def test_none_response(self):
def test_none_response():
app = flask.Flask(__name__)
app.testing = True
@app.route('/')
def test():
return None
@ -939,15 +1046,18 @@ class TestBasicFunctionality(object):
else:
assert "Expected ValueError"
def test_request_locals(self):
def test_request_locals():
assert repr(flask.g) == '<LocalProxy unbound>'
assert not flask.g
def test_test_app_proper_environ(self):
def test_test_app_proper_environ():
app = flask.Flask(__name__)
app.config.update(
SERVER_NAME='localhost.localdomain:5000'
)
@app.route('/')
def index():
return 'Foo'
@ -998,10 +1108,12 @@ class TestBasicFunctionality(object):
rv = app.test_client().get('/', 'http://foo.localhost.localdomain')
assert rv.data == b'Foo SubDomain'
def test_exception_propagation(self):
def test_exception_propagation():
def apprunner(configkey):
app = flask.Flask(__name__)
app.config['LOGGER_HANDLER_POLICY'] = 'never'
@app.route('/')
def index():
1 // 0
@ -1026,17 +1138,21 @@ class TestBasicFunctionality(object):
t.start()
t.join()
def test_max_content_length(self):
def test_max_content_length():
app = flask.Flask(__name__)
app.config['MAX_CONTENT_LENGTH'] = 64
@app.before_request
def always_first():
flask.request.form['myfile']
assert False
@app.route('/accept', methods=['POST'])
def accept_file():
flask.request.form['myfile']
assert False
@app.errorhandler(413)
def catcher(error):
return '42'
@ -1045,7 +1161,8 @@ class TestBasicFunctionality(object):
rv = c.post('/accept', data={'myfile': 'foo' * 100})
assert rv.data == b'42'
def test_url_processors(self):
def test_url_processors():
app = flask.Flask(__name__)
@app.url_defaults
@ -1076,7 +1193,8 @@ class TestBasicFunctionality(object):
assert c.get('/de/about').data == b'/foo'
assert c.get('/foo').data == b'/en/about'
def test_inject_blueprint_url_defaults(self):
def test_inject_blueprint_url_defaults():
app = flask.Flask(__name__)
bp = flask.Blueprint('foo.bar.baz', __name__,
template_folder='template')
@ -1084,8 +1202,10 @@ class TestBasicFunctionality(object):
@bp.url_defaults
def bp_defaults(endpoint, values):
values['page'] = 'login'
@bp.route('/<page>')
def view(page): pass
def view(page):
pass
app.register_blueprint(bp)
@ -1099,7 +1219,8 @@ class TestBasicFunctionality(object):
expected = '/login'
assert url == expected
def test_nonascii_pathinfo(self):
def test_nonascii_pathinfo():
app = flask.Flask(__name__)
app.testing = True
@ -1111,9 +1232,11 @@ class TestBasicFunctionality(object):
rv = c.get(u'/киртест')
assert rv.data == b'Hello World!'
def test_debug_mode_complains_after_first_request(self):
def test_debug_mode_complains_after_first_request():
app = flask.Flask(__name__)
app.debug = True
@app.route('/')
def index():
return 'Awesome'
@ -1129,15 +1252,18 @@ class TestBasicFunctionality(object):
assert False, 'Expected exception'
app.debug = False
@app.route('/foo')
def working():
return 'Meh'
assert app.test_client().get('/foo').data == b'Meh'
assert app.got_first_request
def test_before_first_request_functions(self):
def test_before_first_request_functions():
got = []
app = flask.Flask(__name__)
@app.before_first_request
def foo():
got.append(42)
@ -1148,7 +1274,8 @@ class TestBasicFunctionality(object):
assert got == [42]
assert app.got_first_request
def test_before_first_request_functions_concurrent(self):
def test_before_first_request_functions_concurrent():
got = []
app = flask.Flask(__name__)
@ -1158,6 +1285,7 @@ class TestBasicFunctionality(object):
got.append(42)
c = app.test_client()
def get_and_assert():
c.get("/")
assert got == [42]
@ -1168,9 +1296,11 @@ class TestBasicFunctionality(object):
t.join()
assert app.got_first_request
def test_routing_redirect_debugging(self):
def test_routing_redirect_debugging():
app = flask.Flask(__name__)
app.debug = True
@app.route('/foo/', methods=['GET', 'POST'])
def foo():
return 'success'
@ -1192,7 +1322,8 @@ class TestBasicFunctionality(object):
rv = c.post('/foo', data={}, follow_redirects=True)
assert rv.data == b'success'
def test_route_decorator_custom_endpoint(self):
def test_route_decorator_custom_endpoint():
app = flask.Flask(__name__)
app.debug = True
@ -1218,7 +1349,8 @@ class TestBasicFunctionality(object):
assert c.get('/bar/').data == b'bar'
assert c.get('/bar/123').data == b'123'
def test_preserve_only_once(self):
def test_preserve_only_once():
app = flask.Flask(__name__)
app.debug = True
@ -1238,7 +1370,8 @@ class TestBasicFunctionality(object):
assert flask._request_ctx_stack.top is None
assert flask._app_ctx_stack.top is None
def test_preserve_remembers_exception(self):
def test_preserve_remembers_exception():
app = flask.Flask(__name__)
app.debug = True
errors = []
@ -1270,20 +1403,22 @@ class TestBasicFunctionality(object):
# At this point another request does nothing.
c.get('/success')
assert len(errors) == 3
assert errors[1] == None
assert errors[1] is None
def test_get_method_on_g(self):
def test_get_method_on_g():
app = flask.Flask(__name__)
app.testing = True
with app.app_context():
assert flask.g.get('x') == None
assert flask.g.get('x') is None
assert flask.g.get('x', 11) == 11
flask.g.x = 42
assert flask.g.get('x') == 42
assert flask.g.x == 42
def test_g_iteration_protocol(self):
def test_g_iteration_protocol():
app = flask.Flask(__name__)
app.testing = True
@ -1295,14 +1430,14 @@ class TestBasicFunctionality(object):
assert sorted(flask.g) == ['bar', 'foo']
class TestSubdomain(object):
def test_basic_support(self):
def test_subdomain_basic_support():
app = flask.Flask(__name__)
app.config['SERVER_NAME'] = 'localhost'
@app.route('/')
def normal_index():
return 'normal index'
@app.route('/', subdomain='test')
def test_index():
return 'test index'
@ -1314,9 +1449,11 @@ class TestSubdomain(object):
rv = c.get('/', 'http://test.localhost/')
assert rv.data == b'test index'
def test_subdomain_matching(self):
def test_subdomain_matching():
app = flask.Flask(__name__)
app.config['SERVER_NAME'] = 'localhost'
@app.route('/', subdomain='<user>')
def index(user):
return 'index for %s' % user
@ -1325,9 +1462,11 @@ class TestSubdomain(object):
rv = c.get('/', 'http://mitsuhiko.localhost/')
assert rv.data == b'index for mitsuhiko'
def test_subdomain_matching_with_ports(self):
def test_subdomain_matching_with_ports():
app = flask.Flask(__name__)
app.config['SERVER_NAME'] = 'localhost:3000'
@app.route('/', subdomain='<user>')
def index(user):
return 'index for %s' % user
@ -1336,7 +1475,8 @@ class TestSubdomain(object):
rv = c.get('/', 'http://mitsuhiko.localhost:3000/')
assert rv.data == b'index for mitsuhiko'
def test_multi_route_rules(self):
def test_multi_route_rules():
app = flask.Flask(__name__)
@app.route('/')
@ -1349,8 +1489,10 @@ class TestSubdomain(object):
rv = app.test_client().open('/b/')
assert rv.data == b'b'
def test_multi_route_class_views(self):
def test_multi_route_class_views():
class View(object):
def __init__(self, app):
app.add_url_rule('/', 'index', self.index)
app.add_url_rule('/<test>/', 'index', self.index)

Loading…
Cancel
Save