Add test

Author: Marishka17

cvat/apps/engine/tests/test_rest_api.py

@@ -80,31 +80,40 @@
 logging.getLogger("libav").setLevel(logging.ERROR)
 
 
-def create_db_users(cls):
-    (group_admin, _) = Group.objects.get_or_create(name="admin")
-    (group_user, _) = Group.objects.get_or_create(name="user")
-    (group_annotator, _) = Group.objects.get_or_create(name="worker")
-    (group_somebody, _) = Group.objects.get_or_create(name="somebody")
-
-    user_admin = User.objects.create_superuser(username="admin", email="", password="admin")
-    user_admin.groups.add(group_admin)
-    user_owner = User.objects.create_user(username="user1", password="user1")
-    user_owner.groups.add(group_user)
-    user_assignee = User.objects.create_user(username="user2", password="user2")
-    user_assignee.groups.add(group_annotator)
-    user_annotator = User.objects.create_user(username="user3", password="user3")
-    user_annotator.groups.add(group_annotator)
-    user_somebody = User.objects.create_user(username="user4", password="user4")
-    user_somebody.groups.add(group_somebody)
-    user_dummy = User.objects.create_user(username="user5", password="user5")
-    user_dummy.groups.add(group_user)
-
-    cls.admin = user_admin
-    cls.owner = cls.user1 = user_owner
-    cls.assignee = cls.user2 = user_assignee
-    cls.annotator = cls.user3 = user_annotator
-    cls.somebody = cls.user4 = user_somebody
-    cls.user = cls.user5 = user_dummy
+def create_db_users(
+    cls: type[ApiTestBase],
+    *,
+    admin: bool = True,
+    primary: bool = True,
+    extra: bool = True,
+):
+    if admin:
+        (group_admin, _) = Group.objects.get_or_create(name="admin")
+        user_admin = User.objects.create_superuser(username="admin", email="", password="admin")
+        user_admin.groups.add(group_admin)
+        cls.admin = user_admin
+
+    if primary:
+        (group_user, _) = Group.objects.get_or_create(name="user")
+        (group_annotator, _) = Group.objects.get_or_create(name="worker")
+        user_owner = User.objects.create_user(username="user1", password="user1")
+        user_owner.groups.add(group_user)
+        user_assignee = User.objects.create_user(username="user2", password="user2")
+        user_assignee.groups.add(group_annotator)
+        user_annotator = User.objects.create_user(username="user3", password="user3")
+        user_annotator.groups.add(group_annotator)
+        cls.owner = cls.user1 = user_owner
+        cls.assignee = cls.user2 = user_assignee
+        cls.annotator = cls.user3 = user_annotator
+
+    if extra:
+        (group_somebody, _) = Group.objects.get_or_create(name="somebody")
+        user_somebody = User.objects.create_user(username="user4", password="user4")
+        user_somebody.groups.add(group_somebody)
+        user_dummy = User.objects.create_user(username="user5", password="user5")
+        user_dummy.groups.add(group_user)
+        cls.somebody = cls.user4 = user_somebody
+        cls.user = cls.user5 = user_dummy
 
 
 def create_db_task(data):

cvat/apps/iam/tests/test_rest_api.py

@@ -4,12 +4,14 @@
 # SPDX-License-Identifier: MIT
 
 from allauth.account.views import EmailVerificationSentView
+from django.contrib.auth.models import User
 from django.test import override_settings
 from django.urls import path, re_path, reverse
 from rest_framework import status
 from rest_framework.authtoken.models import Token
-from rest_framework.test import APIClient, APITestCase
 
+from cvat.apps.engine.tests.test_rest_api import create_db_users
+from cvat.apps.engine.tests.utils import ApiTestBase
 from cvat.apps.iam.urls import urlpatterns as iam_url_patterns
 from cvat.apps.iam.views import ConfirmEmailViewEx
 
@@ -27,24 +29,7 @@
 ]
 
 
-class ForceLogin:
-    def __init__(self, user, client):
-        self.user = user
-        self.client = client
-
-    def __enter__(self):
-        if self.user:
-            self.client.force_login(self.user, backend="django.contrib.auth.backends.ModelBackend")
-
-        return self
-
-    def __exit__(self, exception_type, exception_value, traceback):
-        if self.user:
-            self.client.logout()
-
-
-class UserRegisterAPITestCase(APITestCase):
-
+class UserRegisterAPITestCase(ApiTestBase):
     user_data = {
         "first_name": "test_first",
         "last_name": "test_last",
@@ -55,8 +40,10 @@ class UserRegisterAPITestCase(APITestCase):
         "confirmations": [],
     }
 
-    def setUp(self):
-        self.client = APIClient()
+    @classmethod
+    def setUpTestData(cls):
+        # create only admin account
+        create_db_users(cls, primary=False, extra=False)
 
     def _run_api_v2_user_register(self, data):
         url = reverse("rest_register")
@@ -130,3 +117,49 @@ def test_register_account_with_email_verification_mandatory(self):
                 "key": None,
             },
         )
+
+    @override_settings(
+        ACCOUNT_EMAIL_REQUIRED=True,
+        ACCOUNT_EMAIL_VERIFICATION="mandatory",
+        EMAIL_BACKEND="django.core.mail.backends.console.EmailBackend",
+    )
+    def test_register_account_with_different_email_case_than_in_invitation(self):
+        """
+        Ensure a user can log in to the account after being invited to an organization
+        and then registering with the same email but in a different case.
+        """
+        org_slug = "testorg"
+        response = self._post_request(
+            "/api/organizations", self.admin, data={"slug": org_slug, "name": "Test organization"}
+        )
+        self.assertTrue(response.status_code == status.HTTP_201_CREATED)
+        response = self._post_request(
+            "/api/invitations",
+            self.admin,
+            data={"role": "worker", "email": self.user_data["email"].upper()},
+            query_params={"org": org_slug},
+        )
+        self.assertTrue(response.status_code == status.HTTP_201_CREATED)
+
+        response = self._run_api_v2_user_register(self.user_data)
+        self._check_response(
+            response,
+            {
+                "first_name": "test_first",
+                "last_name": "test_last",
+                "username": "test_username",
+                "email": "[email protected]",
+                "email_verification_required": True,
+                "key": None,
+            },
+        )
+        invited_db_user = User.objects.get(email=self.user_data["email"])
+        self.assertTrue(invited_db_user.emailaddress_set.update(verified=True))
+        response = self.client.post(
+            "/api/auth/login",
+            format="json",
+            data={"email": self.user_data["email"], "password": self.user_data["password1"]},
+        )
+        self.assertTrue(response.status_code == status.HTTP_200_OK)
+        self.assertIn("sessionid", response.cookies)
+        self.assertIn("csrftoken", response.cookies)