wehub-resource-sync
112 lines
4.3 KiB
TypeScript
112 lines
4.3 KiB
TypeScript
import { containerTest } from "@internal/testcontainers";
|
|
import { trace } from "@internal/tracing";
|
|
import { describe, expect, vi } from "vitest";
|
|
import type { TriggerScheduledTaskParams } from "../src/engine/types.js";
|
|
import { ScheduleEngine } from "../src/index.js";
|
|
|
|
describe("ScheduleEngine Integration (part 2)", () => {
|
|
// Deploy-moment backward compatibility. At deploy time, in-flight Redis jobs
|
|
// were enqueued by the old engine — their payload has no `lastScheduleTime`
|
|
// field — and `instance.lastScheduledTimestamp` is still populated (last
|
|
// written by the old engine pre-deploy). The new engine must report that DB
|
|
// value as `payload.lastTimestamp` so customers don't see a transient
|
|
// `undefined` for the one fire per schedule that drains the legacy queue.
|
|
containerTest(
|
|
"should fall back to instance.lastScheduledTimestamp when payload lacks lastScheduleTime",
|
|
{ timeout: 30_000 },
|
|
async ({ prisma, redisOptions }) => {
|
|
const triggerCalls: TriggerScheduledTaskParams[] = [];
|
|
const engine = new ScheduleEngine({
|
|
prisma,
|
|
redis: redisOptions,
|
|
distributionWindow: { seconds: 10 },
|
|
worker: {
|
|
concurrency: 1,
|
|
disabled: true, // Don't actually run the worker — calling triggerScheduledTask directly
|
|
pollIntervalMs: 1000,
|
|
},
|
|
tracer: trace.getTracer("test", "0.0.0"),
|
|
onTriggerScheduledTask: async (params) => {
|
|
triggerCalls.push(params);
|
|
return { success: true };
|
|
},
|
|
isDevEnvironmentConnectedHandler: vi.fn().mockResolvedValue(true),
|
|
});
|
|
|
|
try {
|
|
const organization = await prisma.organization.create({
|
|
data: { title: "Legacy Payload Org", slug: "legacy-payload-org" },
|
|
});
|
|
|
|
const project = await prisma.project.create({
|
|
data: {
|
|
name: "Legacy Payload Project",
|
|
slug: "legacy-payload-project",
|
|
externalRef: "legacy-payload-ref",
|
|
organizationId: organization.id,
|
|
},
|
|
});
|
|
|
|
const environment = await prisma.runtimeEnvironment.create({
|
|
data: {
|
|
slug: "legacy-payload-env",
|
|
type: "PRODUCTION",
|
|
projectId: project.id,
|
|
organizationId: organization.id,
|
|
apiKey: "tr_legacy_1234",
|
|
pkApiKey: "pk_legacy_1234",
|
|
shortcode: "legacy-short",
|
|
},
|
|
});
|
|
|
|
const taskSchedule = await prisma.taskSchedule.create({
|
|
data: {
|
|
friendlyId: "sched_legacy_payload",
|
|
taskIdentifier: "legacy-payload-task",
|
|
projectId: project.id,
|
|
deduplicationKey: "legacy-payload-dedup",
|
|
userProvidedDeduplicationKey: false,
|
|
generatorExpression: "*/5 * * * *",
|
|
generatorDescription: "Every 5 minutes",
|
|
timezone: "UTC",
|
|
type: "DECLARATIVE",
|
|
active: true,
|
|
externalId: "legacy-ext",
|
|
},
|
|
});
|
|
|
|
// Pre-populate lastScheduledTimestamp on the instance — simulates the
|
|
// value the old engine wrote to the DB before this PR deployed.
|
|
const preDeployLastFire = new Date("2026-04-30T10:00:00.000Z");
|
|
const scheduleInstance = await prisma.taskScheduleInstance.create({
|
|
data: {
|
|
taskScheduleId: taskSchedule.id,
|
|
environmentId: environment.id,
|
|
projectId: project.id,
|
|
active: true,
|
|
lastScheduledTimestamp: preDeployLastFire,
|
|
},
|
|
});
|
|
|
|
// Call triggerScheduledTask directly without lastScheduleTime,
|
|
// simulating an in-flight Redis job enqueued by the old engine.
|
|
const exactScheduleTime = new Date("2026-04-30T10:05:00.000Z");
|
|
await engine.triggerScheduledTask({
|
|
instanceId: scheduleInstance.id,
|
|
finalAttempt: false,
|
|
exactScheduleTime,
|
|
// lastScheduleTime intentionally omitted — legacy payload shape
|
|
});
|
|
|
|
expect(triggerCalls.length).toBe(1);
|
|
expect(triggerCalls[0].payload.timestamp).toEqual(exactScheduleTime);
|
|
// Falls back to instance.lastScheduledTimestamp from the DB rather
|
|
// than reporting undefined for this one transitional fire.
|
|
expect(triggerCalls[0].payload.lastTimestamp).toEqual(preDeployLastFire);
|
|
} finally {
|
|
await engine.quit();
|
|
}
|
|
}
|
|
);
|
|
});
|