Fix event handler override (#828)

This commit is contained in:
Nikhil Rao 2023-04-15 11:33:52 -07:00 committed by GitHub
parent bd03e9290c
commit e4bf94464b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 1 additions and 8 deletions

View File

@ -182,9 +182,7 @@ class State(Base, ABC):
for name, fn in cls.__dict__.items()
if not name.startswith("_") and isinstance(fn, Callable)
}
for name, fn in events.items():
event_handler = EventHandler(fn=fn)
cls.event_handlers[name] = event_handler
cls.event_handlers = {name: EventHandler(fn=fn) for name, fn in events.items()}
cls.set_handlers()

View File

@ -211,20 +211,15 @@ def test_event_handlers(test_state):
test_state: A state.
"""
expected = {
"change_both",
"do_nothing",
"do_something",
"set_array",
"set_complex",
"set_count",
"set_fig",
"set_key",
"set_mapping",
"set_num1",
"set_num2",
"set_obj",
"set_value",
"set_value2",
}
cls = type(test_state)