fix: worker reuse when worker was timeouted

Author: janbritz

questionpy_server/worker/exception.py

@@ -5,23 +5,27 @@
 from questionpy_server.worker.runtime.messages import BaseWorkerError
 
 
-class WorkerNotRunningError(BaseWorkerError):
-    pass
+class WorkerStoppedError(BaseWorkerError):
+    """Represents an error that results in or was caused by a worker being stopped."""
 
 
 class WorkerStartError(BaseWorkerError):
     pass
 
 
-class WorkerCPUTimeLimitExceededError(BaseWorkerError):
+class WorkerNotRunningError(WorkerStoppedError):
+    pass
+
+
+class WorkerCPUTimeLimitExceededError(WorkerStoppedError):
     def __init__(self, limit: float, worker_name: str):
         self.limit = limit
         super().__init__(
             f"Worker has exceeded its CPU time limit of {limit} seconds and was killed.", worker_name=worker_name
         )
 
 
-class WorkerRealTimeLimitExceededError(BaseWorkerError):
+class WorkerRealTimeLimitExceededError(WorkerStoppedError):
     def __init__(self, limit: float, worker_name: str):
         self.limit = limit
         super().__init__(

questionpy_server/worker/impl/_base.py

@@ -29,6 +29,7 @@
     WorkerNotRunningError,
     WorkerRealTimeLimitExceededError,
     WorkerStartError,
+    WorkerStoppedError,
 )
 from questionpy_server.worker.runtime.messages import (
     BaseWorkerError,
@@ -141,7 +142,7 @@ async def send_and_wait_for_response(
         self.state = WorkerState.SERVER_AWAITS_RESPONSE
         try:
             result = await fut
-        except WorkerNotRunningError:
+        except WorkerStoppedError:
             self.state = WorkerState.NOT_RUNNING
             raise
         finally: