From 9e98b57b58588e0bd7a4ee2f3f19c0cb454a135f Mon Sep 17 00:00:00 2001
From: EnvyDragon <138727357+EnvyDragon@users.noreply.github.com>
Date: Wed, 18 Sep 2024 14:59:39 -0400
Subject: [PATCH] de-flaking new tests

---
 worlds/sc2/test/test_generation.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/worlds/sc2/test/test_generation.py b/worlds/sc2/test/test_generation.py
index bfda3ee8e143..0a904a29287d 100644
--- 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)