add specific redirecting check

ocadotechnology · Aug 28, 2024 · d781dc0 · d781dc0
1 parent c25b25c
commit d781dc0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/game/end_to_end_tests/base_game_test.py b/game/end_to_end_tests/base_game_test.py
@@ -57,6 +57,7 @@ def complete_and_check_level(
         page,
         from_python_den=False,
         next_episode=None,
+        redirects=False,
         check_algorithm_score=True,
         check_route_score=True,
         final_level=False,
@@ -72,7 +73,7 @@ def complete_and_check_level(
         if final_level:
             return page
         if next_episode is None:
-            if next_level is None:
+            if redirects:
                 page.next_level_redirected(from_python_den)
             else:
                 page.next_level(from_python_den)

diff --git a/game/end_to_end_tests/test_play_through.py b/game/end_to_end_tests/test_play_through.py
@@ -460,7 +460,7 @@ def test_python_level_040(self):
         )
 
     def test_python_level_041(self):
-        self._complete_level(41, from_python_den=True)
+        self._complete_level(41, from_python_den=True, redirects=True)
 
     def test_episode_15(self):
         self._complete_episode(