diff options
author | Andrei Zeliankou <zelenkov@nginx.com> | 2024-01-26 14:58:43 +0000 |
---|---|---|
committer | andrey-zelenkov <xim.andrew@gmail.com> | 2024-02-21 14:06:43 +0000 |
commit | 697a58506235e89af1c8cc3cafc92b3d85a3467d (patch) | |
tree | 67d8cf28089e947ed59d00c24df934a8a9eea956 /test | |
parent | f71ead5fa5b8bae378a0eca2bedb689cf08a8eff (diff) | |
download | unit-697a58506235e89af1c8cc3cafc92b3d85a3467d.tar.gz unit-697a58506235e89af1c8cc3cafc92b3d85a3467d.tar.bz2 |
Python: bytearray body support for ASGI module.
@filiphanes requested support for bytearray
and memoryview in the request body here:
<https://github.com/nginx/unit/issues/648>
This patch implements bytearray body support only.
Memoryview body still need to be implemented.
Diffstat (limited to 'test')
-rw-r--r-- | test/python/body_bytearray/asgi.py | 20 | ||||
-rw-r--r-- | test/test_asgi_application.py | 8 |
2 files changed, 28 insertions, 0 deletions
diff --git a/test/python/body_bytearray/asgi.py b/test/python/body_bytearray/asgi.py new file mode 100644 index 00000000..6d2f402f --- /dev/null +++ b/test/python/body_bytearray/asgi.py @@ -0,0 +1,20 @@ +async def application(scope, receive, send): + assert scope['type'] == 'http' + + body = b'' + while True: + m = await receive() + body += m.get('body', b'') + if not m.get('more_body', False): + body = bytearray(body) + break + + await send( + { + 'type': 'http.response.start', + 'status': 200, + 'headers': [(b'content-length', str(len(body)).encode())], + } + ) + + await send({'type': 'http.response.body', 'body': body}) diff --git a/test/test_asgi_application.py b/test/test_asgi_application.py index e6668b2f..226a1ed7 100644 --- a/test/test_asgi_application.py +++ b/test/test_asgi_application.py @@ -218,6 +218,14 @@ def test_asgi_application_shm_ack_handle(): assert resp['body'] == body, 'keep-alive 1' +def test_asgi_application_body_bytearray(): + client.load('body_bytearray') + + body = '0123456789' + + assert client.post(body=body)['body'] == body + + def test_asgi_keepalive_body(): client.load('mirror') |