chore: import upstream snapshot with attribution

2026-07-13 13:32:57 +08:00
commit cd420f9332
4811 changed files with 884702 additions and 0 deletions
@@ -0,0 +1,109 @@
+import { describe, expect, it, vi } from "vitest";
+
+// Module-level db wiring is imported transitively by the service file. The mint
+// helper under test never touches the DB (it is driven with injected deps), so
+// these empty singletons only satisfy the import graph — same boundary pattern
+// as triggerTask.server.test.ts and runEngineBatchTriggerStoreRouting.test.ts.
+vi.mock("~/db.server", () => ({
+  prisma: {},
+  $replica: {},
+  runOpsNewPrisma: {},
+  runOpsLegacyPrisma: {},
+  runOpsNewReplica: {},
+  runOpsLegacyReplica: {},
+}));
+
+import { BatchId, generateRunOpsId, ownerEngine, RunId } from "@trigger.dev/core/v3/isomorphic";
+import type { AuthenticatedEnvironment } from "~/services/apiAuth.server";
+import { BatchTriggerV3Service } from "~/v3/services/batchTriggerV3.server";
+
+vi.setConfig({ testTimeout: 60_000 });
+
+const CUID_LEN = 25;
+const RUN_OPS_ID_LEN = 26;
+
+// Minimal AuthenticatedEnvironment — only the fields the mint path reads
+// (organizationId, id, organization.featureFlags) need to be real. A root batch
+// (no parentRunId) with no run-ops id override mints cuid, which is the env-default
+// branch we assert on below.
+function fakeEnv(): AuthenticatedEnvironment {
+  return {
+    id: "env_123",
+    organizationId: "org_123",
+    organization: { featureFlags: {} },
+  } as unknown as AuthenticatedEnvironment;
+}
+
+// Build the service with resolveMintKind forced to "cuid" (its production default
+// when split is off / org not cut over), proving the CHILD branch overrides the env
+// default purely from the parent's id-shape.
+function buildService() {
+  return new BatchTriggerV3Service(undefined, undefined, {} as any, {} as any, async () => "cuid");
+}
+
+describe("BatchTriggerV3Service child-residency inheritance", () => {
+  it("a run-ops parent yields run-ops id (NEW) child friendlyIds", async () => {
+    const service = buildService();
+    const parentFriendlyId = RunId.toFriendlyId(
+      // v1 internal id (version "1" at index 25) → NEW residency parent
+      "a".repeat(RUN_OPS_ID_LEN - 1) + "1"
+    );
+    expect(ownerEngine(RunId.fromFriendlyId(parentFriendlyId))).toBe("NEW");
+
+    const childFriendlyId = await (service as any).mintChildFriendlyId(fakeEnv(), parentFriendlyId);
+
+    expect(RunId.fromFriendlyId(childFriendlyId).length).toBe(RUN_OPS_ID_LEN);
+    expect(ownerEngine(RunId.fromFriendlyId(childFriendlyId))).toBe("NEW");
+  });
+
+  it("a cuid parent yields cuid (LEGACY) child friendlyIds", async () => {
+    const service = buildService();
+    const parentFriendlyId = RunId.generate().friendlyId; // cuid (25) → LEGACY parent
+    expect(ownerEngine(RunId.fromFriendlyId(parentFriendlyId))).toBe("LEGACY");
+
+    const childFriendlyId = await (service as any).mintChildFriendlyId(fakeEnv(), parentFriendlyId);
+
+    expect(RunId.fromFriendlyId(childFriendlyId).length).toBe(CUID_LEN);
+    expect(ownerEngine(RunId.fromFriendlyId(childFriendlyId))).toBe("LEGACY");
+  });
+
+  it("a ROOT batch (no parentRunId) mints by the env setting (cuid default here)", async () => {
+    const service = buildService();
+    const childFriendlyId = await (service as any).mintChildFriendlyId(fakeEnv(), undefined);
+    expect(RunId.fromFriendlyId(childFriendlyId).length).toBe(CUID_LEN);
+    expect(ownerEngine(RunId.fromFriendlyId(childFriendlyId))).toBe("LEGACY");
+  });
+
+  // A root batch's children are anchored to the batch's friendlyId, NOT to a
+  // re-resolution of the per-org flag. Even with the env flag forced to "cuid" (a flip
+  // away from the batch's residency), a run-ops batch anchor yields run-ops children — so
+  // batch + children stay co-resident and TaskRun.batchId never crosses the seam.
+  it("a run-ops batch anchor yields run-ops children even when the env flag resolves cuid", async () => {
+    const service = buildService(); // resolveMintKind forced to "cuid"
+    const batchFriendlyId = BatchId.toFriendlyId(generateRunOpsId()); // run-ops id (NEW) batch
+    expect(ownerEngine(batchFriendlyId)).toBe("NEW");
+
+    const childFriendlyId = await (service as any).mintChildFriendlyId(fakeEnv(), batchFriendlyId);
+
+    expect(RunId.fromFriendlyId(childFriendlyId).length).toBe(RUN_OPS_ID_LEN);
+    expect(ownerEngine(RunId.fromFriendlyId(childFriendlyId))).toBe("NEW");
+  });
+
+  // The cuid mirror: a cuid batch anchor yields cuid children even if the flag flipped ON.
+  it("a cuid batch anchor yields cuid children even when the env flag resolves 'runOpsId'", async () => {
+    const service = new BatchTriggerV3Service(
+      undefined,
+      undefined,
+      {} as any,
+      {} as any,
+      async () => "runOpsId" // env flag flipped ON mid-batch
+    );
+    const batchFriendlyId = BatchId.generate().friendlyId; // cuid (LEGACY) batch
+    expect(ownerEngine(batchFriendlyId)).toBe("LEGACY");
+
+    const childFriendlyId = await (service as any).mintChildFriendlyId(fakeEnv(), batchFriendlyId);
+
+    expect(RunId.fromFriendlyId(childFriendlyId).length).toBe(CUID_LEN);
+    expect(ownerEngine(RunId.fromFriendlyId(childFriendlyId))).toBe("LEGACY");
+  });
+});