starlette/tests/test_background.py

54 lines
1.2 KiB
Python
Raw Normal View History

import asyncio
from starlette.background import BackgroundTask
from starlette.responses import Response
from starlette.testclient import TestClient
def test_async_task():
2018-10-02 11:29:44 +00:00
TASK_COMPLETE = False
async def async_task():
2018-10-02 11:29:44 +00:00
nonlocal TASK_COMPLETE
TASK_COMPLETE = True
task = BackgroundTask(async_task)
def app(scope):
async def asgi(receive, send):
response = Response(
"task initiated", media_type="text/plain", background=task
)
await response(receive, send)
return asgi
client = TestClient(app)
response = client.get("/")
assert response.text == "task initiated"
2018-10-02 11:29:44 +00:00
assert TASK_COMPLETE
def test_sync_task():
2018-10-02 11:29:44 +00:00
TASK_COMPLETE = False
def sync_task():
2018-10-02 11:29:44 +00:00
nonlocal TASK_COMPLETE
TASK_COMPLETE = True
task = BackgroundTask(sync_task)
def app(scope):
async def asgi(receive, send):
response = Response(
"task initiated", media_type="text/plain", background=task
)
await response(receive, send)
return asgi
client = TestClient(app)
response = client.get("/")
assert response.text == "task initiated"
2018-10-02 11:29:44 +00:00
assert TASK_COMPLETE