Create saved_queries table, refs #1

datasette_saved_queries/__init__.py

@@ -1 +1,19 @@
 from datasette import hookimpl
+import sqlite_utils
+
+
+def create_tables(conn):
+    db = sqlite_utils.Database(conn)
+    if not db["saved_queries"].exists():
+        db["saved_queries"].create(
+            {"name": str, "sql": str, "author_id": str,}, pk="name"
+        )
+
+
+@hookimpl
+def startup(datasette):
+    async def inner():
+        db = datasette.get_database()
+        await db.execute_write_fn(create_tables, block=True)
+
+    return inner

setup.py

@@ -28,9 +28,7 @@ def get_long_description():
     version=VERSION,
     packages=["datasette_saved_queries"],
     entry_points={"datasette": ["saved_queries = datasette_saved_queries"]},
-    install_requires=["datasette"],
-    extras_require={
-        "test": ["pytest", "pytest-asyncio", "httpx"]
-    },
+    install_requires=["datasette>=0.45a1", "sqlite-utils"],
+    extras_require={"test": ["pytest", "pytest-asyncio", "httpx"]},
     tests_require=["datasette-saved-queries[test]"],
 )

tests/test_saved_queries.py

@@ -5,9 +5,21 @@
 
 @pytest.mark.asyncio
 async def test_plugin_is_installed():
-    app = Datasette([], memory=True,).app()
+    app = Datasette([], memory=True).app()
     async with httpx.AsyncClient(app=app) as client:
         response = await client.get("http://localhost/-/plugins.json")
         assert 200 == response.status_code
         installed_plugins = {p["name"] for p in response.json()}
         assert "datasette-saved-queries" in installed_plugins
+
+
+@pytest.mark.asyncio
+async def test_table_created(tmpdir):
+    data = tmpdir / "data.db"
+    ds = Datasette([data])
+    await ds.invoke_startup()
+    app = ds.app()
+    async with httpx.AsyncClient(app=app) as client:
+        response = await client.get("http://localhost/data/saved_queries.json")
+        assert 200 == response.status_code
+        assert ["name", "sql", "author_id"] == response.json()["columns"]