def test_get_event_names(ees: EventEmitterS,
arg_swallower: Callable[..., None]) -> None:
events = {"event", "event1", "event2", "event3"}
for event in events:
ees.on(event, arg_swallower)
assert all(registered_event_name in events
for registered_event_name in ees.event_names())
def test_on_does_not_emits_error_when_not_listening_for_errors(
ees: EventEmitterS, error_helper: "EEExceptionHelper") -> None:
ees.on("event", error_helper.error_raiser)
assert ees.listener_count("event") == 1
assert ees.listener_count("error") == 0
assert ees.emit("event", 1, data=2)
error_helper.assert_error_was_not_emitted()
def test_remove_all_listeners(ees: EventEmitterS,
arg_swallower: Callable[..., None]) -> None:
ees.on("event", arg_swallower)
assert ees.listener_count("event") == 1
assert arg_swallower is ees.listeners("event")[0]
ees.remove_all_listeners()
assert ees.listener_count("event") == 0
def test_on(ees: EventEmitterS, mock: Mock) -> None:
ees.on("event", mock.method)
assert ees.listener_count("event") == 1
assert mock.method is ees.listeners("event")[0]
assert ees.emit("event", 1, data=2)
mock.method.assert_called_with(1, data=2)
assert ees.listener_count("event") == 1
assert mock.method is ees.listeners("event")[0]
async def test_coroutine_listener_does_not_emits_error_when_not_listening_for_errors(
ees_with_event_loop: EventEmitterS,
error_helper: "EEExceptionHelper",
event_loop: AbstractEventLoop,
) -> None:
error_helper.with_deferred(event_loop)
ees_with_event_loop.on("event", error_helper.error_raiser_async)
assert ees_with_event_loop.listener_count("event") == 1
assert ees_with_event_loop.listener_count("error") == 0
assert ees_with_event_loop.emit("event", 1, data=2)
await error_helper.assert_error_was_not_emitted_async()
async def test_coroutine_listener(ees_with_event_loop: EventEmitterS,
mock: Mock, deferred: Future) -> None:
async def handler(*args, **kwargs) -> None:
mock.method(*args, **kwargs)
deferred.set_result(True)
ees_with_event_loop.on("event", handler)
assert ees_with_event_loop.listener_count("event") == 1
assert handler is ees_with_event_loop.listeners("event")[0]
assert ees_with_event_loop.emit("event", 1, data=2)
assert await deferred
mock.method.assert_called_with(1, data=2)
assert ees_with_event_loop.listener_count("event") == 1
assert handler is ees_with_event_loop.listeners("event")[0]
def test_get_listeners_registered_for_event(
ees: EventEmitterS, arg_swallower: Callable[..., None]) -> None:
ees.on("event", arg_swallower)
ees_listeners = ees.listeners("event")
assert len(ees_listeners) == 1
assert ees_listeners[0] is arg_swallower