b2d2719f90
* add type hinting to events * fix pyi * make it a list * add on change * dang it darglintz * add future annotations * add try except becuse i hate this * add check for class * aaaa * sometimes you need to make hard decisions * ono * i hate unions * add rx event * move stuff around * maybe * special case osmething * i don't need no test * remove stray print Co-authored-by: Masen Furer <m_github@0x26.net> * remove another stray print Co-authored-by: Masen Furer <m_github@0x26.net> * add rx event --------- Co-authored-by: Masen Furer <m_github@0x26.net>
131 lines
3.1 KiB
Python
131 lines
3.1 KiB
Python
from typing import Any
|
|
|
|
from reflex import event
|
|
from reflex.components.core.upload import (
|
|
StyledUpload,
|
|
Upload,
|
|
UploadNamespace,
|
|
_on_drop_spec, # type: ignore
|
|
cancel_upload,
|
|
get_upload_url,
|
|
)
|
|
from reflex.event import EventSpec
|
|
from reflex.state import State
|
|
from reflex.vars.base import LiteralVar, Var
|
|
|
|
|
|
class UploadStateTest(State):
|
|
"""Test upload state."""
|
|
|
|
@event
|
|
def drop_handler(self, files: Any):
|
|
"""Handle the drop event.
|
|
|
|
Args:
|
|
files: The files dropped.
|
|
"""
|
|
pass
|
|
|
|
@event
|
|
def not_drop_handler(self, not_files: Any):
|
|
"""Handle the drop event without defining the files argument.
|
|
|
|
Args:
|
|
not_files: The files dropped.
|
|
"""
|
|
pass
|
|
|
|
|
|
def test_cancel_upload():
|
|
spec = cancel_upload("foo_id")
|
|
assert isinstance(spec, EventSpec)
|
|
|
|
|
|
def test_get_upload_url():
|
|
url = get_upload_url("foo_file")
|
|
assert isinstance(url, Var)
|
|
|
|
|
|
def test__on_drop_spec():
|
|
assert isinstance(_on_drop_spec(LiteralVar.create([])), tuple)
|
|
|
|
|
|
def test_upload_create():
|
|
up_comp_1 = Upload.create()
|
|
assert isinstance(up_comp_1, Upload)
|
|
assert up_comp_1.is_used
|
|
|
|
# reset is_used
|
|
Upload.is_used = False
|
|
|
|
up_comp_2 = Upload.create(
|
|
id="foo_id",
|
|
on_drop=UploadStateTest.drop_handler([]), # type: ignore
|
|
)
|
|
assert isinstance(up_comp_2, Upload)
|
|
assert up_comp_2.is_used
|
|
|
|
# reset is_used
|
|
Upload.is_used = False
|
|
|
|
up_comp_3 = Upload.create(
|
|
id="foo_id",
|
|
on_drop=UploadStateTest.drop_handler,
|
|
)
|
|
assert isinstance(up_comp_3, Upload)
|
|
assert up_comp_3.is_used
|
|
|
|
# reset is_used
|
|
Upload.is_used = False
|
|
|
|
up_comp_4 = Upload.create(
|
|
id="foo_id",
|
|
on_drop=UploadStateTest.not_drop_handler([]), # type: ignore
|
|
)
|
|
assert isinstance(up_comp_4, Upload)
|
|
assert up_comp_4.is_used
|
|
|
|
|
|
def test_styled_upload_create():
|
|
styled_up_comp_1 = StyledUpload.create()
|
|
assert isinstance(styled_up_comp_1, StyledUpload)
|
|
assert styled_up_comp_1.is_used
|
|
|
|
# reset is_used
|
|
StyledUpload.is_used = False
|
|
|
|
styled_up_comp_2 = StyledUpload.create(
|
|
id="foo_id",
|
|
on_drop=UploadStateTest.drop_handler([]), # type: ignore
|
|
)
|
|
assert isinstance(styled_up_comp_2, StyledUpload)
|
|
assert styled_up_comp_2.is_used
|
|
|
|
# reset is_used
|
|
StyledUpload.is_used = False
|
|
|
|
styled_up_comp_3 = StyledUpload.create(
|
|
id="foo_id",
|
|
on_drop=UploadStateTest.drop_handler,
|
|
)
|
|
assert isinstance(styled_up_comp_3, StyledUpload)
|
|
assert styled_up_comp_3.is_used
|
|
|
|
# reset is_used
|
|
StyledUpload.is_used = False
|
|
|
|
styled_up_comp_4 = StyledUpload.create(
|
|
id="foo_id",
|
|
on_drop=UploadStateTest.not_drop_handler([]), # type: ignore
|
|
)
|
|
assert isinstance(styled_up_comp_4, StyledUpload)
|
|
assert styled_up_comp_4.is_used
|
|
|
|
|
|
def test_upload_namespace():
|
|
up_ns = UploadNamespace()
|
|
assert isinstance(up_ns, UploadNamespace)
|
|
|
|
assert isinstance(up_ns(id="foo_id"), StyledUpload)
|
|
assert isinstance(up_ns.root(id="foo_id"), Upload)
|