Fix workflow get

Author: dshabin

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

@@ -701,8 +701,9 @@ def execute(self):
                 if self._fill_classes:
                     annotation_classes = self._annotation_classes.get_all()
                     for annotation_class in annotation_classes:
-                        annotation_class.uuid = workflow.class_id
-                        workflow_data["className"] = annotation_class.name
+                        if annotation_class.id == workflow.class_id:
+                            workflow_data["className"] = annotation_class.name
+                            break
                 data.append(workflow_data)
             self._response.data = data
         return self._response

tests/integration/projects/test_basic_project.py

@@ -9,6 +9,93 @@
 from tests.integration.base import BaseTestCase
 
 
+class TestWorkflowGet(BaseTestCase):
+    PROJECT_NAME = "TestWorkflowGet"
+    PROJECT_TYPE = "Vector"
+    PROJECT_DESCRIPTION = "DESCRIPTION"
+
+    def test_workflow_get(self):
+        sa.create_annotation_class(
+            self.PROJECT_NAME,
+            "class1",
+            "#FFAAFF",
+            [
+                {
+                    "name": "tall",
+                    "is_multiselect": 0,
+                    "attributes": [{"name": "yes"}, {"name": "no"}],
+                },
+                {
+                    "name": "age",
+                    "is_multiselect": 0,
+                    "attributes": [{"name": "young"}, {"name": "old"}],
+                },
+            ],
+        )
+        sa.create_annotation_class(
+            self.PROJECT_NAME,
+            "class2",
+            "#FFAAFF",
+            [
+                {
+                    "name": "tall",
+                    "is_multiselect": 0,
+                    "attributes": [{"name": "yes"}, {"name": "no"}],
+                },
+                {
+                    "name": "age",
+                    "is_multiselect": 0,
+                    "attributes": [{"name": "young"}, {"name": "old"}],
+                },
+            ],
+        )
+        sa.set_project_workflow(
+            self.PROJECT_NAME,
+            [
+                {
+                    "step": 1,
+                    "className": "class1",
+                    "tool": 3,
+                    "attribute": [
+                        {
+                            "attribute": {
+                                "name": "young",
+                                "attribute_group": {"name": "age"},
+                            }
+                        },
+                        {
+                            "attribute": {
+                                "name": "yes",
+                                "attribute_group": {"name": "tall"},
+                            }
+                        },
+                    ],
+                },
+                {
+                    "step": 2,
+                    "className": "class2",
+                    "tool": 3,
+                    "attribute": [
+                        {
+                            "attribute": {
+                                "name": "young",
+                                "attribute_group": {"name": "age"},
+                            }
+                        },
+                        {
+                            "attribute": {
+                                "name": "yes",
+                                "attribute_group": {"name": "tall"},
+                            }
+                        },
+                    ],
+                }
+            ],
+        )
+        workflows = sa.get_project_workflow(self.PROJECT_NAME)
+        self.assertEqual(workflows[0]['className'], "class1")
+        self.assertEqual(workflows[1]['className'], "class2")
+
 
 class TestProjectCreateMetadata(BaseTestCase):
     PROJECT_NAME = "sample_basic_project"