third-party-mirrors/EDDN
1
0
Fork 0
mirror of https://github.com/EDCD/EDDN.git synced 2025-04-20 18:47:39 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

scripts/test-sender: Support --gzip=bad

Browse Source
This commit is contained in:
Athanasius 2022-01-11 16:41:25 +00:00
parent 7537a300c0
commit ad8fc57df3
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
scripts/testing/gateway-responses/test-sender.py
View File

@ -27,9 +27,14 @@ send_message:
s = requests.Session()
if args.gzip:
if args.gzip == 'good':
msg = zlib.compress(msg.encode('utf-8'))
s.headers['Content-Encoding'] = 'gzip'
# We assume that the argparse setup is enforcing the value being
# valid, i.e. `'good'` if it's not `'bad'`.
msg = zlib.compress(msg.encode('utf-8'))
s.headers['Content-Encoding'] = 'gzip'
if args.gzip == 'bad':
# Prepend a character so it's not a valid gzip header
msg = b'w' + msg
r = s.post(upload_url, data=msg)
@ -55,7 +60,7 @@ if __name__ == "__main__":
__parser.add_argument(
'--gzip',
choices=('good'),
choices=('good', 'bad'),
help='Specify to gzip-compress the request body',
)