diff --git a/game/end_to_end_tests/base_game_test.py b/game/end_to_end_tests/base_game_test.py index 34e283f2b..aa5c074bb 100644 --- a/game/end_to_end_tests/base_game_test.py +++ b/game/end_to_end_tests/base_game_test.py @@ -101,6 +101,18 @@ def go_to_level(self, level_name, from_python_den=False): return GamePage(self.selenium) + def go_to_level_without_dismissing_dialog(self, level_name, from_python_den=False): + viewname = ( + "play_python_default_level" + if from_python_den + else "play_default_level" + ) + + path = reverse(viewname, kwargs={"level_name": str(level_name)}) + self._go_to_path(path) + + return BasePage(self.selenium) + def go_to_custom_level(self, level): path = reverse("play_custom_level", kwargs={"levelId": str(level.id)}) self._go_to_path(path) diff --git a/game/end_to_end_tests/test_level_selection.py b/game/end_to_end_tests/test_level_selection.py index 6441e0ed5..e949e7370 100644 --- a/game/end_to_end_tests/test_level_selection.py +++ b/game/end_to_end_tests/test_level_selection.py @@ -2,7 +2,6 @@ from common.tests.utils.organisation import create_organisation_directly from common.tests.utils.student import create_school_student_directly from common.tests.utils.teacher import signup_teacher_directly -from django.urls import reverse from hamcrest import assert_that, ends_with, equal_to from selenium.common.exceptions import NoSuchElementException from selenium.webdriver.common.by import By @@ -69,8 +68,7 @@ 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_path(reverse("play_python_default_level", kwargs={"level_name": "41"})) - + page = self.go_to_level_without_dismissing_dialog(41, True) next_button = self.selenium.find_element(By.ID, "next_button") assert WebDriverWait(self.selenium, 10).until( EC.element_to_be_clickable(next_button)