Commit 2f08f0d55b173b099a90553e3bacff7134e047ce
1 parent
9f9d3466
Exists in
master
and in
30 other branches
Fixed PEP8
Showing
1 changed file
with
26 additions
and
31 deletions
Show diff stats
colab/signals/tests/test_signals.py
| ... | ... | @@ -3,50 +3,48 @@ Test Signals class. |
| 3 | 3 | Objective: Test parameters, and behavior. |
| 4 | 4 | """ |
| 5 | 5 | |
| 6 | -from django.test import TestCase | |
| 7 | -from django.apps import apps | |
| 8 | 6 | import django |
| 9 | -from colab.signals.signals import * | |
| 7 | + | |
| 8 | +from django.test import TestCase | |
| 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) | |
| 14 | + | |
| 12 | 15 | |
| 13 | 16 | class SignalsTest(TestCase): |
| 14 | 17 | |
| 15 | 18 | def setUp(self): |
| 16 | - django.setup() | |
| 17 | - self.list_signal = ['a', 'b', 'c'] | |
| 18 | - self.plugin_name = 'test_signal' | |
| 19 | - | |
| 19 | + django.setup() | |
| 20 | + self.list_signal = ['a', 'b', 'c'] | |
| 21 | + self.plugin_name = 'test_signal' | |
| 20 | 22 | |
| 21 | 23 | def test_register_signal_(self): |
| 22 | - register_signal(self.plugin_name, self.list_signal) | |
| 23 | - | |
| 24 | - signal_name ='a' | |
| 25 | - signal_list = ['test_signal'] | |
| 26 | - self.assertEqual(len(registered_signals[signal_name]), 1) | |
| 27 | - self.assertEqual(registered_signals[signal_name], signal_list) | |
| 24 | + register_signal(self.plugin_name, self.list_signal) | |
| 25 | + signal_name = 'a' | |
| 26 | + signal_list = ['test_signal'] | |
| 27 | + self.assertEqual(len(registered_signals[signal_name]), 1) | |
| 28 | + self.assertEqual(registered_signals[signal_name], signal_list) | |
| 28 | 29 | |
| 29 | - | |
| 30 | 30 | def test_register_signal_already_registered(self): |
| 31 | - signal_name ='a' | |
| 32 | - signal_list = ['test_signal'] | |
| 31 | + signal_name = 'a' | |
| 32 | + signal_list = ['test_signal'] | |
| 33 | 33 | |
| 34 | - register_signal(self.plugin_name, self.list_signal) | |
| 35 | - self.assertEqual(len(registered_signals[signal_name]), 1) | |
| 34 | + register_signal(self.plugin_name, self.list_signal) | |
| 35 | + self.assertEqual(len(registered_signals[signal_name]), 1) | |
| 36 | 36 | |
| 37 | - register_signal(self.plugin_name, self.list_signal) | |
| 38 | - self.assertEqual(len(registered_signals[signal_name]), 1) | |
| 39 | - self.assertEqual(registered_signals[signal_name], signal_list) | |
| 37 | + register_signal(self.plugin_name, self.list_signal) | |
| 38 | + self.assertEqual(len(registered_signals[signal_name]), 1) | |
| 39 | + self.assertEqual(registered_signals[signal_name], signal_list) | |
| 40 | 40 | |
| 41 | - | |
| 42 | 41 | def test_connect_non_registered_signal(self): |
| 43 | 42 | sender = 'Test' |
| 44 | 43 | handling_method = 'Test' |
| 45 | 44 | signal_name = 'Test' |
| 46 | - | |
| 47 | - self.assertRaises(Exception, connect_signal, signal_name, | |
| 48 | - sender, handling_method) | |
| 49 | 45 | |
| 46 | + self.assertRaises(Exception, connect_signal, signal_name, | |
| 47 | + sender, handling_method) | |
| 50 | 48 | |
| 51 | 49 | @patch('colab.signals.signals.Signal.connect') |
| 52 | 50 | def test_connect_already_registered_signal(self, mock): |
| ... | ... | @@ -54,23 +52,21 @@ class SignalsTest(TestCase): |
| 54 | 52 | handling_method = MagicMock |
| 55 | 53 | type(handling_method).delay = PropertyMock(return_value='Test') |
| 56 | 54 | signal_name = 'a' |
| 57 | - | |
| 55 | + | |
| 58 | 56 | register_signal(self.plugin_name, self.list_signal) |
| 59 | 57 | |
| 60 | 58 | connect_signal(signal_name, sender, handling_method.delay) |
| 61 | 59 | args, kwargs = mock.call_args |
| 62 | - | |
| 60 | + | |
| 63 | 61 | self.assertEqual(args[0], handling_method) |
| 64 | 62 | self.assertEqual(kwargs['sender'], sender) |
| 65 | 63 | self.assertTrue(mock.is_called) |
| 66 | - | |
| 67 | 64 | |
| 68 | 65 | @patch('colab.signals.signals.Signal.send') |
| 69 | 66 | def test_send_signal(self, mock): |
| 70 | 67 | sender = 'Test' |
| 71 | - handling_method = 'Test' | |
| 72 | 68 | signal_name = 'a' |
| 73 | - | |
| 69 | + | |
| 74 | 70 | register_signal(self.plugin_name, self.list_signal) |
| 75 | 71 | send(signal_name, sender) |
| 76 | 72 | |
| ... | ... | @@ -78,7 +74,6 @@ class SignalsTest(TestCase): |
| 78 | 74 | |
| 79 | 75 | self.assertEqual(kwargs['sender'], sender) |
| 80 | 76 | self.assertTrue(mock.is_called) |
| 81 | - | |
| 82 | 77 | |
| 83 | 78 | def test_send_signal_not_registered(self): |
| 84 | 79 | self.assertRaises(Exception, send, 'test_signal', 'test') | ... | ... |