From 818b0b76a2d58f7c2d850570efcdc22d345b4059 Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Sun, 8 Aug 2021 16:07:52 -0700
Subject: [PATCH] Test table render_cell async as well as query results, refs
 #1425

---
 tests/test_plugins.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/tests/test_plugins.py b/tests/test_plugins.py
index 9bda7420..ec8ff0c5 100644
--- a/tests/test_plugins.py
+++ b/tests/test_plugins.py
@@ -185,8 +185,11 @@ def test_hook_render_cell_demo(app_client):
     } == json.loads(td.string)
 
 
-def test_hook_render_cell_async(app_client):
-    response = app_client.get("/fixtures?sql=select+'RENDER_CELL_ASYNC'")
+@pytest.mark.parametrize(
+    "path", ("/fixtures?sql=select+'RENDER_CELL_ASYNC'", "/fixtures/simple_primary_key")
+)
+def test_hook_render_cell_async(app_client, path):
+    response = app_client.get(path)
     assert b"RENDER_CELL_ASYNC_RESULT" in response.body