ccd4c9615c
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
22 lines
639 B
Python
22 lines
639 B
Python
from sanic.response import text
|
|
from sanic.server import HttpProtocol
|
|
|
|
|
|
class CustomHttpProtocol(HttpProtocol):
|
|
def write_response(self, response):
|
|
if isinstance(response, str):
|
|
response = text(response)
|
|
self.transport.write(response.output(self.request.version))
|
|
self.transport.close()
|
|
|
|
|
|
def test_use_custom_protocol(app):
|
|
@app.route("/1")
|
|
async def handler_1(request):
|
|
return "OK"
|
|
|
|
server_kwargs = {"protocol": CustomHttpProtocol}
|
|
request, response = app.test_client.get("/1", server_kwargs=server_kwargs)
|
|
assert response.status == 200
|
|
assert response.text == "OK"
|