c3848d0db4
* use $ syntax * missing test case change * try being a little smart * improve merge imports logic * add public as well * oops missed that one * last one there
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
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)
|