Fixed pin image

Author: VaghinakDev

pytest.ini

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

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,6 +5,7 @@
 
 import pytest
 
+from src.superannotate import AppException
 from src.superannotate import SAClient
 from tests.integration.base import BaseTestCase
 
@@ -46,11 +47,15 @@ def test_vector_annotations_with_tag(self):
         metadata = sa.get_item_metadata(self.PROJECT_NAME, item_name=self.EXAMPLE_IMAGE_1)
         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):
-        sa.attach_items(self.PROJECT_NAME, [{"url": "some url", "name": self.EXAMPLE_IMAGE_1}])
-        item = sa.search_items(self.PROJECT_NAME)[0]
-        assert not item['is_pinned']
-        sa.pin_image(self.PROJECT_NAME, self.EXAMPLE_IMAGE_1 + 'NEW NAME')
+        with self.assertRaisesRegexp(AppException, "Item not found."):
+            sa.pin_image(self.PROJECT_NAME, 'NEW NAME')
 
 
 class TestVectorAnnotationsWithTagFolderUpload(BaseTestCase):