From 9f6dd985bc0eff70f8a9ce65c6578bc43d2e172b Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Fri, 9 Oct 2020 17:39:45 -0700
Subject: [PATCH] Fix broken CSV/JSON export on query page, refs #1010

---
 datasette/templates/query.html | 2 +-
 tests/test_html.py             | 7 +++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/datasette/templates/query.html b/datasette/templates/query.html
index 0add74a8..c6574f31 100644
--- a/datasette/templates/query.html
+++ b/datasette/templates/query.html
@@ -58,7 +58,7 @@
 </form>
 
 {% if display_rows %}
-  <p class="export-links">This data as {% for name, url in renderers.items() %}<a href="{{ base_url }}{{ url }}">{{ name }}</a>{{ ", " if not loop.last }}{% endfor %}, <a href="{{ base_url }}{{ url_csv }}">CSV</a></p>
+  <p class="export-links">This data as {% for name, url in renderers.items() %}<a href="{{ url }}">{{ name }}</a>{{ ", " if not loop.last }}{% endfor %}, <a href="{{ url_csv }}">CSV</a></p>
 <table class="rows-and-columns">
     <thead>
         <tr>
diff --git a/tests/test_html.py b/tests/test_html.py
index aca4eedd..3f8cb178 100644
--- a/tests/test_html.py
+++ b/tests/test_html.py
@@ -672,6 +672,13 @@ def test_row_json_export_link(app_client):
     assert '<a href="/fixtures/simple_primary_key/1.json">json</a>' in response.text
 
 
+def test_query_json_csv_export_links(app_client):
+    response = app_client.get("/fixtures?sql=select+1")
+    assert response.status == 200
+    assert '<a href="/fixtures.json?sql=select+1">json</a>' in response.text
+    assert '<a href="/fixtures.csv?sql=select+1&amp;_size=max">CSV</a>' in response.text
+
+
 def test_csv_json_export_links_include_labels_if_foreign_keys(app_client):
     response = app_client.get("/fixtures/facetable")
     assert response.status == 200