62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
from reflex.components.core.banner import (
|
|
ConnectionBanner,
|
|
ConnectionModal,
|
|
ConnectionPulser,
|
|
WebsocketTargetURL,
|
|
)
|
|
from reflex.components.radix.themes.base import RadixThemesComponent
|
|
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(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(_imports) == sorted(
|
|
(
|
|
"react",
|
|
"$/utils/context",
|
|
"$/utils/state",
|
|
"@emotion/react",
|
|
RadixThemesComponent().library or "",
|
|
"$/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(_imports) == sorted(
|
|
(
|
|
"react",
|
|
"$/utils/context",
|
|
"$/utils/state",
|
|
"@emotion/react",
|
|
RadixThemesComponent().library or "",
|
|
"$/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)
|