reflex/tests/components/core/test_banner.py

from reflex.components.core.banner import (
    ConnectionBanner,
    ConnectionModal,
    ConnectionPulser,
    WebsocketTargetURL,
)
from reflex.components.radix.themes.typography.text import Text


def test_websocket_target_url():
    url = WebsocketTargetURL.create()
    var_data = url._get_all_var_data()
    assert var_data is not None
    assert sorted(tuple((key for key, _ in var_data.imports))) == sorted(
        ("/utils/state", "/env.json")
    )


def test_connection_banner():
    banner = ConnectionBanner.create()
    _imports = banner._get_all_imports(collapse=True)
    assert sorted(tuple(_imports)) == sorted(
        (
            "react",
            "/utils/context",
            "/utils/state",
            "@radix-ui/themes@^3.0.0",
            "/env.json",
        )
    )

    msg = "Connection error"
    custom_banner = ConnectionBanner.create(Text.create(msg))
    assert msg in str(custom_banner.render())


def test_connection_modal():
    modal = ConnectionModal.create()
    _imports = modal._get_all_imports(collapse=True)
    assert sorted(tuple(_imports)) == sorted(
        (
            "react",
            "/utils/context",
            "/utils/state",
            "@radix-ui/themes@^3.0.0",
            "/env.json",
        )
    )

    msg = "Connection error"
    custom_modal = ConnectionModal.create(Text.create(msg))
    assert msg in str(custom_modal.render())


def test_connection_pulser():
    pulser = ConnectionPulser.create()
    _custom_code = pulser._get_all_custom_code()
    _imports = pulser._get_all_imports(collapse=True)