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