def form_valid(self, form): """ If form is valid, send activation email and login user """ super(CreateIdentityView, self).form_valid(form) mailer = IdentityMailer(self.object) mailer.send_activation() token = mailer.token token.app_data.entree['origin_site'] = self.kwargs['origin_site'] token.app_data.entree['next_url'] = self.kwargs.get('next_url') token.save() return self.entree_login(self.object)
def form_valid(self, form): """ If form is valid, send activation email and login user """ super(CreateIdentityView, self).form_valid(form) mailer = IdentityMailer(self.object) mailer.send_activation() token = mailer.token token.extra_data['origin_site'] = self.kwargs['origin_site'] token.extra_data['next_url'] = self.kwargs.get('next_url') token.save() return self.entree_login(self.object)
def get(self, request, *args, **kwargs): mailer = IdentityMailer(request.entree_user) try: send_status = mailer.send_activation() except ValidationError: send_status = False if request.is_ajax(): return self.render_to_response({'send_status': send_status}) return HttpResponseRedirect(reverse("verify_identity"))
def get(self, request, *args, **kwargs): mailer = IdentityMailer(request.entree_user) try: send_status = mailer.send_activation() except ValidationError: send_status = False if request.is_ajax(): return self.render_to_response({'send_status': send_status}) return HttpResponseRedirect(reverse("verify_identity"))
class TestMailer(TestCase): def setUp(self): super(TestMailer, self).setUp() cache.clear() self.user = Identity.objects.create(email='*****@*****.**') self.mailer = IdentityMailer(self.user) def test_dont_send_already_activated_mail(self): self.user.mail_verified = True self.user.save() assert_raises(ValidationError, self.mailer.send_activation) def test_send_before_cooldown(self): self.mailer.send_activation() assert_raises(ValidationError, self.mailer.send_activation) def test_token_autocreate_on_send(self): assert_equals(True, self.mailer.send_activation()) def test_moar_tokens_exists_but_handled(self): t1 = self.user.create_token(token_type=MAIL_TOKEN) t2 = self.user.create_token(token_type=MAIL_TOKEN) t1.touched = datetime.now() - timedelta(days=1, minutes=2) t1.save() t2.touched = datetime.now() - timedelta(days=1, minutes=1) t2.save() token_left = self.user.create_token(token_type=MAIL_TOKEN) token_left.touched = datetime.now() - timedelta(days=1) token_left.save() assert_equals(True, self.mailer.send_activation()) assert_equals(1, LoginToken.objects.count()) assert_equals(token_left.pk, LoginToken.objects.all()[0].pk) def test_send_more_mails_cooldown_raises(self): assert_equals(True, self.mailer.send_activation()) assert_raises(ValidationError, self.mailer.send_activation) @patch.object(settings, 'DEBUG', True) def test_send_more_mails_cooldown_with_debug_pass(self): assert_equals(True, self.mailer.send_activation()) assert_equals(True, self.mailer.send_activation()) @patch('entree.enauth.mailer.send_mail') def test_mail_failed_token_timestamp_not_updated(self, mocked_send_mail): mocked_send_mail.side_effect = Exception('x') token_str = 'TOKENVAL' day_ago = datetime.now() - timedelta(days=1) original_token = LoginToken.objects.create(token_type=MAIL_TOKEN, touched=day_ago, value=token_str, user=self.user) assert_equals(False, self.mailer.send_activation()) updated_token = LoginToken.objects.get(value=token_str) assert_equals(updated_token.touched, original_token.touched) @patch('entree.enauth.mailer.send_mail') def test_mail_token_timestamp_updated(self, mocked_send_mail): mocked_send_mail.return_value = True day_ago = datetime.now() - timedelta(days=1) original_token = LoginToken.objects.create(token_type=MAIL_TOKEN, touched=day_ago, value='fo', user=self.user) assert_equals(True, self.mailer.send_activation()) updated_token = LoginToken.objects.get(pk=original_token.pk) assert_equals(True, (datetime.now() - updated_token.touched) < timedelta(seconds=10)) def test_pwd_reset_token_autocreated(self): self.mailer.send_pwd_reset_verify()
class TestMailer(TestCase): def setUp(self): super(TestMailer, self).setUp() cache.clear() self.user = Identity.objects.create(email='*****@*****.**') self.mailer = IdentityMailer(self.user) def test_dont_send_already_activated_mail(self): self.user.mail_verified = True self.user.save() assert_raises(ValidationError, self.mailer.send_activation) def test_send_before_cooldown(self): self.mailer.send_activation() assert_raises(ValidationError, self.mailer.send_activation) def test_token_autocreate_on_send(self): assert_equals(True, self.mailer.send_activation()) def test_moar_tokens_exists_but_handled(self): t1 = self.user.create_token(token_type=MAIL_TOKEN) t2 = self.user.create_token(token_type=MAIL_TOKEN) t1.touched = datetime.now() - timedelta(days=1, minutes=2) t1.save() t2.touched = datetime.now() - timedelta(days=1, minutes=1) t2.save() token_left = self.user.create_token(token_type=MAIL_TOKEN) token_left.touched = datetime.now() - timedelta(days=1) token_left.save() assert_equals(True, self.mailer.send_activation()) assert_equals(1, LoginToken.objects.count()) assert_equals(token_left.pk, LoginToken.objects.all()[0].pk) def test_send_more_mails_cooldown_raises(self): assert_equals(True, self.mailer.send_activation()) assert_raises(ValidationError, self.mailer.send_activation) @patch.object(settings, 'DEBUG', True) def test_send_more_mails_cooldown_with_debug_pass(self): assert_equals(True, self.mailer.send_activation()) assert_equals(True, self.mailer.send_activation()) @patch('entree.enauth.mailer.send_mail') def test_mail_failed_token_timestamp_not_updated(self, mocked_send_mail): mocked_send_mail.side_effect = Exception('x') token_str = 'TOKENVAL' day_ago = datetime.now() - timedelta(days=1) original_token = LoginToken.objects.create(token_type=MAIL_TOKEN, touched=day_ago, value=token_str, user=self.user) assert_equals(False, self.mailer.send_activation()) updated_token = LoginToken.objects.get(value=token_str) assert_equals(updated_token.touched, original_token.touched) @patch('entree.enauth.mailer.send_mail') def test_mail_token_timestamp_updated(self, mocked_send_mail): mocked_send_mail.return_value = True day_ago = datetime.now() - timedelta(days=1) original_token = LoginToken.objects.create(token_type=MAIL_TOKEN, touched=day_ago, value='fo', user=self.user) assert_equals(True, self.mailer.send_activation()) updated_token = LoginToken.objects.get(pk=original_token.pk) assert_equals( True, (datetime.now() - updated_token.touched) < timedelta(seconds=10)) def test_pwd_reset_token_autocreated(self): self.mailer.send_pwd_reset_verify()