mirror of https://github.com/mitsuhiko/flask.git
Armin Ronacher
15 years ago
1 changed files with 74 additions and 0 deletions
@ -0,0 +1,74 @@ |
|||||||
|
# -*- coding: utf-8 -*- |
||||||
|
""" |
||||||
|
Flask Tests |
||||||
|
~~~~~~~~~~~ |
||||||
|
|
||||||
|
Tests Flask itself. The majority of Flask is already tested |
||||||
|
as part of Werkzeug. |
||||||
|
|
||||||
|
:copyright: (c) 2010 by Armin Ronacher. |
||||||
|
:license: BSD, see LICENSE for more details. |
||||||
|
""" |
||||||
|
from __future__ import with_statement |
||||||
|
import flask |
||||||
|
import unittest |
||||||
|
import tempfile |
||||||
|
|
||||||
|
|
||||||
|
class ContextTestCase(unittest.TestCase): |
||||||
|
|
||||||
|
def test_context_binding(self): |
||||||
|
app = flask.Flask(__name__) |
||||||
|
@app.route('/') |
||||||
|
def index(): |
||||||
|
return 'Hello %s!' % flask.request.args['name'] |
||||||
|
@app.route('/meh') |
||||||
|
def meh(): |
||||||
|
return flask.request.url |
||||||
|
|
||||||
|
with app.test_request_context('/?name=World'): |
||||||
|
assert index() == 'Hello World!' |
||||||
|
with app.test_request_context('/meh'): |
||||||
|
assert meh() == 'http://localhost/meh' |
||||||
|
|
||||||
|
def test_request_dispatching(self): |
||||||
|
app = flask.Flask(__name__) |
||||||
|
@app.route('/') |
||||||
|
def index(): |
||||||
|
return flask.request.method |
||||||
|
@app.route('/more', methods=['GET', 'POST']) |
||||||
|
def more(): |
||||||
|
return flask.request.method |
||||||
|
|
||||||
|
c = app.test_client() |
||||||
|
assert c.get('/').data == 'GET' |
||||||
|
rv = c.post('/') |
||||||
|
assert rv.status_code == 405 |
||||||
|
assert sorted(rv.allow) == ['GET', 'HEAD'] |
||||||
|
rv = c.head('/') |
||||||
|
assert rv.status_code == 200 |
||||||
|
assert not rv.data # head truncates |
||||||
|
assert c.post('/more').data == 'POST' |
||||||
|
assert c.get('/more').data == 'GET' |
||||||
|
rv = c.delete('/more') |
||||||
|
assert rv.status_code == 405 |
||||||
|
assert sorted(rv.allow) == ['GET', 'HEAD', 'POST'] |
||||||
|
|
||||||
|
def test_session(self): |
||||||
|
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'] |
||||||
|
|
||||||
|
c = app.test_client() |
||||||
|
assert c.post('/set', data={'value': '42'}).data == 'value set' |
||||||
|
assert c.get('/get').data == '42' |
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
unittest.main() |
Loading…
Reference in new issue