From d7e12ea8a713ca2f3c788498e05764f3faad16ac Mon Sep 17 00:00:00 2001 From: Sergio Oliveira Date: Thu, 26 Nov 2015 15:48:56 -0200 Subject: [PATCH] Added test cases for user sync signals --- colab/accounts/models.py | 5 +++-- colab/accounts/tests/test_user.py | 19 +++++++++++++++++++ 2 files changed, 22 insertions(+), 2 deletions(-) diff --git a/colab/accounts/models.py b/colab/accounts/models.py index 49322f3..a6ca51f 100644 --- a/colab/accounts/models.py +++ b/colab/accounts/models.py @@ -14,9 +14,10 @@ from .utils import mailman class ColabUserManager(UserManager): - def _create_user(self, *args, **kwargs): + def _create_user(self, username, email, password, + is_staff, is_superuser, **kwargs): + args = (username, email, password, is_staff, is_superuser) user = super(ColabUserManager, self)._create_user(*args, **kwargs) - password = kwargs.get('password') user_created.send(user.__class__, user=user, password=password) return user diff --git a/colab/accounts/tests/test_user.py b/colab/accounts/tests/test_user.py index a57b9e3..558c64f 100644 --- a/colab/accounts/tests/test_user.py +++ b/colab/accounts/tests/test_user.py @@ -426,3 +426,22 @@ class UserTest(TestCase): self.authenticate_user() response = self.client.get(url, follow=True) self.assertIn(message, response.content) + + @mock.patch('colab.accounts.signals.user_password_changed.send') + @mock.patch('colab.accounts.signals.user_created.send') + def test_user_created_signal(self, user_created_send, + user_password_changed_send): + user = User.objects.create_user( + username='test_user', + password='12345', + email='test@example.com', + ) + user_created_send.assert_called_with(User, user=user, password='12345') + user_password_changed_send.assert_not_called() + + @mock.patch('colab.accounts.signals.user_password_changed.send') + def test_user_password_changed_signal(self, user_password_changed_send): + user = User.objects.first() + user.set_password('54321') + user_password_changed_send.assert_called_with(User, user=user, + password='54321') -- libgit2 0.21.2