summaryrefslogtreecommitdiffstats
path: root/test/wpt/tests/xhr/resources/gzip.py
blob: fd1ca926fb7d6b5ffa292a4bce77a1116803f296 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import gzip as gzip_module

from io import BytesIO

def main(request, response):
    if b"content" in request.GET:
        output = request.GET[b"content"]
    else:
        output = request.body

    out = BytesIO()
    with gzip_module.GzipFile(fileobj=out, mode="w") as f:
        f.write(output)
    output = out.getvalue()

    headers = [(b"Content-type", b"text/plain"),
               (b"Content-Encoding", b"gzip"),
               (b"X-Request-Method", request.method),
               (b"X-Request-Query", request.url_parts.query if request.url_parts.query else b"NO"),
               (b"X-Request-Content-Length", request.headers.get(b"Content-Length", b"NO")),
               (b"X-Request-Content-Type", request.headers.get(b"Content-Type", b"NO")),
               (b"Content-Length", len(output))]

    return headers, output