0af4770180
* build pyi files when building/publishing 3rd party * fix typo in workflow * add future annotation * add tests to pass coverage check * add more unit tests * omit pyi_generator from test coverage * change black from dev deps to direct deps * remake all pyi * format pyi if black is present, return as if otherwise * fix requested changes --------- Co-authored-by: Masen Furer <m_github@0x26.net>
52 lines
1.4 KiB
Python
52 lines
1.4 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()
|
|
_imports = url.get_imports(collapse=True)
|
|
assert list(_imports.keys()) == ["/utils/state", "/env.json"]
|
|
|
|
|
|
def test_connection_banner():
|
|
banner = ConnectionBanner.create()
|
|
_imports = banner.get_imports(collapse=True)
|
|
assert list(_imports.keys()) == [
|
|
"react",
|
|
"/utils/context",
|
|
"/utils/state",
|
|
"@radix-ui/themes@^2.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_imports(collapse=True)
|
|
assert list(_imports.keys()) == [
|
|
"react",
|
|
"/utils/context",
|
|
"/utils/state",
|
|
"@radix-ui/themes@^2.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_custom_code()
|
|
_imports = pulser.get_imports(collapse=True)
|