forked from orbit-oss/flask
Remove class from test_views
This commit is contained in:
Markus Unterwaditzer
parent
7e8f1053bb
commit
e77bc2d86c
1 changed files with 134 additions and 142 deletions
|
|
@ -17,155 +17,147 @@ import unittest
|
||||||
from tests import TestFlask
|
from tests import TestFlask
|
||||||
from werkzeug.http import parse_set_header
|
from werkzeug.http import parse_set_header
|
||||||
|
|
||||||
class TestView(TestFlask):
|
def common_test(app):
|
||||||
|
c = app.test_client()
|
||||||
|
|
||||||
def common_test(self, app):
|
assert c.get('/').data == b'GET'
|
||||||
c = app.test_client()
|
assert c.post('/').data == b'POST'
|
||||||
|
assert c.put('/').status_code == 405
|
||||||
|
meths = parse_set_header(c.open('/', method='OPTIONS').headers['Allow'])
|
||||||
|
assert sorted(meths) == ['GET', 'HEAD', 'OPTIONS', 'POST']
|
||||||
|
|
||||||
assert c.get('/').data == b'GET'
|
def test_basic_view():
|
||||||
assert c.post('/').data == b'POST'
|
app = flask.Flask(__name__)
|
||||||
assert c.put('/').status_code == 405
|
|
||||||
meths = parse_set_header(c.open('/', method='OPTIONS').headers['Allow'])
|
|
||||||
assert sorted(meths) == ['GET', 'HEAD', 'OPTIONS', 'POST']
|
|
||||||
|
|
||||||
def test_basic_view(self):
|
class Index(flask.views.View):
|
||||||
app = flask.Flask(__name__)
|
methods = ['GET', 'POST']
|
||||||
|
def dispatch_request(self):
|
||||||
|
return flask.request.method
|
||||||
|
|
||||||
class Index(flask.views.View):
|
app.add_url_rule('/', view_func=Index.as_view('index'))
|
||||||
methods = ['GET', 'POST']
|
common_test(app)
|
||||||
def dispatch_request(self):
|
|
||||||
return flask.request.method
|
|
||||||
|
|
||||||
app.add_url_rule('/', view_func=Index.as_view('index'))
|
def test_method_based_view():
|
||||||
self.common_test(app)
|
app = flask.Flask(__name__)
|
||||||
|
|
||||||
def test_method_based_view(self):
|
class Index(flask.views.MethodView):
|
||||||
app = flask.Flask(__name__)
|
def get(self):
|
||||||
|
return 'GET'
|
||||||
|
def post(self):
|
||||||
|
return 'POST'
|
||||||
|
|
||||||
class Index(flask.views.MethodView):
|
app.add_url_rule('/', view_func=Index.as_view('index'))
|
||||||
def get(self):
|
|
||||||
return 'GET'
|
|
||||||
def post(self):
|
|
||||||
return 'POST'
|
|
||||||
|
|
||||||
|
common_test(app)
|
||||||
|
|
||||||
|
def test_view_patching():
|
||||||
|
app = flask.Flask(__name__)
|
||||||
|
|
||||||
|
class Index(flask.views.MethodView):
|
||||||
|
def get(self):
|
||||||
|
1 // 0
|
||||||
|
def post(self):
|
||||||
|
1 // 0
|
||||||
|
|
||||||
|
class Other(Index):
|
||||||
|
def get(self):
|
||||||
|
return 'GET'
|
||||||
|
def post(self):
|
||||||
|
return 'POST'
|
||||||
|
|
||||||
|
view = Index.as_view('index')
|
||||||
|
view.view_class = Other
|
||||||
|
app.add_url_rule('/', view_func=view)
|
||||||
|
common_test(app)
|
||||||
|
|
||||||
|
def test_view_inheritance():
|
||||||
|
app = flask.Flask(__name__)
|
||||||
|
|
||||||
|
class Index(flask.views.MethodView):
|
||||||
|
def get(self):
|
||||||
|
return 'GET'
|
||||||
|
def post(self):
|
||||||
|
return 'POST'
|
||||||
|
|
||||||
|
class BetterIndex(Index):
|
||||||
|
def delete(self):
|
||||||
|
return 'DELETE'
|
||||||
|
|
||||||
|
app.add_url_rule('/', view_func=BetterIndex.as_view('index'))
|
||||||
|
c = app.test_client()
|
||||||
|
|
||||||
|
meths = parse_set_header(c.open('/', method='OPTIONS').headers['Allow'])
|
||||||
|
assert sorted(meths) == ['DELETE', 'GET', 'HEAD', 'OPTIONS', 'POST']
|
||||||
|
|
||||||
|
def test_view_decorators():
|
||||||
|
app = flask.Flask(__name__)
|
||||||
|
|
||||||
|
def add_x_parachute(f):
|
||||||
|
def new_function(*args, **kwargs):
|
||||||
|
resp = flask.make_response(f(*args, **kwargs))
|
||||||
|
resp.headers['X-Parachute'] = 'awesome'
|
||||||
|
return resp
|
||||||
|
return new_function
|
||||||
|
|
||||||
|
class Index(flask.views.View):
|
||||||
|
decorators = [add_x_parachute]
|
||||||
|
def dispatch_request(self):
|
||||||
|
return 'Awesome'
|
||||||
|
|
||||||
|
app.add_url_rule('/', view_func=Index.as_view('index'))
|
||||||
|
c = app.test_client()
|
||||||
|
rv = c.get('/')
|
||||||
|
assert rv.headers['X-Parachute'] == 'awesome'
|
||||||
|
assert rv.data == b'Awesome'
|
||||||
|
|
||||||
|
def test_implicit_head():
|
||||||
|
app = flask.Flask(__name__)
|
||||||
|
|
||||||
|
class Index(flask.views.MethodView):
|
||||||
|
def get(self):
|
||||||
|
return flask.Response('Blub', headers={
|
||||||
|
'X-Method': flask.request.method
|
||||||
|
})
|
||||||
|
|
||||||
|
app.add_url_rule('/', view_func=Index.as_view('index'))
|
||||||
|
c = app.test_client()
|
||||||
|
rv = c.get('/')
|
||||||
|
assert rv.data == b'Blub'
|
||||||
|
assert rv.headers['X-Method'] == 'GET'
|
||||||
|
rv = c.head('/')
|
||||||
|
assert rv.data == b''
|
||||||
|
assert rv.headers['X-Method'] == 'HEAD'
|
||||||
|
|
||||||
|
def test_explicit_head():
|
||||||
|
app = flask.Flask(__name__)
|
||||||
|
|
||||||
|
class Index(flask.views.MethodView):
|
||||||
|
def get(self):
|
||||||
|
return 'GET'
|
||||||
|
def head(self):
|
||||||
|
return flask.Response('', headers={'X-Method': 'HEAD'})
|
||||||
|
|
||||||
|
app.add_url_rule('/', view_func=Index.as_view('index'))
|
||||||
|
c = app.test_client()
|
||||||
|
rv = c.get('/')
|
||||||
|
assert rv.data == b'GET'
|
||||||
|
rv = c.head('/')
|
||||||
|
assert rv.data == b''
|
||||||
|
assert rv.headers['X-Method'] == 'HEAD'
|
||||||
|
|
||||||
|
def test_endpoint_override():
|
||||||
|
app = flask.Flask(__name__)
|
||||||
|
app.debug = True
|
||||||
|
|
||||||
|
class Index(flask.views.View):
|
||||||
|
methods = ['GET', 'POST']
|
||||||
|
def dispatch_request(self):
|
||||||
|
return flask.request.method
|
||||||
|
|
||||||
|
app.add_url_rule('/', view_func=Index.as_view('index'))
|
||||||
|
|
||||||
|
with pytest.raises(AssertionError):
|
||||||
app.add_url_rule('/', view_func=Index.as_view('index'))
|
app.add_url_rule('/', view_func=Index.as_view('index'))
|
||||||
|
|
||||||
self.common_test(app)
|
# But these tests should still pass. We just log a warning.
|
||||||
|
common_test(app)
|
||||||
def test_view_patching(self):
|
|
||||||
app = flask.Flask(__name__)
|
|
||||||
|
|
||||||
class Index(flask.views.MethodView):
|
|
||||||
def get(self):
|
|
||||||
1 // 0
|
|
||||||
def post(self):
|
|
||||||
1 // 0
|
|
||||||
|
|
||||||
class Other(Index):
|
|
||||||
def get(self):
|
|
||||||
return 'GET'
|
|
||||||
def post(self):
|
|
||||||
return 'POST'
|
|
||||||
|
|
||||||
view = Index.as_view('index')
|
|
||||||
view.view_class = Other
|
|
||||||
app.add_url_rule('/', view_func=view)
|
|
||||||
self.common_test(app)
|
|
||||||
|
|
||||||
def test_view_inheritance(self):
|
|
||||||
app = flask.Flask(__name__)
|
|
||||||
|
|
||||||
class Index(flask.views.MethodView):
|
|
||||||
def get(self):
|
|
||||||
return 'GET'
|
|
||||||
def post(self):
|
|
||||||
return 'POST'
|
|
||||||
|
|
||||||
class BetterIndex(Index):
|
|
||||||
def delete(self):
|
|
||||||
return 'DELETE'
|
|
||||||
|
|
||||||
app.add_url_rule('/', view_func=BetterIndex.as_view('index'))
|
|
||||||
c = app.test_client()
|
|
||||||
|
|
||||||
meths = parse_set_header(c.open('/', method='OPTIONS').headers['Allow'])
|
|
||||||
assert sorted(meths) == ['DELETE', 'GET', 'HEAD', 'OPTIONS', 'POST']
|
|
||||||
|
|
||||||
def test_view_decorators(self):
|
|
||||||
app = flask.Flask(__name__)
|
|
||||||
|
|
||||||
def add_x_parachute(f):
|
|
||||||
def new_function(*args, **kwargs):
|
|
||||||
resp = flask.make_response(f(*args, **kwargs))
|
|
||||||
resp.headers['X-Parachute'] = 'awesome'
|
|
||||||
return resp
|
|
||||||
return new_function
|
|
||||||
|
|
||||||
class Index(flask.views.View):
|
|
||||||
decorators = [add_x_parachute]
|
|
||||||
def dispatch_request(self):
|
|
||||||
return 'Awesome'
|
|
||||||
|
|
||||||
app.add_url_rule('/', view_func=Index.as_view('index'))
|
|
||||||
c = app.test_client()
|
|
||||||
rv = c.get('/')
|
|
||||||
assert rv.headers['X-Parachute'] == 'awesome'
|
|
||||||
assert rv.data == b'Awesome'
|
|
||||||
|
|
||||||
def test_implicit_head(self):
|
|
||||||
app = flask.Flask(__name__)
|
|
||||||
|
|
||||||
class Index(flask.views.MethodView):
|
|
||||||
def get(self):
|
|
||||||
return flask.Response('Blub', headers={
|
|
||||||
'X-Method': flask.request.method
|
|
||||||
})
|
|
||||||
|
|
||||||
app.add_url_rule('/', view_func=Index.as_view('index'))
|
|
||||||
c = app.test_client()
|
|
||||||
rv = c.get('/')
|
|
||||||
assert rv.data == b'Blub'
|
|
||||||
assert rv.headers['X-Method'] == 'GET'
|
|
||||||
rv = c.head('/')
|
|
||||||
assert rv.data == b''
|
|
||||||
assert rv.headers['X-Method'] == 'HEAD'
|
|
||||||
|
|
||||||
def test_explicit_head(self):
|
|
||||||
app = flask.Flask(__name__)
|
|
||||||
|
|
||||||
class Index(flask.views.MethodView):
|
|
||||||
def get(self):
|
|
||||||
return 'GET'
|
|
||||||
def head(self):
|
|
||||||
return flask.Response('', headers={'X-Method': 'HEAD'})
|
|
||||||
|
|
||||||
app.add_url_rule('/', view_func=Index.as_view('index'))
|
|
||||||
c = app.test_client()
|
|
||||||
rv = c.get('/')
|
|
||||||
assert rv.data == b'GET'
|
|
||||||
rv = c.head('/')
|
|
||||||
assert rv.data == b''
|
|
||||||
assert rv.headers['X-Method'] == 'HEAD'
|
|
||||||
|
|
||||||
def test_endpoint_override(self):
|
|
||||||
app = flask.Flask(__name__)
|
|
||||||
app.debug = True
|
|
||||||
|
|
||||||
class Index(flask.views.View):
|
|
||||||
methods = ['GET', 'POST']
|
|
||||||
def dispatch_request(self):
|
|
||||||
return flask.request.method
|
|
||||||
|
|
||||||
app.add_url_rule('/', view_func=Index.as_view('index'))
|
|
||||||
|
|
||||||
with pytest.raises(AssertionError):
|
|
||||||
app.add_url_rule('/', view_func=Index.as_view('index'))
|
|
||||||
|
|
||||||
# But these tests should still pass. We just log a warning.
|
|
||||||
self.common_test(app)
|
|
||||||
|
|
||||||
|
|
||||||
def suite():
|
|
||||||
suite = unittest.TestSuite()
|
|
||||||
suite.addTest(unittest.makeSuite(TestView))
|
|
||||||
return suite
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue