diff --git a/small_small_hr/emails.py b/small_small_hr/emails.py index 576d6d9..9e3637c 100644 --- a/small_small_hr/emails.py +++ b/small_small_hr/emails.py @@ -8,7 +8,7 @@ from django.utils.translation import ugettext as _ -def send_email( # pylint: disable=too-many-locals +def send_email( # pylint: disable=too-many-arguments, too-many-locals name: str, email: str, subject: str, message: str, obj: object = None, cc_list: list = None, template: str = 'generic'): """ diff --git a/tests/test_forms.py b/tests/test_forms.py index 7ccceb2..4576117 100644 --- a/tests/test_forms.py +++ b/tests/test_forms.py @@ -470,6 +470,8 @@ def test_leave_oversubscribe_off(self, mock): """ Test leave oversubscribe when SSHR_ALLOW_OVERSUBSCRIBE is False """ + mock.return_value = None + user = mommy.make('auth.User', first_name='Bob', last_name='Ndoe') staffprofile = mommy.make('small_small_hr.StaffProfile', user=user) staffprofile.leave_days = 21