Add test to for add_contributors_to_project (pending users)

Author: VaghinakDev

pytest.ini

@@ -3,4 +3,4 @@ minversion = 3.7
 log_cli=true
 python_files = test_*.py
 ;pytest_plugins = ['pytest_profiling']
-addopts = -n 6 --dist loadscope
+;addopts = -n 6 --dist loadscope

src/superannotate/lib/core/usecases/projects.py

@@ -860,12 +860,11 @@ def execute(self):
                 project_id=self._project.id,
             ).data
             project_emails = {user.email for user in project_users}
-            users = self._service_provider.work_management.list_users(EmptyQuery()).data
-            pending_invitations = []
+            users = self._service_provider.work_management.list_users(
+                EmptyQuery(), parent_entity=CustomFieldEntityEnum.TEAM
+            ).data
             for user in users:
-                if user.state == WMUserStateEnum.Pending.value:
-                    pending_invitations.append(user)
-                elif user.role == constants.UserRole.CONTRIBUTOR.value:
+                if user.role == constants.UserRole.CONTRIBUTOR.value:
                     team_users.add(user.email)
 
             role_email_map = defaultdict(list)

tests/integration/work_management/test_list_users.py

@@ -1,3 +1,4 @@
+import pytest
 from superannotate import SAClient
 from tests.integration.base import BaseTestCase
 
@@ -22,6 +23,14 @@ def setUp(self):
             self.PROJECT_NAME, [scapegoat["email"]], "Annotator"
         )
 
+    @pytest.mark.skip(reason="For not send real email")
+    def test_pending_users(self):
+        test_email = "test@superannotate.com"
+        sa.invite_contributors_to_team(emails=[test_email])
+        sa.add_contributors_to_project(self.PROJECT_NAME, [test_email], "Annotator")
+        project = sa.get_project_metadata(self.PROJECT_NAME, include_contributors=True)
+        assert project['contributors'][1]["state"] == "Pending"
+
     def test_list_users_by_project_name(self):
         project_users = sa.list_users(project=self.PROJECT_NAME)
         assert len(project_users) == 1