82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
# -*- 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'
|
|
|
|
def test_url_generation(self):
|
|
app = flask.Flask(__name__)
|
|
@app.route('/hello/<name>', methods=['POST'])
|
|
def hello():
|
|
pass
|
|
with app.test_request_context():
|
|
assert flask.url_for('hello', name='test x') == '/hello/test%20x'
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|