From 3121e8a7a9e74845c7c99765c0ef86c4f9a2e3f6 Mon Sep 17 00:00:00 2001 From: Masen Furer Date: Thu, 14 Mar 2024 16:16:29 -0700 Subject: [PATCH] test_custom_component_get_imports: test imports from wrapped custom components --- tests/components/test_component.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/tests/components/test_component.py b/tests/components/test_component.py index 04468b2ba..d5413229f 100644 --- a/tests/components/test_component.py +++ b/tests/components/test_component.py @@ -1269,3 +1269,28 @@ def test_deprecated_props(capsys): assert "type={`type1`}" in c2_1_render["props"] assert "min={`min1`}" in c2_1_render["props"] assert "max={`max1`}" in c2_1_render["props"] + + +def test_custom_component_get_imports(): + class Inner(Component): + tag = "Inner" + library = "inner" + + class Other(Component): + tag = "Other" + library = "other" + + @rx.memo + def wrapper(): + return Inner.create() + + @rx.memo + def outer(c: Component): + return Other.create(c) + + custom_comp = wrapper() + assert "inner" in custom_comp.get_imports() + + outer_comp = outer(c=wrapper()) + assert "inner" in outer_comp.get_imports() + assert "other" in outer_comp.get_imports()