From 88ac538b41a4753c3de9b509c3a0e13077f66182 Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Wed, 2 Dec 2020 15:47:37 -0800
Subject: [PATCH] transfer-encoding: chunked for DB downloads, refs #749

This should get >32MB downloads working on Cloud Run.
---
 datasette/views/database.py | 1 +
 tests/test_html.py          | 1 +
 2 files changed, 2 insertions(+)

diff --git a/datasette/views/database.py b/datasette/views/database.py
index 17c78150..f6fd579c 100644
--- a/datasette/views/database.py
+++ b/datasette/views/database.py
@@ -169,6 +169,7 @@ class DatabaseDownload(DataView):
         headers = {}
         if self.ds.cors:
             headers["Access-Control-Allow-Origin"] = "*"
+        headers["Transfer-Encoding"] = "chunked"
         return AsgiFileDownload(
             filepath,
             filename=os.path.basename(filepath),
diff --git a/tests/test_html.py b/tests/test_html.py
index ecbf89b4..b9d3afcd 100644
--- a/tests/test_html.py
+++ b/tests/test_html.py
@@ -1028,6 +1028,7 @@ def test_database_download_for_immutable():
             download_response.headers["content-disposition"]
             == 'attachment; filename="fixtures.db"'
         )
+        assert download_response.headers["transfer-encoding"] == "chunked"
 
 
 def test_database_download_disallowed_for_mutable(app_client):