Pass view_name=page to custom pages, refs #648

This commit is contained in:
Simon Willison 2020-04-26 10:49:57 -07:00
commit b5509a0b0a
3 changed files with 26 additions and 5 deletions

View file

@ -11,6 +11,7 @@ import json
import pathlib
import pytest
import re
import textwrap
import urllib.parse
@ -1243,16 +1244,31 @@ def test_base_url_config(base_url, path):
}
def test_custom_template_page(tmpdir):
def test_custom_pages(tmpdir):
template_dir = tmpdir.mkdir("page-templates")
extra_plugins = {
"view_name.py": textwrap.dedent(
"""
from datasette import hookimpl
@hookimpl
def extra_template_vars(view_name):
return {
"view_name": view_name,
}
"""
)
}
pages_dir = template_dir.mkdir("pages")
(pages_dir / "about.html").write_text("ABOUT!", "utf-8")
(pages_dir / "about.html").write_text("ABOUT! view_name:{{ view_name }}", "utf-8")
nested_dir = pages_dir.mkdir("nested")
(nested_dir / "nest.html").write_text("Nest!", "utf-8")
for client in make_app_client(template_dir=str(template_dir)):
for client in make_app_client(
template_dir=str(template_dir), extra_plugins=extra_plugins
):
response = client.get("/about")
assert 200 == response.status
assert "ABOUT!" == response.text
assert "ABOUT! view_name:page" == response.text
response = client.get("/nested/nest")
assert 200 == response.status
assert "Nest!" == response.text