try to fix test

ocadotechnology · Aug 28, 2024 · bd59090 · bd59090
1 parent e1ca2a7
commit bd59090
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/game/end_to_end_tests/test_level_selection.py b/game/end_to_end_tests/test_level_selection.py
@@ -68,15 +68,15 @@ def test_redirect_to_levelless_episode(self):
         levels_page = self.go_to_reverse("python_levels")
         expected_url = levels_page.browser.current_url
 
-        page = self.go_to_level("41", True)
+        page = self.go_to_level(41, True)
         assert WebDriverWait(self.selenium, 10).until(
             EC.visibility_of_element_located((By.ID, "next_button"))
         )
-        page.browser.find_element(By.ID, "next_button").click()
+        self.selenium.find_element(By.ID, "next_button").click()
 
         current_url = page.browser.current_url
         assert current_url == expected_url
 
-        episode_20_header = page.find_element(By.ID, "episode-20")
+        episode_20_header = self.selenium.find_element(By.ID, "episode-20")
         episode_20_expanded = episode_20_header.get_attribute("aria-expanded")
         assert episode_20_expanded == True