define and use new function

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)

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)