pin image fix

Author: VaghinakDev

src/superannotate/lib/app/interface/sdk_interface.py

@@ -703,11 +703,14 @@ def pin_image(
         :type pin: bool
         """
         project, folder = self.controller.get_project_folder_by_path(project)
-        item = self.controller.items.get_by_name(project, folder, image_name).data
+        response = self.controller.items.get_by_name(project, folder, image_name)
+        if response.errors:
+            raise AppException(response.errors)
+        item = response.data
         item.is_pinned = int(pin)
         self.controller.items.update(
             project=project,
-            item=item,
+            item=item
         )
 
     def delete_items(self, project: str, items: Optional[List[str]] = None):

tests/integration/test_basic_images.py

@@ -5,10 +5,11 @@
 
 import pytest
 
+from src.superannotate import AppException
 from src.superannotate import SAClient
+from tests.integration.base import BaseTestCase
 
 sa = SAClient()
-from tests.integration.base import BaseTestCase
 
 
 class TestVectorAnnotationsWithTag(BaseTestCase):
@@ -44,7 +45,17 @@ def test_vector_annotations_with_tag(self):
         self.assertEqual(annotations['instances'][0]['attributes'][0]['groupName'], 'g1')
         sa.pin_image(self.PROJECT_NAME, self.EXAMPLE_IMAGE_1, pin=True)
         metadata = sa.get_item_metadata(self.PROJECT_NAME, item_name=self.EXAMPLE_IMAGE_1)
-        assert metadata['is_pinned'] == True
+        assert metadata['is_pinned'] is True
+
+
+class TestPinImage(BaseTestCase):
+    PROJECT_NAME = "TestPinImage"
+    PROJECT_TYPE = "Vector"
+    PROJECT_DESCRIPTION = "TestPinImage"
+
+    def test_pin_image_negative_name(self):
+        with self.assertRaisesRegexp(AppException, "Item not found."):
+            sa.pin_image(self.PROJECT_NAME, 'NEW NAME')
 
 
 class TestVectorAnnotationsWithTagFolderUpload(BaseTestCase):