summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/fetch/api/resources/huge-response.py
blob: 16a60078e53f5a9deba88283eecd2ce204467562 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# A Python script that generates a huge response. Implemented as a script to
# avoid needing to add a huge file to the repository.

TOTAL_SIZE = 8 * 1024 * 1024 * 1024  # 8 GB
CHUNK_SIZE = 1024 * 1024  # 1 MB

assert TOTAL_SIZE % CHUNK_SIZE == 0


def main(request, response):
    response.headers.set(b"Content-type", b"text/plain")
    response.headers.set(b"Content-Length", str(TOTAL_SIZE).encode())
    response.headers.set(b"Cache-Control", b"max-age=86400")
    response.write_status_headers()

    chunk = bytes(CHUNK_SIZE)
    total_sent = 0

    while total_sent < TOTAL_SIZE:
        if not response.writer.write(chunk):
            break
        total_sent += CHUNK_SIZE