""" Тесты. """ from unittest.mock import patch 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, reverse_lazy 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) -> None: """ Функция предтестовых настроек. """ 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) -> None: """ Функция тестирования успешно завершенной регистрации. """ 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): """ Функция тестирования отправки email. # 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 enumerate(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): """ Функция тестирования регистрации пользователя (сверяем имя с именем в Zendesk. """ 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): """ Функция тестирования проверке присвоения роли admin. """ 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')) class MakeEngineerTestCase(TestCase): """ Класс тестов для проверки функции назначения роли engineer. """ fixtures = ['fixtures/test_make_engineer.json'] def setUp(self): """ Предустановленные значения для проведения тестов. """ self.light_agent = '123@test.ru' self.admin = 'admin@gmail.com' self.engineer = 'customer@example.com' self.client = Client() self.client.force_login(get_user_model().objects.get(email=self.light_agent)) self.admin_client = Client() self.admin_client.force_login(get_user_model().objects.get(email=self.admin)) @patch('main.extra_func.zenpy') def test_redirect(self): """ Функция проверки переадресации пользователя на рабочую страницу после назначения роли engineer. """ user = get_user_model().objects.get(email=self.light_agent) resp = self.client.post(reverse_lazy('work_become_engineer')) self.assertRedirects(resp, reverse('work', args=[user.id])) self.assertEqual(resp.status_code, 302) @patch('main.extra_func.zenpy') def test_light_agent_make_engineer(self, zenpy_mock): """ Функция проверки назначения light_agent на роль engineer. """ self.client.post(reverse_lazy('work_become_engineer')) self.assertEqual(zenpy_mock.update_user.call_args[0][0].custom_role_id, sets.ZENDESK_ROLES['engineer']) @patch('main.extra_func.zenpy') def test_admin_make_engineer(self, zenpy_mock): """ Функция проверки назначения admin на роль engineer. """ self.admin_client.post(reverse_lazy('work_become_engineer')) self.assertEqual(zenpy_mock.update_user.call_args[0][0].custom_role_id, sets.ZENDESK_ROLES['engineer']) @patch('main.extra_func.zenpy') def test_engineer_make_engineer(self, zenpy_mock): """ Функция проверки назначения engineer на роль engineer. """ client = Client() client.force_login(get_user_model().objects.get(email=self.engineer)) client.post(reverse_lazy('work_become_engineer')) self.assertEqual(zenpy_mock.update_user.call_args[0][0].custom_role_id, sets.ZENDESK_ROLES['engineer']) @patch('main.extra_func.zenpy') def test_control_page_make_one(self, zenpy_mock): """ Функция проверки назначения администратором на роль engineer одного пользователя. """ self.admin_client.post( reverse_lazy('control'), data={'users': [get_user_model().objects.get(email=self.light_agent).userprofile.id], 'engineer': 'engineer'} ) call_list = zenpy_mock.update_user.call_args_list mock_object = call_list[0][0][0] self.assertEqual(len(call_list), 1) self.assertEqual(mock_object.custom_role_id, sets.ZENDESK_ROLES['engineer']) @patch('main.extra_func.zenpy') def test_control_page_make_many(self, zenpy_mock): """ Функция проверки назначения администратором на роль engineer нескольких пользователей. """ self.admin_client.post( reverse_lazy('control'), data={ 'users': [ get_user_model().objects.get(email=self.light_agent).userprofile.id, get_user_model().objects.get(email=self.engineer).userprofile.id, ], 'engineer': 'engineer' } ) call_list = zenpy_mock.update_user.call_args_list mock_objects = list(call_list) self.assertEqual(len(call_list), 2) for obj in mock_objects: self.assertEqual(obj[0][0].custom_role_id, sets.ZENDESK_ROLES['engineer'])