Merge pull request #591 from superannotateai/async_threads

Added tests for the async functions in thread

Author: nareksa

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

@@ -2,6 +2,7 @@
 import copy
 import io
 import json
+import logging
 import os
 import sys
 import warnings
@@ -63,7 +64,10 @@
 from lib.core.types import Project
 from lib.infrastructure.utils import extract_project_folder
 from lib.infrastructure.validators import wrap_error
-import logging
+
+import nest_asyncio
+
+nest_asyncio.apply()
 
 logger = logging.getLogger("sa")
 

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

@@ -1,4 +1,3 @@
-import nest_asyncio
 from lib.core.usecases.annotations import *  # noqa: F403 F401
 from lib.core.usecases.classes import *  # noqa: F403 F401
 from lib.core.usecases.custom_fields import *  # noqa: F403 F401
@@ -8,5 +7,3 @@
 from lib.core.usecases.items import *  # noqa: F403 F401
 from lib.core.usecases.models import *  # noqa: F403 F401
 from lib.core.usecases.projects import *  # noqa: F403 F401
-
-nest_asyncio.apply()

tests/unit/test_async_functions.py

@@ -1,4 +1,5 @@
 import asyncio
+import concurrent.futures
 from unittest import TestCase
 
 from superannotate import SAClient
@@ -81,10 +82,26 @@ async def gather_test():
 
     def test_upload_annotations_in_running_event_loop(self):
         async def _test():
-            sa.attach_items(self.PROJECT_NAME, self.ATTACH_PAYLOAD)
             annotations = sa.upload_annotations(
                 self.PROJECT_NAME, annotations=self.UPLOAD_PAYLOAD
             )
             assert len(annotations["succeeded"]) == 4
 
         asyncio.run(_test())
+
+    def test_upload_in_threads(self):
+        def _test():
+            annotations = sa.upload_annotations(
+                self.PROJECT_NAME, annotations=self.UPLOAD_PAYLOAD
+            )
+            assert len(annotations["succeeded"]) == 4
+            return True
+
+        with concurrent.futures.ThreadPoolExecutor(max_workers=8) as executor:
+            futures = []
+            for i in range(8):
+                futures.append(executor.submit(_test))
+            results = []
+            for f in concurrent.futures.as_completed(futures):
+                results.append(f.result())
+            assert all(results)