From de2feca166b94b9fde459e4e4931fb97e4485594 Mon Sep 17 00:00:00 2001
From: Tim Schilling <tim@aspiredu.com>
Date: Sun, 26 May 2024 08:42:46 -0500
Subject: [PATCH] Fix theme selenium integration test.

The click actions were being triggered on the wrong element.
---
 tests/test_integration.py | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/tests/test_integration.py b/tests/test_integration.py
index 70522ff8a..465804651 100644
--- a/tests/test_integration.py
+++ b/tests/test_integration.py
@@ -802,19 +802,21 @@ def test_theme_toggle(self):
         self.assertEqual(toolbar.get_attribute("data-theme"), "auto")
 
         # The theme toggle button is shown on the toolbar
-        self.assertIn('<a id="djToggleThemeButton', toolbar.text)
+        toggle_button = self.selenium.find_element(By.ID, "djToggleThemeButton")
+        self.assertTrue(toggle_button.is_displayed())
 
         # The theme changes when user clicks the button
-        self.selenium.find_element(By.ID, "djDebugToolbar").click()
+        toggle_button.click()
         self.assertEqual(toolbar.get_attribute("data-theme"), "light")
-        self.selenium.find_element(By.ID, "djDebugToolbar").click()
+        toggle_button.click()
         self.assertEqual(toolbar.get_attribute("data-theme"), "dark")
+        toggle_button.click()
+        self.assertEqual(toolbar.get_attribute("data-theme"), "auto")
+        # Switch back to light.
+        toggle_button.click()
+        self.assertEqual(toolbar.get_attribute("data-theme"), "light")
 
         # Enter the page again to check that user settings is saved
         self.get("/regular/basic/")
         toolbar = self.selenium.find_element(By.ID, "djDebug")
-        self.assertEqual(toolbar.get_attribute("data-theme"), "dark")
-
-        # Set the default again and check that dark changes to auto
-        self.selenium.find_element(By.ID, "djDebugToolbar").click()
-        self.assertEqual(toolbar.get_attribute("data-theme"), "auto")
+        self.assertEqual(toolbar.get_attribute("data-theme"), "light")