Fix task polling and test sleep mock

Author: IgorChvyrov-sm

src/conductor/client/automator/task_runner.py

@@ -93,11 +93,15 @@ def __poll_task(self) -> Task:
                 )
             if auth_exception.invalid_token:
                 logger.error(
-                    "Failed to poll task: %s; reason: invalid auth token", task_definition_name
+                    "Failed to poll task: %s; reason: invalid auth token",
+                    task_definition_name,
                 )
             else:
                 logger.error(
-                    "Failed to poll task: %s; status: %s - {auth_exception.error_code}", task_definition_name, auth_exception.status, auth_exception.error_code
+                    "Failed to poll task: %s; status: %s - %s",
+                    task_definition_name,
+                    auth_exception.status,
+                    auth_exception.error_code,
                 )
             return None
         except ApiException as e:
@@ -106,11 +110,20 @@ def __poll_task(self) -> Task:
                     task_definition_name, type(e)
                 )
             logger.error(
-                "Failed to poll task for: %s, reason: %s",
+                "Failed to poll task: %s, reason: %s, code: %s",
                 task_definition_name,
-                traceback.format_exc(),
+                e.reason,
+                e.code,
             )
             return None
+        except Exception as e:
+            if self.metrics_collector is not None:
+                self.metrics_collector.increment_task_poll_error(
+                    task_definition_name, type(e)
+                )
+            logger.error("Failed to poll task: %s; reason: %s", task_definition_name, e)
+            return None
+
         if task is not None:
             logger.debug(
                 "Polled task: %s; worker_id: %s; domain: %s",

tests/unit/automator/test_async_task_runner.py

@@ -279,7 +279,7 @@ async def test_update_task_with_invalid_task_result():
 
 @pytest.mark.asyncio
 async def test_update_task_with_faulty_task_api(mocker):
-    mocker.patch("time.sleep", return_value=None)
+    mocker.patch("asyncio.sleep", return_value=None)
     mocker.patch.object(TaskResourceApiAdapter, "update_task", side_effect=Exception())
     task_runner = get_valid_task_runner()
     task_result = get_valid_task_result()