From 400d15f72eb5bec7d8fa4b3dfe4709ea7b6a2a13 Mon Sep 17 00:00:00 2001 From: Lendemor Date: Wed, 11 Dec 2024 19:43:31 +0100 Subject: [PATCH] revert test changes --- tests/units/components/test_component.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/units/components/test_component.py b/tests/units/components/test_component.py index e375be414..8d26b02c6 100644 --- a/tests/units/components/test_component.py +++ b/tests/units/components/test_component.py @@ -31,7 +31,6 @@ from reflex.utils.exceptions import EventFnArgMismatch from reflex.utils.imports import ImportDict, ImportVar, ParsedImportDict, parse_imports from reflex.vars import VarData from reflex.vars.base import LiteralVar, Var -from reflex.vars.hooks import HookVar @pytest.fixture @@ -2078,11 +2077,11 @@ def test_component_add_hooks_var(): ), ] - assert list(HookComponent()._get_all_hooks()) == [ - HookVar.create("const hook3 = useRef(null)"), + assert list(hook for hook in HookComponent()._get_all_hooks()) == [ + "const hook3 = useRef(null)", "const hook1 = 42", "const hook2 = 43", - HookVar.create("useEffect(() => () => {}, [])"), + "useEffect(() => () => {}, [])", ] imports = HookComponent()._get_all_imports() assert len(imports) == 1