84 lines
4.1 KiB
Python
84 lines
4.1 KiB
Python
from urllib.parse import urlparse
|
||
|
||
from django.contrib.auth import get_user_model
|
||
from django.core import mail
|
||
from django.test import TestCase, Client
|
||
from django.urls import reverse
|
||
from django.utils import translation
|
||
|
||
import access_controller.settings as sets
|
||
from main.zendesk_admin import zenpy
|
||
|
||
|
||
class RegistrationTestCase(TestCase):
|
||
fixtures = ['fixtures/data.json']
|
||
|
||
def setUp(self):
|
||
self.email_backend = 'django.core.mail.backends.locmem.EmailBackend'
|
||
self.any_zendesk_user_email = 'idar.sokurov.05@mail.ru'
|
||
self.zendesk_admin_email = 'idar.sokurov.05@mail.ru'
|
||
self.client = Client()
|
||
|
||
def test_registration_complete_redirect(self):
|
||
with self.settings(EMAIL_BACKEND=self.email_backend):
|
||
resp = self.client.post(reverse('registration'), data={'email': self.any_zendesk_user_email})
|
||
self.assertRedirects(resp, reverse('password_reset_done'))
|
||
|
||
def test_registration_fail_redirect(self):
|
||
with self.settings(EMAIL_BACKEND=self.email_backend):
|
||
resp = self.client.post(reverse('registration'), data={'email': self.any_zendesk_user_email + 'asd'})
|
||
self.assertRedirects(resp, reverse('django_registration_disallowed'))
|
||
|
||
def test_registration_user_already_exist(self):
|
||
with self.settings(EMAIL_BACKEND=self.email_backend) and translation.override('ru'):
|
||
resp = self.client.post(reverse('registration'), data={'email': '123@test.ru'})
|
||
self.assertContains(resp, 'Этот адрес электронной почты уже используется', count=1, status_code=200)
|
||
|
||
def test_registration_email_sending(self):
|
||
# TODO: Найти способ лучше проверять сообщения
|
||
email_template = [
|
||
'',
|
||
'Вы получили это письмо, потому что вы (или кто-то другой) запросили восстановление пароля '
|
||
'от учётной записи на сайте testserver, которая связана с этим адресом электронной почты.',
|
||
'',
|
||
'Пожалуйста, перейдите на эту страницу и введите новый пароль:',
|
||
'',
|
||
'url',
|
||
'',
|
||
f'Ваше имя пользователя (на случай, если вы его забыли): {self.any_zendesk_user_email}',
|
||
'',
|
||
'Спасибо, что используете наш сайт!',
|
||
'',
|
||
'Команда сайта testserver',
|
||
'',
|
||
'',
|
||
'',
|
||
]
|
||
with self.settings(EMAIL_BACKEND=self.email_backend) and translation.override('ru'):
|
||
self.client.post(reverse('registration'), data={'email': self.any_zendesk_user_email})
|
||
|
||
self.assertEqual(len(mail.outbox), 1)
|
||
self.assertEqual(mail.outbox[0].to, [self.zendesk_admin_email])
|
||
self.assertEqual(mail.outbox[0].from_email, sets.DEFAULT_FROM_EMAIL)
|
||
|
||
message = mail.outbox[0].body.split('\n')
|
||
for i in range(len(message)):
|
||
if email_template[i] != 'url':
|
||
self.assertEqual(message[i], email_template[i])
|
||
else:
|
||
self.assertTrue(urlparse(message[i]).scheme)
|
||
|
||
def test_registration_user_creating(self):
|
||
with self.settings(EMAIL_BACKEND=self.email_backend):
|
||
self.client.post(reverse('registration'), data={'email': self.any_zendesk_user_email})
|
||
user = get_user_model().objects.get(email=self.any_zendesk_user_email)
|
||
zendesk_user = zenpy.get_user(self.any_zendesk_user_email)
|
||
self.assertEqual(user.userprofile.name, zendesk_user.name)
|
||
|
||
def test_permissions_applying(self):
|
||
with self.settings(EMAIL_BACKEND=self.email_backend):
|
||
self.client.post(reverse('registration'), data={'email': self.zendesk_admin_email})
|
||
user = get_user_model().objects.get(email=self.zendesk_admin_email)
|
||
self.assertEqual(user.userprofile.role, 'admin')
|
||
self.assertTrue(user.has_perm('main.has_control_access'))
|