Solution: Random redirect

• random.choice

from flask import Flask, redirect
import random

app = Flask(__name__)

urls = [
    'https://code-maven.com/',
    'https://perlmaven.com/',
    'https://hostlocal.com/',
    'https://pydigger.com/',
    'https://szabgab.com/',
]

@app.get('/')
def main_page():
    return '<a href="/random">Random</a>'

@app.get('/random')
def random_redirect():
    return redirect(random.choice(urls))

import random_redirect
import pytest

@pytest.fixture()
def web():
    return random_redirect.app.test_client()

def test_main_page(web):
    rv = web.get('/')
    assert rv.status == '200 OK'
    assert rv.data == b'<a href="/random">Random</a>'

def test_random(web):
    rv = web.get('/random')
    assert rv.status == '302 FOUND'
    assert 'https://' in rv.headers['Location']
    print(rv.headers['Location'])
    assert rv.headers['Location'] in random_redirect.urls