[unit tests] Remove sleep usage

cvat-ai · Jan 3, 2025 · ea45933 · ea45933
1 parent a82db50
commit ea45933
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/cvat/apps/engine/tests/test_rest_api.py b/cvat/apps/engine/tests/test_rest_api.py
@@ -3096,7 +3096,7 @@ def test_can_remove_export_cache_automatically_after_successful_export(self):
         task_id = self.tasks[0]["id"]
         user = self.admin
 
-        TASK_CACHE_TTL = timedelta(seconds=5)
+        TASK_CACHE_TTL = timedelta(seconds=20)
         with (
             mock.patch('cvat.apps.dataset_manager.views.TASK_CACHE_TTL', new=TASK_CACHE_TTL),
             mock.patch('cvat.apps.dataset_manager.views.TTL_CONSTS', new={'task': TASK_CACHE_TTL}),
@@ -3127,10 +3127,12 @@ def test_can_remove_export_cache_automatically_after_successful_export(self):
             file_path = job.return_value()
             self.assertTrue(os.path.isfile(file_path))
 
-            sleep(TASK_CACHE_TTL.total_seconds() + 1)
-
-            cron_export_cache_cleanup()
-            mock_clear_export_cache.assert_called_once()
+            with (
+                mock.patch('cvat.apps.dataset_manager.views.TASK_CACHE_TTL', new=timedelta(seconds=0)),
+                mock.patch('cvat.apps.dataset_manager.views.TTL_CONSTS', new={'task': timedelta(seconds=0)}),
+            ):
+                cron_export_cache_cleanup()
+                mock_clear_export_cache.assert_called_once()
             self.assertFalse(os.path.exists(file_path))