def __register_receiver(self, sender, receiver):
""" Register a receiver
"""
sender_key = get_sender_key(sender)
receivers = self.receivers.get(sender_key, [])
receivers.append(receiver)
self.receivers[sender_key] = receivers
def __get_receivers(self, sender):
""" Returns receivers of the sender
"""
sender_key = get_sender_key(sender)
return self.receivers.get(sender_key, [])
def test_when_signal_connect_sender_is_a_function():
test_signal = Signal()
test_signal.connect(sender=sender_function, receiver=coroutine_receiver)
key = get_sender_key(sender_function)
assert coroutine_receiver in test_signal.receivers[key]
def test_signal_connect_when_is_coroutine():
test_signal = Signal()
test_signal.connect(sender=Sender, receiver=coroutine_receiver)
key = get_sender_key(Sender)
assert coroutine_receiver in test_signal.receivers[key]