updated item tests

Author: Narek Mkhitaryan

tests/integration/items/test_list_items.py

@@ -1,4 +1,6 @@
 import os
+import random
+import string
 from pathlib import Path
 
 from src.superannotate import AppException
@@ -47,3 +49,15 @@ def test_invalid_filter(self):
             sa.list_items(self.PROJECT_NAME, assignments__user_role="Dummy")
         with self.assertRaisesRegexp(AppException, "Invalid status provided."):
             sa.list_items(self.PROJECT_NAME, annotation_status="Dummy")
+
+    def test_list_items_URL_limit(self):
+        items_for_attache = []
+        item_names = []
+        for i in range(125):
+            name = f"{''.join(random.choice(string.ascii_letters + string.digits) for _ in range(120))}"
+            item_names.append(name)
+            items_for_attache.append({"name": name, "url": f"{name}-{i}"})
+
+        sa.attach_items(self.PROJECT_NAME, items_for_attache)
+        items = sa.list_items(self.PROJECT_NAME, name__in=item_names)
+        assert len(items) == 125

tests/integration/items/test_move_items.py

@@ -17,6 +17,17 @@ class TestMoveItems(BaseTestCase):
     FOLDER_2 = "folder_2"
     CSV_PATH = "data_set/attach_urls.csv"
 
+    Attachment = [
+        {
+            "url": "https://drive.google.com/uc?export=download&id=1vwfCpTzcjxoEA4hhDxqapPOVvLVeS7ZS",
+            "name": IMAGE_NAME,
+        },
+        {
+            "url": "https://drive.google.com/uc?export=download&id=1vwfCpTzcjxoEA4hhDxqapPOVvLVeS7Zw",
+            "name": IMAGE_NAME_2,
+        },
+    ]
+
     @property
     def scv_path(self):
         return os.path.join(Path(__file__).parent.parent.parent, self.CSV_PATH)
@@ -46,21 +57,11 @@ def test_move_items_from_folder(self):
         assert len(sa.search_items(f"{self.PROJECT_NAME}/{self.FOLDER_2}")) == 7
         assert len(sa.search_items(f"{self.PROJECT_NAME}/{self.FOLDER_1}")) == 0
 
-    def test_move_items_from_folder_with_replace_strategy(self):
-        attachment = [
-            {
-                "url": "https://drive.google.com/uc?export=download&id=1vwfCpTzcjxoEA4hhDxqapPOVvLVeS7ZS",
-                "name": self.IMAGE_NAME,
-            },
-            {
-                "url": "https://drive.google.com/uc?export=download&id=1vwfCpTzcjxoEA4hhDxqapPOVvLVeS7Zw",
-                "name": self.IMAGE_NAME_2,
-            },
-        ]
+    def test_move_items_from_folder_with_replace(self):
         sa.create_folder(self.PROJECT_NAME, self.FOLDER_1)
         sa.create_folder(self.PROJECT_NAME, self.FOLDER_2)
         uploaded, _, _ = sa.attach_items(
-            f"{self.PROJECT_NAME}/{self.FOLDER_1}", attachment
+            f"{self.PROJECT_NAME}/{self.FOLDER_1}", self.Attachment
         )
         assert len(uploaded) == 2
         sa.set_approval_statuses(
@@ -75,7 +76,7 @@ def test_move_items_from_folder_with_replace_strategy(self):
         )
 
         uploaded_2, _, _ = sa.attach_items(
-            f"{self.PROJECT_NAME}/{self.FOLDER_2}", attachment
+            f"{self.PROJECT_NAME}/{self.FOLDER_2}", self.Attachment
         )
         assert len(uploaded_2) == 2
         folder_2_items = sa.search_items(f"{self.PROJECT_NAME}/{self.FOLDER_2}")
@@ -96,3 +97,64 @@ def test_move_items_from_folder_with_replace_strategy(self):
         folder_2_items = sa.search_items(f"{self.PROJECT_NAME}/{self.FOLDER_2}")
         assert folder_2_items[0]["annotation_status"] == "Completed"
         assert folder_2_items[0]["approval_status"] == "Approved"
+
+    def test_move_items_from_folder_with_replace_annotations_only(self):
+        sa.create_folder(self.PROJECT_NAME, self.FOLDER_1)
+        sa.create_folder(self.PROJECT_NAME, self.FOLDER_2)
+        uploaded, _, _ = sa.attach_items(
+            f"{self.PROJECT_NAME}/{self.FOLDER_1}", self.Attachment
+        )
+        assert len(uploaded) == 2
+        sa.set_approval_statuses(
+            f"{self.PROJECT_NAME}/{self.FOLDER_1}",
+            "Approved",
+            items=[self.IMAGE_NAME, self.IMAGE_NAME_2],
+        )
+        sa.set_annotation_statuses(
+            f"{self.PROJECT_NAME}/{self.FOLDER_1}",
+            "Completed",
+            items=[self.IMAGE_NAME, self.IMAGE_NAME_2],
+        )
+
+        uploaded_2, _, _ = sa.attach_items(
+            f"{self.PROJECT_NAME}/{self.FOLDER_2}", self.Attachment
+        )
+        assert len(uploaded_2) == 2
+        folder_2_items = sa.search_items(f"{self.PROJECT_NAME}/{self.FOLDER_2}")
+        assert folder_2_items[0]["annotation_status"] == "NotStarted"
+        assert not folder_2_items[0]["approval_status"]
+
+        skipped_items = sa.move_items(
+            f"{self.PROJECT_NAME}/{self.FOLDER_1}",
+            f"{self.PROJECT_NAME}/{self.FOLDER_2}",
+            duplicate_strategy="replace_annotations_only",
+        )
+        assert len(skipped_items) == 0
+        folder_1_items = sa.search_items(f"{self.PROJECT_NAME}/{self.FOLDER_1}")
+        folder_2_items = sa.search_items(f"{self.PROJECT_NAME}/{self.FOLDER_2}")
+        assert len(folder_1_items) == 0
+        assert len(folder_2_items) == 2
+
+        folder_2_items = sa.search_items(f"{self.PROJECT_NAME}/{self.FOLDER_2}")
+        assert folder_2_items[0]["annotation_status"] == "NotStarted"
+        assert not folder_2_items[0]["approval_status"]
+
+    def test_move_items_from_folder_with_skip(self):
+        sa.create_folder(self.PROJECT_NAME, self.FOLDER_1)
+        sa.create_folder(self.PROJECT_NAME, self.FOLDER_2)
+        uploaded, _, _ = sa.attach_items(
+            f"{self.PROJECT_NAME}/{self.FOLDER_1}", self.Attachment
+        )
+        assert len(uploaded) == 2
+
+        uploaded_2, _, _ = sa.attach_items(
+            f"{self.PROJECT_NAME}/{self.FOLDER_2}", self.Attachment
+        )
+        assert len(uploaded_2) == 2
+
+        skipped_items = sa.move_items(
+            f"{self.PROJECT_NAME}/{self.FOLDER_1}",
+            f"{self.PROJECT_NAME}/{self.FOLDER_2}",
+            duplicate_strategy="skip",
+        )
+        assert len(skipped_items) == 2