Commit 2f08f0d55b173b099a90553e3bacff7134e047ce

Authored by Sergio Oliveira
1 parent 9f9d3466

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