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') | ... | ... |