de-flaking new tests

EnvyDragon · Sep 18, 2024 · 9e98b57 · 9e98b57
1 parent c5d9f98
commit 9e98b57
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/worlds/sc2/test/test_generation.py b/worlds/sc2/test/test_generation.py
@@ -537,6 +537,12 @@ def test_excluding_mission_short_name_excludes_all_variants_of_mission(self):
             'mission_order': options.MissionOrder.option_grid,
             'selected_races': options.SelectRaces.option_all,
             'enable_race_swap': options.EnableRaceSwapVariants.option_shuffle_all,
+            'enable_prophecy_missions': False,
+            'enable_hots_missions': False,
+            'enable_lotv_prologue_missions': False,
+            'enable_lotv_missions': False,
+            'enable_epilogue_missions': False,
+            'enable_nco_missions': False,
         }
         self.generate_world(world_options)
         missions = get_all_missions(self.world.custom_mission_order)
@@ -553,6 +559,12 @@ def test_excluding_mission_variant_excludes_just_that_variant(self):
             'mission_order': options.MissionOrder.option_grid,
             'selected_races': options.SelectRaces.option_all,
             'enable_race_swap': options.EnableRaceSwapVariants.option_shuffle_all,
+            'enable_prophecy_missions': False,
+            'enable_hots_missions': False,
+            'enable_lotv_prologue_missions': False,
+            'enable_lotv_missions': False,
+            'enable_epilogue_missions': False,
+            'enable_nco_missions': False,
         }
         self.generate_world(world_options)
         missions = get_all_missions(self.world.custom_mission_order)