sanic/tests/test_request_data.py
Adam Hopkins ccd4c9615c Create requests-async based TestClient, remove aiohttp dependency, drop Python 3.5
Update all tests to be compatible with requests-async
Cleanup testing client changes with black and isort
Remove Python 3.5 and other meta doc cleanup
rename pyproject and fix pep517 error
Add black config to tox.ini
Cleanup tests and remove aiohttp
tox.ini change for easier development commands
Remove aiohttp from changelog and requirements
Cleanup imports and Makefile
2019-04-30 15:26:06 +03:00

47 lines
1.1 KiB
Python

import random
from sanic.response import json
try:
from ujson import loads
except ImportError:
from json import loads
def test_storage(app):
@app.middleware("request")
def store(request):
request["user"] = "sanic"
request["sidekick"] = "tails"
del request["sidekick"]
@app.route("/")
def handler(request):
return json(
{"user": request.get("user"), "sidekick": request.get("sidekick")}
)
request, response = app.test_client.get("/")
response_json = loads(response.text)
assert response_json["user"] == "sanic"
assert response_json.get("sidekick") is None
def test_app_injection(app):
expected = random.choice(range(0, 100))
@app.listener("after_server_start")
async def inject_data(app, loop):
app.injected = expected
@app.get("/")
async def handler(request):
return json({"injected": request.app.injected})
request, response = app.test_client.get("/")
response_json = loads(response.text)
assert response_json["injected"] == expected