Flask Redirect

from flask import Flask, redirect

app = Flask(__name__)

@app.get('/')
def main_page():
    return '<a href="/cm">Go to Code Maven</a>'

@app.get('/cm')
def cm():
    return redirect('https://code-maven.com/')

import redirect

def test_main_page():
    web = redirect.app.test_client()

    rv = web.get('/')
    assert rv.status == '200 OK'
    assert rv.data == b'<a href="/cm">Go to Code Maven</a>'

def test_redirect():
    web = redirect.app.test_client()

    rv = web.get('/cm')
    assert rv.status == '302 FOUND'
    assert rv.headers['Location'] == 'https://code-maven.com/'
    assert b'<p>You should be redirected automatically to the target URL: <a href="https://code-maven.com/">https://code-maven.com/</a>' in rv.data