improve error reporting

golemfactory · Dec 22, 2021 · 1259892 · 1259892
1 parent 1de9c6d
commit 1259892
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/goth/assertions/assertions.py b/goth/assertions/assertions.py
@@ -152,7 +152,7 @@ async def func_wrapper():
             try:
                 return await self._func(self)
             except asyncio.CancelledError:
-                raise AssertionError("Assertion cancelled")
+                raise AssertionError(f"Assertion cancelled: {self}")
             finally:
                 self._notify_update_events()
 

diff --git a/goth/runner/probe/__init__.py b/goth/runner/probe/__init__.py
@@ -8,6 +8,7 @@
 import logging
 import os
 from pathlib import Path
+import traceback
 from typing import (
     AsyncIterator,
     Dict,
@@ -394,9 +395,10 @@ async def run_command_on_host(
                 logger.debug("Command task has finished")
 
         except Exception as e:
-            logger.error(f"Cancelling command on error: {e!r}")
+            logger.error(f"Cancelling command on error: {e!r}, command: `{command}`")
             if cmd_task and not cmd_task.done():
                 cmd_task.cancel()
+            traceback.print_exc()
             raise
 
         finally: