Commit f1ff2fc7373ad36fdd4c4df1e644cf5cf7bc24e2

Authored by Lucas Kanashiro
Committed by Sergio Oliveira
1 parent 5975ae7b

colab.signals: raise exception in send method

If signal is not registered before call send.

Signed-off-by: Lucas Moura <lucas.moura128@gmail.com>
colab/signals/signals.py
... ... @@ -29,3 +29,5 @@ class Signals():
29 29 def send(self, signal_name, sender, **kwargs):
30 30 if signal_name in self.signal_instances:
31 31 self.signal_instances[signal_name].send(sender=sender)
  32 + else:
  33 + raise Exception("Signal does not exist!")
... ...
colab/signals/tests/test_signals.py
... ... @@ -79,4 +79,5 @@ class SignalsTest(TestCase):
79 79 self.assertTrue(mock.is_called)
80 80  
81 81  
82   -
  82 + def test_send_signal_not_registered(self):
  83 + self.assertRaises(Exception, self.signals.send, 'test_signal', 'test')
... ...