Merge pull request #803 from superannotateai/fix_categories

Fix categories

Author: nareksa

tests/integration/work_management/test_contributors_categories.py

@@ -32,7 +32,7 @@ def setUpClass(cls, *args, **kwargs) -> None:
             cls.PROJECT_TYPE,
             settings=[
                 {"attribute": "TemplateState", "value": 1},
-                {"attribute": "CategorizeItems", "value": 1},
+                {"attribute": "CategorizeItems", "value": 2},
             ],
         )
         team = sa.controller.team
@@ -138,15 +138,20 @@ def test_set_contributors_categories_by_id(self):
         test_categories = ["ID_Cat_A", "ID_Cat_B"]
         sa.create_categories(project=self.PROJECT_NAME, categories=test_categories)
 
+        scapegoat_project_id = sa.list_users(
+            project=self.PROJECT_NAME,
+            email=self.scapegoat["email"],
+        )[0]["id"]
+
         sa.set_contributors_categories(
             project=self.PROJECT_NAME,
-            contributors=[self.scapegoat["id"]],
+            contributors=[scapegoat_project_id],
             categories=test_categories,
         )
 
         # Verify categories were assigned
         project_users = sa.list_users(
-            project=self.PROJECT_NAME, id=self.scapegoat["id"], include=["categories"]
+            project=self.PROJECT_NAME, id=scapegoat_project_id, include=["categories"]
         )
         assigned_categories = [cat["name"] for cat in project_users[0]["categories"]]
         for category in test_categories: