Skip to content

Commit b4d9526

Browse files
registerTaskDef with retryCount 0 for TaskManager tests
1 parent b8d1ef2 commit b4d9526

1 file changed

Lines changed: 14 additions & 2 deletions

File tree

src/task/__tests__/TaskManager.test.ts

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import { expect, describe, test, jest } from "@jest/globals";
2-
import { simpleTask, WorkflowExecutor } from "../../core";
2+
import { simpleTask, taskDefinition, WorkflowExecutor } from "../../core";
33
import { orkesConductorClient } from "../../orkes";
44
import { TaskManager, ConductorWorker } from "../index";
55
import { mockLogger } from "./mockLogger";
@@ -69,6 +69,12 @@ describe("TaskManager", () => {
6969
},
7070
};
7171

72+
await client.metadataResource.registerTaskDef([taskDefinition({
73+
name: "taskmanager-error-handler-test-unique",
74+
timeoutSeconds: 0,
75+
retryCount: 0,
76+
})]);
77+
7278
const manager = new TaskManager(client, [worker], {
7379
options: { pollInterval: BASE_TIME },
7480
onError: mockErrorHandler,
@@ -93,7 +99,7 @@ describe("TaskManager", () => {
9399
correlationId: "errorHandlerTestIdentifierUnique"
94100
});
95101

96-
const workflowStatus = await TestUtil.waitForWorkflowCompletion(executor, status, BASE_TIME * 4);
102+
const workflowStatus = await TestUtil.waitForWorkflowCompletion(executor, status, BASE_TIME * 6);
97103

98104
expect(workflowStatus.status).toEqual("FAILED");
99105
expect(mockErrorHandler).toBeCalledTimes(1);
@@ -111,6 +117,12 @@ describe("TaskManager", () => {
111117
},
112118
};
113119

120+
await client.metadataResource.registerTaskDef([taskDefinition({
121+
name: "taskmanager-error-test",
122+
timeoutSeconds: 0,
123+
retryCount: 0,
124+
})]);
125+
114126
const manager = new TaskManager(client, [worker], {
115127
options: { pollInterval: BASE_TIME },
116128
});

0 commit comments

Comments
 (0)