From 18bcaa85a27890de47e623c630a65dddb3d644d4 Mon Sep 17 00:00:00 2001 From: Fabian Dill Date: Mon, 10 Feb 2025 19:18:14 +0100 Subject: [PATCH] Test: ensure get_all_state() does not error in between steps (#4612) --- test/general/test_state.py | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 test/general/test_state.py diff --git a/test/general/test_state.py b/test/general/test_state.py new file mode 100644 index 00000000..460fc3d6 --- /dev/null +++ b/test/general/test_state.py @@ -0,0 +1,29 @@ +import unittest + +from worlds.AutoWorld import AutoWorldRegister, call_all +from . import setup_solo_multiworld + + +class TestBase(unittest.TestCase): + gen_steps = ( + "generate_early", + "create_regions", + ) + + test_steps = ( + "create_items", + "set_rules", + "connect_entrances", + "generate_basic", + "pre_fill", + ) + + def test_all_state_is_available(self): + """Ensure all_state can be created at certain steps.""" + for game_name, world_type in AutoWorldRegister.world_types.items(): + with self.subTest("Game", game=game_name): + multiworld = setup_solo_multiworld(world_type, self.gen_steps) + for step in self.test_steps: + with self.subTest("Step", step=step): + call_all(multiworld, step) + self.assertTrue(multiworld.get_all_state(False, True))