Commit f1ff2fc7373ad36fdd4c4df1e644cf5cf7bc24e2
Committed by
Sergio Oliveira
1 parent
5975ae7b
Exists in
master
and in
30 other branches
colab.signals: raise exception in send method
If signal is not registered before call send. Signed-off-by: Lucas Moura <lucas.moura128@gmail.com>
Showing
2 changed files
with
4 additions
and
1 deletions
Show diff stats
colab/signals/signals.py
@@ -29,3 +29,5 @@ class Signals(): | @@ -29,3 +29,5 @@ class Signals(): | ||
29 | def send(self, signal_name, sender, **kwargs): | 29 | def send(self, signal_name, sender, **kwargs): |
30 | if signal_name in self.signal_instances: | 30 | if signal_name in self.signal_instances: |
31 | self.signal_instances[signal_name].send(sender=sender) | 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,4 +79,5 @@ class SignalsTest(TestCase): | ||
79 | self.assertTrue(mock.is_called) | 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') |