From feddd61789983dc75f127956176cac493d877be8 Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Tue, 12 Mar 2024 17:01:51 -0700
Subject: [PATCH] Fix tests I broke in #2302

---
 datasette/templates/patterns.html | 2 +-
 tests/test_auth.py                | 7 +++++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/datasette/templates/patterns.html b/datasette/templates/patterns.html
index 33db3d1a..cb0daf9a 100644
--- a/datasette/templates/patterns.html
+++ b/datasette/templates/patterns.html
@@ -26,7 +26,7 @@
                 <li><a href="/-/plugins">Installed plugins</a></li>
                 <li><a href="/-/versions">Version info</a></li>
             </ul>
-            <form action="/-/logout" method="post">
+            <form class="nav-menu-logout" action="/-/logout" method="post">
                 <button class="button-as-link">Log out</button>
             </form>
         </div>
diff --git a/tests/test_auth.py b/tests/test_auth.py
index f2359df7..74e8283a 100644
--- a/tests/test_auth.py
+++ b/tests/test_auth.py
@@ -110,7 +110,7 @@ async def test_logout_button_in_navigation(ds_client, path):
     anon_response = await ds_client.get(path)
     for fragment in (
         "<strong>test</strong>",
-        '<form action="/-/logout" method="post">',
+        '<form class="nav-menu-logout" action="/-/logout" method="post">',
     ):
         assert fragment in response.text
         assert fragment not in anon_response.text
@@ -121,7 +121,10 @@ async def test_logout_button_in_navigation(ds_client, path):
 async def test_no_logout_button_in_navigation_if_no_ds_actor_cookie(ds_client, path):
     response = await ds_client.get(path + "?_bot=1")
     assert "<strong>bot</strong>" in response.text
-    assert '<form action="/-/logout" method="post">' not in response.text
+    assert (
+        '<form class="nav-menu-logout" action="/-/logout" method="post">'
+        not in response.text
+    )
 
 
 @pytest.mark.parametrize(