test: Improve navigation in US04 tests by ensuring back button is clickable before clicking

Author: 02loveslollipop

tests/us4.py

@@ -302,7 +302,10 @@ def test_us04_verify_calculation_of_current_weighted_average(self):
 
             # --- Navigate back for Test Step 2 ---
             logger.info("Navigating back to home page for 2nd grade.")
-            self.driver.find_element(By.XPATH, self.NAV_BACK_BUTTON_XPATH).click()
+            nav_back_button_step1 = self.wait_long.until(
+                EC.element_to_be_clickable((By.XPATH, self.NAV_BACK_BUTTON_XPATH))
+            )
+            nav_back_button_step1.click()
             self.wait_long.until(EC.presence_of_element_located((By.CSS_SELECTOR, self.HOME_CONTAINER_SELECTOR)))
             logger.info(f"Navigated back to home page for 2nd grade.")
 
@@ -325,7 +328,10 @@ def test_us04_verify_calculation_of_current_weighted_average(self):
 
             # --- Navigate back for Test Step 3 ---
             logger.info("Navigating back to home page for 3rd grade.")
-            self.driver.find_element(By.XPATH, self.NAV_BACK_BUTTON_XPATH).click()
+            nav_back_button_step2 = self.wait_long.until(
+                EC.element_to_be_clickable((By.XPATH, self.NAV_BACK_BUTTON_XPATH))
+            )
+            nav_back_button_step2.click()
             self.wait_long.until(EC.presence_of_element_located((By.CSS_SELECTOR, self.HOME_CONTAINER_SELECTOR)))
             logger.info(f"Navigated back to home page for 3rd grade.")