470 lines
21 KiB
Python
470 lines
21 KiB
Python
"""
|
||
Тесты.
|
||
"""
|
||
|
||
|
||
import random
|
||
from unittest.mock import patch, Mock
|
||
|
||
from django.contrib.auth import get_user_model
|
||
from django.core import mail
|
||
from django.http import HttpResponseRedirect
|
||
from django.template.loader import render_to_string
|
||
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_send_email(self):
|
||
"""
|
||
Функция тестирования отправки email.
|
||
"""
|
||
with self.settings(EMAIL_BACKEND=self.email_backend):
|
||
response: HttpResponseRedirect = \
|
||
self.client.post(reverse('registration'), data={'email': self.any_zendesk_user_email})
|
||
self.assertEqual(response.status_code, 302)
|
||
self.assertEqual(len(mail.outbox), 1)
|
||
self.assertEqual(mail.outbox[0].to, [self.any_zendesk_user_email])
|
||
|
||
# context that the email template was rendered with
|
||
email_context = response.context[0].dicts[1]
|
||
correct_subject = render_to_string('registration/password_reset_subject.txt', email_context, response.request)
|
||
self.assertEqual(mail.outbox[0].subject, correct_subject.strip())
|
||
correct_body = render_to_string('registration/password_reset_email.html', email_context, response.request)
|
||
self.assertEqual(mail.outbox[0].body, correct_body)
|
||
|
||
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, zenpy_mock):
|
||
"""
|
||
Функция проверки переадресации пользователя на рабочую страницу после назначения роли 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)
|
||
self.assertFalse(zenpy_mock.called)
|
||
|
||
@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'])
|
||
|
||
|
||
class PasswordResetTestCase(TestCase):
|
||
fixtures = ['fixtures/test_make_engineer.json']
|
||
|
||
def setUp(self):
|
||
self.user = '123@test.ru'
|
||
self.email_backend = 'django.core.mail.backends.locmem.EmailBackend'
|
||
self.client = Client()
|
||
self.client.force_login(get_user_model().objects.get(email=self.user))
|
||
|
||
def test_redirect(self):
|
||
with self.settings(EMAIL_BACKEND=self.email_backend):
|
||
resp = self.client.post(reverse_lazy('password_reset'), data={'email': self.user})
|
||
self.assertRedirects(resp, reverse('password_reset_done'))
|
||
self.assertEqual(resp.status_code, 302)
|
||
|
||
def test_send_email(self):
|
||
with self.settings(EMAIL_BACKEND=self.email_backend):
|
||
response: HttpResponseRedirect = \
|
||
self.client.post(reverse_lazy('password_reset'), data={'email': self.user})
|
||
self.assertEqual(response.status_code, 302)
|
||
self.assertEqual(len(mail.outbox), 1)
|
||
self.assertEqual(mail.outbox[0].to, [self.user])
|
||
|
||
# context that the email template was rendered with
|
||
email_context = response.context[0].dicts[1]
|
||
correct_subject = render_to_string('registration/password_reset_subject.txt', email_context, response.request)
|
||
self.assertEqual(mail.outbox[0].subject, correct_subject.strip())
|
||
correct_body = render_to_string('registration/password_reset_email.html', email_context, response.request)
|
||
self.assertEqual(mail.outbox[0].body, correct_body)
|
||
|
||
def test_email_invalid(self):
|
||
with self.settings(EMAIL_BACKEND=self.email_backend) and translation.override('ru'):
|
||
resp = self.client.post(reverse_lazy('password_reset'), data={'email': 1})
|
||
self.assertContains(resp, 'Введите правильный адрес электронной почты.', count=1, status_code=200)
|
||
|
||
def test_user_does_not_exist(self):
|
||
with self.settings(EMAIL_BACKEND=self.email_backend):
|
||
resp = self.client.post(reverse_lazy('password_reset'), data={'email': self.user + str(random.random())})
|
||
self.assertRedirects(resp, reverse('password_reset_done'))
|
||
self.assertEqual(resp.status_code, 302)
|
||
self.assertEqual(len(mail.outbox), 0)
|
||
|
||
|
||
class PasswordChangeTestCase(TestCase):
|
||
fixtures = ['fixtures/test_make_engineer.json']
|
||
|
||
def setUp(self):
|
||
self.user = '123@test.ru'
|
||
self.client = Client()
|
||
self.client.force_login(get_user_model().objects.get(email=self.user))
|
||
self.set_password()
|
||
|
||
def set_password(self):
|
||
user: get_user_model() = get_user_model().objects.get(email=self.user)
|
||
user.set_password('ImpossiblyHardPassword')
|
||
user.save()
|
||
self.client.force_login(get_user_model().objects.get(email=self.user))
|
||
|
||
def test_change_successful(self):
|
||
self.client.post(
|
||
reverse_lazy('password_change'),
|
||
data={
|
||
'old_password': 'ImpossiblyHardPassword',
|
||
'new_password1': 'EasyPassword',
|
||
'new_password2': 'EasyPassword',
|
||
}
|
||
)
|
||
user = get_user_model().objects.get(email=self.user)
|
||
self.assertTrue(user.check_password('EasyPassword'))
|
||
|
||
def test_invalid_old_password(self):
|
||
with translation.override('ru'):
|
||
resp = self.client.post(
|
||
reverse_lazy('password_change'),
|
||
data={
|
||
'old_password': 'EasyPassword',
|
||
'new_password1': 'EasyPassword',
|
||
'new_password2': 'EasyPassword',
|
||
}
|
||
)
|
||
self.assertContains(resp, 'Ваш старый пароль введен неправильно', count=1, status_code=200)
|
||
|
||
def test_different_new_passwords(self):
|
||
with translation.override('ru'):
|
||
resp = self.client.post(
|
||
reverse_lazy('password_change'),
|
||
data={
|
||
'old_password': 'ImpossiblyHardPassword',
|
||
'new_password1': 'EasyPassword',
|
||
'new_password2': 'EasyPassword1',
|
||
}
|
||
)
|
||
self.assertContains(resp, 'Введенные пароли не совпадают', count=1, status_code=200)
|
||
|
||
def test_invalid_new_password1(self):
|
||
with translation.override('ru'):
|
||
resp = self.client.post(
|
||
reverse_lazy('password_change'),
|
||
data={
|
||
'old_password': 'ImpossiblyHardPassword',
|
||
'new_password1': 'short',
|
||
'new_password2': 'short',
|
||
}
|
||
)
|
||
self.assertContains(resp, 'Введённый пароль слишком короткий', count=1, status_code=200)
|
||
|
||
def test_invalid_new_password2(self):
|
||
with translation.override('ru'):
|
||
resp = self.client.post(
|
||
reverse_lazy('password_change'),
|
||
data={
|
||
'old_password': 'ImpossiblyHardPassword',
|
||
'new_password1': '123123123123123123132123123',
|
||
'new_password2': '123123123123123123132123123',
|
||
}
|
||
)
|
||
self.assertContains(resp, 'Введённый пароль состоит только из цифр', count=1, status_code=200)
|
||
|
||
def test_invalid_new_password3(self):
|
||
with translation.override('ru'):
|
||
resp = self.client.post(
|
||
reverse_lazy('password_change'),
|
||
data={
|
||
'old_password': 'ImpossiblyHardPassword',
|
||
'new_password1': self.user,
|
||
'new_password2': self.user,
|
||
}
|
||
)
|
||
self.assertContains(resp, 'Введённый пароль слишком похож на имя пользователя', count=1, status_code=200)
|
||
|
||
|
||
class GetTicketsTestCase(TestCase):
|
||
"""
|
||
Класс тестов для проверки функции получения тикетов.
|
||
"""
|
||
fixtures = ['fixtures/test_make_engineer.json']
|
||
|
||
def setUp(self):
|
||
"""
|
||
Предустановленные значения для проведения тестов.
|
||
"""
|
||
self.light_agent = '123@test.ru'
|
||
self.engineer = 'customer@example.com'
|
||
self.client = Client()
|
||
self.client.force_login(get_user_model().objects.get(email=self.engineer))
|
||
self.light_agent_client = Client()
|
||
self.light_agent_client.force_login(get_user_model().objects.get(email=self.light_agent))
|
||
|
||
@patch('main.views.zenpy.get_user')
|
||
@patch('main.extra_func.zenpy')
|
||
def test_redirect(self, ZenpyMock, GetUserMock):
|
||
"""
|
||
Функция проверки переадресации пользователя на рабочую страницу.
|
||
"""
|
||
GetUserMock.return_value = Mock()
|
||
user = get_user_model().objects.get(email=self.engineer)
|
||
resp = self.client.post(reverse('work_get_tickets'))
|
||
self.assertRedirects(resp, reverse('work', args=[user.id]))
|
||
self.assertEqual(resp.status_code, 302)
|
||
|
||
@patch('main.views.zenpy')
|
||
@patch('main.views.get_tickets_list_for_group')
|
||
def test_take_one_ticket(self, TicketsMock, ZenpyViewsMock):
|
||
"""
|
||
Функция проверки назначения одного тикета на engineer.
|
||
"""
|
||
TicketsMock.return_value = [Mock()]
|
||
ZenpyViewsMock.get_user.return_value = Mock(role='agent', custom_role_id=sets.ZENDESK_ROLES['engineer'])
|
||
self.client.post(reverse('work_get_tickets'), data={'count_tickets': 1})
|
||
tickets = ZenpyViewsMock.update_tickets.call_args
|
||
self.assertEqual(tickets[0][0][0].assignee, ZenpyViewsMock.get_user.return_value)
|
||
|
||
@patch('main.views.get_tickets_list_for_group')
|
||
@patch('main.views.zenpy')
|
||
def test_take_many_tickets(self, ZenpyMock, TicketsMock):
|
||
"""
|
||
Функция проверки назначения нескольких тикетов на engineer.
|
||
"""
|
||
TicketsMock.return_value = [Mock()] * 3
|
||
ZenpyMock.get_user.return_value = Mock(role='agent', custom_role_id=sets.ZENDESK_ROLES['engineer'])
|
||
self.client.post(reverse('work_get_tickets'), data={'count_tickets': 3})
|
||
tickets = ZenpyMock.update_tickets.call_args
|
||
for ticket in tickets[0][0]:
|
||
self.assertEqual(ticket.assignee, ZenpyMock.get_user.return_value)
|
||
|
||
@patch('main.views.zenpy.get_user')
|
||
@patch('main.views.zenpy')
|
||
def test_light_agent_take_ticket(self, ZenpyMock, GetUserMock):
|
||
"""
|
||
Функция проверки попытки назначения тикета на light_agent.
|
||
"""
|
||
GetUserMock.return_value = Mock(role='agent', custom_role_id=sets.ZENDESK_ROLES['light_agent'])
|
||
self.light_agent_client.post(reverse('work_get_tickets'), data={'count_tickets': 3})
|
||
tickets = ZenpyMock.update_tickets.call_args
|
||
self.assertIsNone(tickets)
|
||
|
||
@patch('main.views.zenpy')
|
||
@patch('main.views.get_tickets_list_for_group')
|
||
def test_take_zero_tickets(self, TicketsMock, ZenpyMock):
|
||
"""
|
||
Функция проверки попытки назначения нуля тикета на engineer.
|
||
"""
|
||
TicketsMock.return_value = [Mock()] * 3
|
||
ZenpyMock.get_user.return_value = Mock(role='agent', custom_role_id=sets.ZENDESK_ROLES['engineer'])
|
||
self.client.post(reverse('work_get_tickets'), data={'count_tickets': 0})
|
||
tickets = ZenpyMock.update_tickets.call_args[0][0]
|
||
self.assertListEqual(tickets, [])
|
||
|
||
@patch('main.views.get_tickets_list_for_group')
|
||
@patch('main.views.zenpy')
|
||
def test_take_invalid_count_tickets(self, ZenpyMock, TicketsMock, ):
|
||
"""
|
||
Функция проверки попытки назначения нуля тикетов на engineer.
|
||
"""
|
||
TicketsMock.return_value = [Mock()] * 3
|
||
ZenpyMock.get_user.return_value = Mock(role='agent', custom_role_id=sets.ZENDESK_ROLES['engineer'])
|
||
self.client.post(reverse('work_get_tickets'), data={'count_tickets': 'asd'})
|
||
tickets = ZenpyMock.update_tickets.call_args
|
||
self.assertIsNone(tickets)
|
||
|
||
|
||
class ProfileTestCase(TestCase):
|
||
"""
|
||
Класс тестов для проверки синхронизации профиля пользователя.
|
||
"""
|
||
fixtures = ['fixtures/profile.json']
|
||
|
||
def setUp(self):
|
||
"""
|
||
Предустановленные значения для проведения тестов.
|
||
"""
|
||
self.zendesk_agent_email = 'krav-88@mail.ru'
|
||
self.zendesk_admin_email = 'idar.sokurov.05@mail.ru'
|
||
self.client = Client()
|
||
self.client.force_login(get_user_model().objects.get(email=self.zendesk_agent_email))
|
||
self.admin_client = Client()
|
||
self.admin_client.force_login(get_user_model().objects.get(email=self.zendesk_admin_email))
|
||
|
||
def test_correct_username(self):
|
||
"""
|
||
Функция проверки синхронизации имени пользователя.
|
||
"""
|
||
resp = self.client.get(reverse('profile'))
|
||
self.assertEqual(resp.context['profile'].name, zenpy.get_user(self.zendesk_agent_email).name)
|
||
|
||
def test_correct_email(self):
|
||
"""
|
||
Функция проверки синхронизации почты пользователя.
|
||
"""
|
||
resp = self.client.get(reverse('profile'))
|
||
self.assertEqual(resp.context['profile'].user.email, zenpy.get_user(self.zendesk_agent_email).email)
|
||
|
||
def test_correct_role(self):
|
||
"""
|
||
Функция проверки синхронизации роли пользователя.
|
||
"""
|
||
resp = self.client.get(reverse('profile'))
|
||
self.assertEqual(resp.context['profile'].role, zenpy.get_user(self.zendesk_agent_email).role)
|
||
resp = self.admin_client.get(reverse('profile'))
|
||
self.assertEqual(resp.context['profile'].role, zenpy.get_user(self.zendesk_admin_email).role)
|
||
|
||
def test_correct_custom_role_id(self):
|
||
"""
|
||
Функция проверки синхронизации рабочей роли пользователя.
|
||
"""
|
||
resp = self.client.get(reverse('profile'))
|
||
user = zenpy.get_user(self.zendesk_agent_email)
|
||
self.assertEqual(resp.context['profile'].custom_role_id, user.custom_role_id if user.custom_role_id else 0)
|
||
resp = self.admin_client.get(reverse('profile'))
|
||
user = zenpy.get_user(self.zendesk_admin_email)
|
||
self.assertEqual(resp.context['profile'].custom_role_id, user.custom_role_id if user.custom_role_id else 0)
|
||
|
||
def test_correct_image(self):
|
||
"""
|
||
Функция проверки синхронизации изображения пользователя.
|
||
"""
|
||
resp = self.client.get(reverse('profile'))
|
||
user = zenpy.get_user(self.zendesk_agent_email)
|
||
self.assertEqual(resp.context['profile'].image, user.photo['content_url'] if user.photo else None)
|