import json import pytest import sqlite3 from datasette.app import Datasette from .fixtures import TestClient PLUGIN = """ from datasette import hookimpl @hookimpl def extra_template_vars(): print("this is template vars") return { "from_plugin": "hooray" } """ METADATA = {"title": "This is from metadata"} CONFIG = { "default_cache_ttl": 60, "allow_sql": False, } CSS = """ body { margin-top: 3em} """ @pytest.fixture(scope="session") def config_dir_client(tmp_path_factory): config_dir = tmp_path_factory.mktemp("config-dir") plugins_dir = config_dir / "plugins" plugins_dir.mkdir() (plugins_dir / "hooray.py").write_text(PLUGIN, "utf-8") templates_dir = config_dir / "templates" templates_dir.mkdir() (templates_dir / "row.html").write_text( "Show row here. Plugin says {{ from_plugin }}", "utf-8" ) static_dir = config_dir / "static" static_dir.mkdir() (static_dir / "hello.css").write_text(CSS, "utf-8") (config_dir / "metadata.json").write_text(json.dumps(METADATA), "utf-8") (config_dir / "config.json").write_text(json.dumps(CONFIG), "utf-8") for dbname in ("demo.db", "immutable.db"): db = sqlite3.connect(str(config_dir / dbname)) db.executescript( """ CREATE TABLE cities ( id integer primary key, name text ); INSERT INTO cities (id, name) VALUES (1, 'San Francisco') ; """ ) # Mark "immutable.db" as immutable (config_dir / "inspect-data.json").write_text( json.dumps( { "immutable": { "hash": "hash", "size": 8192, "file": "immutable.db", "tables": {"cities": {"count": 1}}, } } ), "utf-8", ) ds = Datasette([], config_dir=config_dir) client = TestClient(ds.app()) client.ds = ds yield client def test_metadata(config_dir_client): response = config_dir_client.get("/-/metadata.json") assert 200 == response.status assert METADATA == json.loads(response.text) def test_config(config_dir_client): response = config_dir_client.get("/-/config.json") assert 200 == response.status config = json.loads(response.text) assert 60 == config["default_cache_ttl"] assert not config["allow_sql"] def test_plugins(config_dir_client): response = config_dir_client.get("/-/plugins.json") assert 200 == response.status assert "hooray.py" in {p["name"] for p in json.loads(response.text)} def test_templates_and_plugin(config_dir_client): response = config_dir_client.get("/demo/cities/1") assert 200 == response.status assert "Show row here. Plugin says hooray" == response.text def test_static(config_dir_client): response = config_dir_client.get("/static/hello.css") assert 200 == response.status assert CSS == response.text assert "text/css" == response.headers["content-type"] def test_databases(config_dir_client): response = config_dir_client.get("/-/databases.json") assert 200 == response.status databases = json.loads(response.text) assert 2 == len(databases) databases.sort(key=lambda d: d["name"]) assert "demo" == databases[0]["name"] assert databases[0]["is_mutable"] assert "immutable" == databases[1]["name"] assert not databases[1]["is_mutable"]