91 lines
2.4 KiB
Python
91 lines
2.4 KiB
Python
import pytest
|
|
|
|
from pynecone import event, utils
|
|
from pynecone.event import Event, EventHandler, EventSpec
|
|
from pynecone.var import Var
|
|
|
|
|
|
def make_var(value) -> Var:
|
|
"""Make a variable.
|
|
|
|
Args:
|
|
value: The value of the var.
|
|
|
|
Returns:
|
|
The var.
|
|
"""
|
|
var = Var.create(value)
|
|
assert var is not None
|
|
return var
|
|
|
|
|
|
def test_create_event():
|
|
"""Test creating an event."""
|
|
event = Event(token="token", name="state.do_thing", payload={"arg": "value"})
|
|
assert event.token == "token"
|
|
assert event.name == "state.do_thing"
|
|
assert event.payload == {"arg": "value"}
|
|
|
|
|
|
def test_call_event_handler():
|
|
"""Test that calling an event handler creates an event spec."""
|
|
|
|
def test_fn():
|
|
pass
|
|
|
|
def test_fn_with_args(_, arg1, arg2):
|
|
pass
|
|
|
|
handler = EventHandler(fn=test_fn)
|
|
event_spec = handler()
|
|
|
|
assert event_spec.handler == handler
|
|
assert event_spec.local_args == ()
|
|
assert event_spec.args == ()
|
|
|
|
handler = EventHandler(fn=test_fn_with_args)
|
|
event_spec = handler(make_var("first"), make_var("second"))
|
|
|
|
assert event_spec.handler == handler
|
|
assert event_spec.local_args == ()
|
|
assert event_spec.args == (("arg1", "first"), ("arg2", "second"))
|
|
|
|
first, second = 123, "456"
|
|
handler = EventHandler(fn=test_fn_with_args)
|
|
event_spec = handler(first, second) # type: ignore
|
|
|
|
assert event_spec.handler == handler
|
|
assert event_spec.local_args == ()
|
|
assert event_spec.args == (
|
|
("arg1", utils.json_dumps(first)),
|
|
("arg2", utils.json_dumps(second)),
|
|
)
|
|
|
|
handler = EventHandler(fn=test_fn_with_args)
|
|
with pytest.raises(TypeError):
|
|
handler(test_fn) # type: ignore
|
|
|
|
|
|
def test_event_redirect():
|
|
"""Test the event redirect function."""
|
|
spec = event.redirect("/path")
|
|
assert isinstance(spec, EventSpec)
|
|
assert spec.handler.fn.__qualname__ == "_redirect"
|
|
assert spec.args == (("path", "/path"),)
|
|
|
|
|
|
def test_event_console_log():
|
|
"""Test the event console log function."""
|
|
spec = event.console_log("message")
|
|
assert isinstance(spec, EventSpec)
|
|
assert spec.handler.fn.__qualname__ == "_console"
|
|
assert spec.args == (("message", "message"),)
|
|
|
|
|
|
def test_event_window_alert():
|
|
"""Test the event window alert function."""
|
|
spec = event.window_alert("message")
|
|
assert isinstance(spec, EventSpec)
|
|
assert spec.handler.fn.__qualname__ == "_alert"
|
|
assert spec.args == (("message", "message"),)
|