Skip to content

Commit fa8bfc9

Browse files
committed
fix worker group test again
1 parent 0b0b7da commit fa8bfc9

File tree

1 file changed

+21
-6
lines changed

1 file changed

+21
-6
lines changed

apps/webapp/test/workerGroup.test.ts

Lines changed: 21 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,6 @@ import {
1818
import { HEADER_NAME } from "@trigger.dev/worker";
1919
import { RunEngine } from "@internal/run-engine";
2020
import { trace } from "@opentelemetry/api";
21-
import { TriggerTaskService } from "~/v3/services/triggerTask.server";
22-
import { env } from "~/env.server";
2321

2422
describe("worker", () => {
2523
const defaultInstanceName = "test_worker";
@@ -70,7 +68,7 @@ describe("worker", () => {
7068
headers: {
7169
Authorization: `Bearer ${token.plaintext}`,
7270
[HEADER_NAME.WORKER_INSTANCE_NAME]: defaultInstanceName,
73-
[HEADER_NAME.WORKER_MANAGED_SECRET]: env.MANAGED_WORKER_SECRET,
71+
[HEADER_NAME.WORKER_MANAGED_SECRET]: "managed-secret",
7472
},
7573
});
7674

@@ -88,7 +86,7 @@ describe("worker", () => {
8886
headers: {
8987
Authorization: `Bearer ${token.plaintext}`,
9088
[HEADER_NAME.WORKER_INSTANCE_NAME]: secondInstanceName,
91-
[HEADER_NAME.WORKER_MANAGED_SECRET]: env.MANAGED_WORKER_SECRET,
89+
[HEADER_NAME.WORKER_MANAGED_SECRET]: "managed-secret",
9290
},
9391
});
9492
const secondAuth = await tokenService.authenticate(secondRequest);
@@ -117,7 +115,6 @@ describe("worker", () => {
117115
assert(deployment, "deployment should be defined");
118116

119117
const engine = setupRunEngine(prisma, redisContainer);
120-
const triggerService = new TriggerTaskService({ prisma, engine });
121118

122119
const { token, workerGroupService, workerGroup } = await setupWorkerGroup({
123120
prisma,
@@ -151,7 +148,25 @@ describe("worker", () => {
151148
);
152149

153150
// Trigger
154-
const run = await triggerService.call(taskIdentifier, authenticatedEnvironment, {});
151+
const run = await engine.trigger(
152+
{
153+
number: 1,
154+
friendlyId: "run_1234",
155+
environment: authenticatedEnvironment,
156+
taskIdentifier,
157+
payload: "{}",
158+
payloadType: "application/json",
159+
context: {},
160+
traceContext: {},
161+
traceId: "t12345",
162+
spanId: "s12345",
163+
masterQueue: "main",
164+
queueName: "task/test-task",
165+
isTest: false,
166+
tags: [],
167+
},
168+
prisma
169+
);
155170
assert(run, "run should be defined");
156171

157172
// Check this is a V2 run

0 commit comments

Comments
 (0)