middleware working and tests passing

Author: ericallam

internal-packages/run-engine/src/engine/errors.ts

@@ -13,6 +13,9 @@ export function runStatusFromError(error: TaskRunError): TaskRunStatus {
   //e.g. a bug
   switch (error.code) {
     case "RECURSIVE_WAIT_DEADLOCK":
+    case "TASK_INPUT_ERROR":
+    case "TASK_OUTPUT_ERROR":
+    case "TASK_MIDDLEWARE_ERROR":
       return "COMPLETED_WITH_ERRORS";
     case "TASK_RUN_CANCELLED":
       return "CANCELED";
@@ -41,8 +44,6 @@ export function runStatusFromError(error: TaskRunError): TaskRunStatus {
     case "TASK_RUN_STALLED_EXECUTING_WITH_WAITPOINTS":
     case "TASK_HAS_N0_EXECUTION_SNAPSHOT":
     case "GRACEFUL_EXIT_TIMEOUT":
-    case "TASK_INPUT_ERROR":
-    case "TASK_OUTPUT_ERROR":
     case "POD_EVICTED":
     case "POD_UNKNOWN_ERROR":
     case "TASK_EXECUTION_ABORTED":

packages/core/src/v3/errors.ts

@@ -305,6 +305,7 @@ export function shouldRetryError(error: TaskRunError): boolean {
         case "HANDLE_ERROR_ERROR":
         case "TASK_INPUT_ERROR":
         case "TASK_OUTPUT_ERROR":
+        case "TASK_MIDDLEWARE_ERROR":
         case "POD_EVICTED":
         case "POD_UNKNOWN_ERROR":
         case "TASK_EXECUTION_ABORTED":

packages/core/src/v3/schemas/common.ts

@@ -165,6 +165,7 @@ export const TaskRunInternalError = z.object({
     "TASK_RUN_CANCELLED",
     "TASK_INPUT_ERROR",
     "TASK_OUTPUT_ERROR",
+    "TASK_MIDDLEWARE_ERROR",
     "HANDLE_ERROR_ERROR",
     "GRACEFUL_EXIT_TIMEOUT",
     "TASK_RUN_HEARTBEAT_TIMEOUT",