diff --git a/reflex/state.py b/reflex/state.py index 92ce351b5..d567ed40c 100644 --- a/reflex/state.py +++ b/reflex/state.py @@ -286,7 +286,7 @@ def get_var_for_field(cls: Type[BaseState], f: ModelField): # Keep track of all state instances to calculate minified state names -state_count = 0 +state_count: int = 0 all_state_names: Set[str] = set() diff --git a/tests/test_minify_state.py b/tests/test_minify_state.py new file mode 100644 index 000000000..e4dea43ef --- /dev/null +++ b/tests/test_minify_state.py @@ -0,0 +1,17 @@ +from typing import Set + +from reflex.state import all_state_names, next_minified_state_name + + +def test_next_minified_state_name(): + """Test that the next_minified_state_name function returns unique state names.""" + current_state_count = len(all_state_names) + state_names: Set[str] = set() + gen: int = 10000 + for _ in range(gen): + state_name = next_minified_state_name() + assert state_name not in state_names + state_names.add(state_name) + assert len(state_names) == gen + + assert len(all_state_names) == current_state_count + gen