Pylint for lesson

main/models.py

@@ -40,9 +40,8 @@ class UserProfile(models.Model):
 
         :return: Роль пользователя в Zendesk
         """
-        id = self.custom_role_id
         for role, r_id in ZENDESK_ROLES.items():
-            if r_id == id:
+            if r_id == self.custom_role_id:
                 return role
         return 'UNDEFINED'
 

main/tests.py

@@ -1,3 +1,8 @@
+"""
+Тесты.
+"""
+
+
 from urllib.parse import urlparse
 
 from django.contrib.auth import get_user_model
@@ -11,31 +16,49 @@ from main.zendesk_admin import zenpy
 
 
 class RegistrationTestCase(TestCase):
+    """
+    Класс тестирования регистрации пользователя.
+    """
     fixtures = ['fixtures/data.json']
 
-    def setUp(self):
+    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):
+    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 = [
             '',
             'Вы получили это письмо, потому что вы (или кто-то другой) запросили восстановление пароля '
@@ -62,13 +85,16 @@ class RegistrationTestCase(TestCase):
         self.assertEqual(mail.outbox[0].from_email, sets.DEFAULT_FROM_EMAIL)
 
         message = mail.outbox[0].body.split('\n')
-        for i in range(len(message)):
+        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)
@@ -76,6 +102,9 @@ class RegistrationTestCase(TestCase):
         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)