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,65 @@
+import type * as Core from "@trigger.dev/core/v3";
+import { realtimeStreams } from "@trigger.dev/core/v3";
+import { beforeEach, describe, expect, it, vi } from "vitest";
+import { streams } from "./streams.js";
+
+vi.mock("@trigger.dev/core/v3", async (importOriginal) => {
+  const original = await importOriginal<typeof Core>();
+  return {
+    ...original,
+    taskContext: {
+      ctx: {
+        run: {
+          id: "run_123",
+          // parentTaskRunId and rootTaskRunId are undefined for root tasks
+        },
+      },
+    },
+    realtimeStreams: {
+      pipe: vi.fn().mockReturnValue({
+        wait: () => Promise.resolve(),
+        stream: new ReadableStream(),
+      }),
+    },
+  };
+});
+
+describe("streams.pipe consistency", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it("should not throw and should use self runId when target is 'root' in a root task", async () => {
+    const mockStream = new ReadableStream();
+
+    // This should not throw anymore
+    const { waitUntilComplete: _waitUntilComplete } = streams.pipe("test-key", mockStream, {
+      target: "root",
+    });
+
+    expect(realtimeStreams.pipe).toHaveBeenCalledWith(
+      "test-key",
+      mockStream,
+      expect.objectContaining({
+        target: "run_123",
+      })
+    );
+  });
+
+  it("should not throw and should use self runId when target is 'parent' in a root task", async () => {
+    const mockStream = new ReadableStream();
+
+    // This should not throw anymore
+    const { waitUntilComplete: _waitUntilComplete } = streams.pipe("test-key", mockStream, {
+      target: "parent",
+    });
+
+    expect(realtimeStreams.pipe).toHaveBeenCalledWith(
+      "test-key",
+      mockStream,
+      expect.objectContaining({
+        target: "run_123",
+      })
+    );
+  });
+});