Update BE tests

albinazs · Sep 25, 2024 · aad183c · aad183c
1 parent 87b4ef1
commit aad183c
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/wagtail/admin/tests/test_account_management.py b/wagtail/admin/tests/test_account_management.py
@@ -232,6 +232,7 @@ def post_form(self, extra_post_data):
             "locale-current_time_zone": "Europe/London",
             "theme-theme": "dark",
             "theme-density": "default",
+            "theme-contrast": "default",
         }
         post_data.update(extra_post_data)
         return self.client.post(reverse("wagtailadmin_account"), post_data)
@@ -480,7 +481,7 @@ def test_change_language_preferences(self):
         response = self.client.get(reverse("wagtailadmin_home"))
         self.assertContains(
             response,
-            '<html lang="es" dir="ltr" class="w-theme-dark w-density-default">',
+            '<html lang="es" dir="ltr" class="w-theme-dark w-density-default w-contrast-default">',
         )
 
     def test_unset_language_preferences(self):
@@ -610,6 +611,21 @@ def test_change_theme_post(self):
 
         self.assertEqual(profile.theme, "light")
 
+    def test_change_contrast_post(self):
+        response = self.post_form(
+            {
+                "theme-contrast": "high_contrast",
+            }
+        )
+
+        # Check that the user was redirected to the account page
+        self.assertRedirects(response, reverse("wagtailadmin_account"))
+
+        profile = UserProfile.get_for_user(self.user)
+        profile.refresh_from_db()
+
+        self.assertEqual(profile.contrast, "high_contrast")
+
     def test_change_density_post(self):
         response = self.post_form(
             {

diff --git a/wagtail/admin/tests/test_views.py b/wagtail/admin/tests/test_views.py
@@ -81,15 +81,15 @@ def test_language_code(self):
         response = self.client.get(reverse("wagtailadmin_login"))
         self.assertContains(
             response,
-            '<html lang="de" dir="ltr" class="w-theme-system w-density-default">',
+            '<html lang="de" dir="ltr" class="w-theme-system w-density-default w-contrast-default">',
         )
 
     @override_settings(LANGUAGE_CODE="he")
     def test_bidi_language_changes_dir_attribute(self):
         response = self.client.get(reverse("wagtailadmin_login"))
         self.assertContains(
             response,
-            '<html lang="he" dir="rtl" class="w-theme-system w-density-default">',
+            '<html lang="he" dir="rtl" class="w-theme-system w-density-default w-contrast-default">',
         )
 
     @override_settings(