Como todos sabem que é uma boa prática fazer testes unitários em aplicações.

Uma Api Python usando Flask não seria diferente.

Para começar a demonstrar, vamos criar uma rota de exemplo, para isso vamos criar um arquivo rotas.py

from flask import Flask

app = Flask(__name__)

@app.route('/')
def ola():
    return 'Olá sou a rota do base usando FLASK!'

agora com a nossa rota de exemplo criada, vamos criar o nosso test. Então vamos criar outro arquivo test_rotas.py

from rotas import app

def test_ola():
    response = app.test_client().get('/')

    assert response.status_code == 200
    assert response.data == b'Olá sou a rota do base usando FLASK!'

Com isso é possível fazer os testes unitários em rotas criadas com Flask.

Caso a rota retorne um Json, como no exemplo abaixo.

@app.route('/add', methods=['POST'])
def add():
    data = request.get_json()
    return jsonify({'sum': data['a'] + data['b']})

Para testarmos o retorno é bem simples

def test_add():        
    response = app.test_client().post(
        '/add',
        data=json.dumps({'a': 1, 'b': 2}),
        content_type='application/json',
    )

    data = json.loads(response.get_data(as_text=True))

    assert response.status_code == 200
    assert data['sum'] == 3