feat: added additional tests

src/task/__tests__/TaskRunner.test.ts

@@ -1,64 +1,118 @@
-import {jest, test, expect} from "@jest/globals"
-import type {Mocked} from "jest-mock"
+import { jest, test, expect } from "@jest/globals";
+import type { Mocked } from "jest-mock";
 
-import {RunnerArgs, TaskRunner} from "../TaskRunner"
-import {mockLogger} from "./mockLogger"
-import {TaskResourceService} from "../../common/open-api"
+import { RunnerArgs, TaskRunner } from "../TaskRunner";
+import { mockLogger } from "./mockLogger";
+import { TaskResourceService } from "../../common/open-api";
 
-const taskClientStub: Mocked<Pick<TaskResourceService, "poll" | "updateTask1">> = {
-  poll: jest.fn(),
-  updateTask1: jest.fn()
-}
-const mockTaskClient = taskClientStub as unknown as TaskResourceService
-
-test('polls tasks', async () => {
-  const workerID = "worker-id"
+test("polls tasks", async () => {
+  const taskClientStub: Mocked<
+    Pick<TaskResourceService, "poll" | "updateTask1">
+  > = {
+    poll: jest.fn(),
+    updateTask1: jest.fn(),
+  };
+  const mockTaskClient = taskClientStub as unknown as TaskResourceService;
+  const workerID = "worker-id";
   const args: RunnerArgs = {
     worker: {
       taskDefName: "test",
-      execute: async ({inputData}) =>  {
+      execute: async ({ inputData }) => {
         return {
           outputData: {
-            "hello": "from worker",
-            ...inputData
+            hello: "from worker",
+            ...inputData,
           },
-          status: "COMPLETED"
-        }
+          status: "COMPLETED",
+        };
       },
     },
     options: {
       pollInterval: 10,
-      domain: '',
+      domain: "",
       concurrency: 1,
-      workerID
+      workerID,
     },
     logger: mockLogger,
-    taskResource: mockTaskClient
-  }
-  const workflowInstanceId =  "fake-workflow-id"
-  const taskId = "fake-task-id"
+    taskResource: mockTaskClient,
+  };
+  const workflowInstanceId = "fake-workflow-id";
+  const taskId = "fake-task-id";
   taskClientStub.poll.mockResolvedValue({
     taskId,
     workflowInstanceId,
     status: "IN_PROGRESS",
     reasonForIncompletion: undefined,
     inputData: {
-      "input": "from workflow"
-    }
-  })
+      input: "from workflow",
+    },
+  });
 
-  const runner = new TaskRunner(args)
-  runner.startPolling()
+  const runner = new TaskRunner(args);
+  runner.startPolling();
   await new Promise((r) => setTimeout(() => r(true), 10));
-  runner.stopPolling()
+  runner.stopPolling();
   const expected = {
     taskId,
     workflowInstanceId,
     status: "COMPLETED",
     outputData: {
-      "hello": "from worker",
-      "input": "from workflow"
-    }
-  }
-  expect(taskClientStub.updateTask1).toHaveBeenCalledWith(expected)
-})
+      hello: "from worker",
+      input: "from workflow",
+    },
+  };
+  expect(taskClientStub.updateTask1).toHaveBeenCalledWith(expected);
+});
+
+test("Should set the task as failed if the task has an error", async () => {
+  const taskClientStub: Mocked<
+    Pick<TaskResourceService, "poll" | "updateTask1">
+  > = {
+    poll: jest.fn(),
+    updateTask1: jest.fn(),
+  };
+  const mockTaskClient = taskClientStub as unknown as TaskResourceService;
+
+  const workerID = "worker-id";
+  const args: RunnerArgs = {
+    worker: {
+      taskDefName: "test",
+      execute: async ({ inputData }) => {
+        throw new Error("Error from worker");
+      },
+    },
+    options: {
+      pollInterval: 10,
+      domain: "",
+      concurrency: 1,
+      workerID,
+    },
+    logger: mockLogger,
+    taskResource: mockTaskClient,
+  };
+  const workflowInstanceId = "fake-workflow-id";
+  const taskId = "fake-task-id";
+  taskClientStub.poll.mockResolvedValue({
+    taskId,
+    workflowInstanceId,
+    status: "IN_PROGRESS",
+    reasonForIncompletion: undefined,
+    inputData: {
+      input: "from workflow",
+    },
+  });
+
+  const runner = new TaskRunner(args);
+  runner.startPolling();
+  await new Promise((r) => setTimeout(() => r(true), 10));
+  runner.stopPolling();
+  const expected = {
+    taskId,
+    workflowInstanceId,
+    status: "FAILED",
+    outputData: {
+    },
+    reasonForIncompletion: "Error from worker",
+  };
+  expect(taskClientStub.updateTask1).toHaveBeenCalledWith(expected);
+});