Testes Assíncronos¶
Você já viu como testar as suas aplicações FastAPI utilizando o TestClient
que é fornecido. Até agora, você viu apenas como escrever testes síncronos, sem utilizar funções async
.
Ser capaz de utilizar funções assíncronas em seus testes pode ser útil, por exemplo, quando você está realizando uma consulta em seu banco de dados de maneira assíncrona. Imagine que você deseja testar realizando requisições para a sua aplicação FastAPI e depois verificar que a sua aplicação inseriu corretamente as informações no banco de dados, ao utilizar uma biblioteca assíncrona para banco de dados.
Vamos ver como nós podemos fazer isso funcionar.
pytest.mark.anyio¶
Se quisermos chamar funções assíncronas em nossos testes, as nossas funções de teste precisam ser assíncronas. O AnyIO oferece um plugin bem legal para isso, que nos permite especificar que algumas das nossas funções de teste precisam ser chamadas de forma assíncrona.
HTTPX¶
Mesmo que a sua aplicação FastAPI utilize funções normais com def
no lugar de async def
, ela ainda é uma aplicação async
por baixo dos panos.
O TestClient
faz algumas mágicas para invocar a aplicação FastAPI assíncrona em suas funções def
normais, utilizando o pytest padrão. Porém a mágica não acontece mais quando nós estamos utilizando dentro de funções assíncronas. Ao executar os nossos testes de forma assíncrona, nós não podemos mais utilizar o TestClient
dentro das nossas funções de teste.
O TestClient
é baseado no HTTPX, e felizmente nós podemos utilizá-lo diretamente para testar a API.
Exemplo¶
Para um exemplos simples, vamos considerar uma estrutura de arquivos semelhante ao descrito em Bigger Applications e Testing:
.
├── app
│ ├── __init__.py
│ ├── main.py
│ └── test_main.py
O arquivo main.py
teria:
from fastapi import FastAPI
app = FastAPI()
@app.get("/")
async def root():
return {"message": "Tomato"}
O arquivo test_main.py
teria os testes para para o arquivo main.py
, ele poderia ficar assim:
import pytest
from httpx import ASGITransport, AsyncClient
from .main import app
@pytest.mark.anyio
async def test_root():
async with AsyncClient(
transport=ASGITransport(app=app), base_url="http://test"
) as ac:
response = await ac.get("/")
assert response.status_code == 200
assert response.json() == {"message": "Tomato"}
Executá-lo¶
Você pode executar os seus testes normalmente via:
$ pytest
---> 100%
Em Detalhes¶
O marcador @pytest.mark.anyio
informa ao pytest que esta função de teste deve ser invocada de maneira assíncrona:
import pytest
from httpx import ASGITransport, AsyncClient
from .main import app
@pytest.mark.anyio
async def test_root():
async with AsyncClient(
transport=ASGITransport(app=app), base_url="http://test"
) as ac:
response = await ac.get("/")
assert response.status_code == 200
assert response.json() == {"message": "Tomato"}
"Dica"
Note que a função de teste é async def
agora, no lugar de apenas def
como quando estávamos utilizando o TestClient
anteriormente.
Então podemos criar um AsyncClient
com a aplicação, e enviar requisições assíncronas para ela utilizando await
.
import pytest
from httpx import ASGITransport, AsyncClient
from .main import app
@pytest.mark.anyio
async def test_root():
async with AsyncClient(
transport=ASGITransport(app=app), base_url="http://test"
) as ac:
response = await ac.get("/")
assert response.status_code == 200
assert response.json() == {"message": "Tomato"}
Isso é equivalente a:
response = client.get('/')
...que nós utilizamos para fazer as nossas requisições utilizando o TestClient
.
"Dica"
Note que nós estamos utilizando async/await com o novo AsyncClient
- a requisição é assíncrona.
"Aviso"
Se a sua aplicação depende dos eventos de vida útil (lifespan), o AsyncClient
não acionará estes eventos. Para garantir que eles são acionados, utilize o LifespanManager
do florimondmanca/asgi-lifespan.
Outras Chamadas de Funções Assíncronas¶
Como a função de teste agora é assíncrona, você pode chamar (e esperar
) outras funções async
além de enviar requisições para a sua aplicação FastAPI em seus testes, exatamente como você as chamaria em qualquer outro lugar do seu código.
"Dica"
Se você se deparar com um RuntimeError: Task attached to a different loop
ao integrar funções assíncronas em seus testes (e.g. ao utilizar o MotorClient do MongoDB) Lembre-se de instanciar objetos que precisam de um loop de eventos (event loop) apenas em funções assíncronas, e.g. um "callback" '@app.on_event("startup")
.