from flask import Flask
app = Flask(__name__)
@app.route("/")
def index():
return '''
Main<br>
<a href="/user/23">23</a><br>
<a href="/user/42">42</a><br>
<a href="/user/Joe">Joe</a><br>
'''
@app.route("/user/", defaults={ 'uid': 'zero' })
@app.route("/user/<uid>")
def api_info(uid):
return uid
import app
import pytest
@pytest.fixture()
def web():
return app.app.test_client()
def test_app(web):
rv = web.get('/')
assert rv.status == '200 OK'
assert b'Main<br>' in rv.data
@pytest.mark.parametrize('uid', ['23', '42', 'Joe'])
def test_user(web, uid):
rv = web.get(f'/user/{uid}')
assert rv.status == '200 OK'
assert uid == rv.data.decode('utf-8')
def test_user_root_slash(web):
rv = web.get(f'/user/')
assert rv.status == '200 OK'
assert b'zero' == rv.data
def test_user_root(web):
rv = web.get(f'/user')
assert rv.status == '308 PERMANENT REDIRECT'
assert rv.headers['Location'] == 'http://localhost/user/'
assert b'<p>You should be redirected automatically to target URL:' in rv.data