From a2a5dff709c6f1676ac30b5e734c2763002562cf Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Thu, 27 Oct 2022 12:08:26 -0700
Subject: [PATCH] Missing tests for insert row API, refs #1851

---
 tests/test_api_write.py | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 tests/test_api_write.py

diff --git a/tests/test_api_write.py b/tests/test_api_write.py
new file mode 100644
index 00000000..86c221d0
--- /dev/null
+++ b/tests/test_api_write.py
@@ -0,0 +1,38 @@
+from datasette.app import Datasette
+from datasette.utils import sqlite3
+import pytest
+import time
+
+
+@pytest.fixture
+def ds_write(tmp_path_factory):
+    db_directory = tmp_path_factory.mktemp("dbs")
+    db_path = str(db_directory / "data.db")
+    db = sqlite3.connect(str(db_path))
+    db.execute("vacuum")
+    db.execute("create table docs (id integer primary key, title text, score float)")
+    ds = Datasette([db_path])
+    yield ds
+    db.close()
+
+
+@pytest.mark.asyncio
+async def test_write_row(ds_write):
+    token = "dstok_{}".format(
+        ds_write.sign(
+            {"a": "root", "token": "dstok", "t": int(time.time())}, namespace="token"
+        )
+    )
+    response = await ds_write.client.post(
+        "/data/docs",
+        json={"insert": {"title": "Test", "score": 1.0}},
+        headers={
+            "Authorization": "Bearer {}".format(token),
+            "Content-Type": "application/json",
+        },
+    )
+    expected_row = {"id": 1, "title": "Test", "score": 1.0}
+    assert response.status_code == 201
+    assert response.json()["inserted_row"] == expected_row
+    rows = (await ds_write.get_database("data").execute("select * from docs")).rows
+    assert dict(rows[0]) == expected_row