Commit fe71a2b1a96024b11b2c46146f226094daedca95
1 parent
2f08f0d5
Exists in
master
and in
30 other branches
Using custom exception
Showing
4 changed files
with
17 additions
and
6 deletions
Show diff stats
colab/signals/signals.py
| 1 | 1 | from django.dispatch import Signal |
| 2 | 2 | from colab.signals.celery import app |
| 3 | 3 | |
| 4 | +from .exceptions import SignalDoesNotExist | |
| 5 | + | |
| 4 | 6 | |
| 5 | 7 | registered_signals = {} |
| 6 | 8 | signal_instances = {} |
| ... | ... | @@ -28,11 +30,11 @@ def connect_signal(signal_name, sender, handling_method): |
| 28 | 30 | signal_instances[signal_name].connect(handling_method.delay, |
| 29 | 31 | sender=sender) |
| 30 | 32 | else: |
| 31 | - raise Exception("Signal does not exist!") | |
| 33 | + raise SignalDoesNotExist | |
| 32 | 34 | |
| 33 | 35 | |
| 34 | 36 | def send(signal_name, sender, **kwargs): |
| 35 | 37 | if signal_name in signal_instances: |
| 36 | 38 | signal_instances[signal_name].send(sender=sender, **kwargs) |
| 37 | 39 | else: |
| 38 | - raise Exception("Signal does not exist!") | |
| 40 | + raise SignalDoesNotExist | ... | ... |
colab/signals/tests/test_signals.py
| ... | ... | @@ -9,8 +9,8 @@ from django.test import TestCase |
| 9 | 9 | |
| 10 | 10 | from mock import patch, MagicMock, PropertyMock |
| 11 | 11 | |
| 12 | -from colab.signals.signals import (registered_signals, register_signal, | |
| 13 | - connect_signal, send) | |
| 12 | +from ..signals import registered_signals, register_signal, connect_signal, send | |
| 13 | +from ..exceptions import SignalDoesNotExist | |
| 14 | 14 | |
| 15 | 15 | |
| 16 | 16 | class SignalsTest(TestCase): |
| ... | ... | @@ -43,7 +43,7 @@ class SignalsTest(TestCase): |
| 43 | 43 | handling_method = 'Test' |
| 44 | 44 | signal_name = 'Test' |
| 45 | 45 | |
| 46 | - self.assertRaises(Exception, connect_signal, signal_name, | |
| 46 | + self.assertRaises(SignalDoesNotExist, connect_signal, signal_name, | |
| 47 | 47 | sender, handling_method) |
| 48 | 48 | |
| 49 | 49 | @patch('colab.signals.signals.Signal.connect') |
| ... | ... | @@ -76,4 +76,4 @@ class SignalsTest(TestCase): |
| 76 | 76 | self.assertTrue(mock.is_called) |
| 77 | 77 | |
| 78 | 78 | def test_send_signal_not_registered(self): |
| 79 | - self.assertRaises(Exception, send, 'test_signal', 'test') | |
| 79 | + self.assertRaises(SignalDoesNotExist, send, 'test_signal', 'test') | ... | ... |