From 7537a300c0f0b465507d1b36d8383c17ec1da465 Mon Sep 17 00:00:00 2001 From: Athanasius Date: Tue, 11 Jan 2022 16:31:05 +0000 Subject: [PATCH] scripts/test-sender: Support for valid gzip compression --- scripts/testing/gateway-responses/test-sender.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/scripts/testing/gateway-responses/test-sender.py b/scripts/testing/gateway-responses/test-sender.py index d5ac96b..52e7aca 100644 --- a/scripts/testing/gateway-responses/test-sender.py +++ b/scripts/testing/gateway-responses/test-sender.py @@ -3,6 +3,7 @@ import argparse import requests +import zlib upload_url = 'https://dev.eddn.edcd.io:4432/upload/' @@ -25,6 +26,11 @@ send_message: s = requests.Session() + if args.gzip: + if args.gzip == 'good': + msg = zlib.compress(msg.encode('utf-8')) + s.headers['Content-Encoding'] = 'gzip' + r = s.post(upload_url, data=msg) print(f'Response: {r!r}') @@ -47,6 +53,12 @@ if __name__ == "__main__": help='Specify to form-encode the request body', ) + __parser.add_argument( + '--gzip', + choices=('good'), + help='Specify to gzip-compress the request body', + ) + __parser.add_argument( 'messagefile', metavar='',