From 7bf15b4f44eb3f6c83f42b825393610d6d1c4184 Mon Sep 17 00:00:00 2001 From: Benedikt Bartscher Date: Thu, 25 Jul 2024 23:14:07 +0200 Subject: [PATCH] add simple test that minified state names are unique --- reflex/state.py | 2 +- tests/test_minify_state.py | 17 +++++++++++++++++ 2 files changed, 18 insertions(+), 1 deletion(-) create mode 100644 tests/test_minify_state.py 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