| ... | ... | @@ -2850,6 +2850,73 @@ it.layer(layerTest)("ProjectionStoreV2", (it) => { |
| 2850 | 2850 | }), |
| 2851 | 2851 | ); |
| 2852 | 2852 | |
| 2853 | it.effect("selects the latest waiting secret only from active runs", () => |
| 2854 | Effect.gen(function* () { |
| 2855 | const projectionStore = yield* ProjectionStore.ProjectionStoreV2; |
| 2856 | const sql = yield* SqlClient.SqlClient; |
| 2857 | const suffix = "shell-pending-secret"; |
| 2858 | const threadId = yield* addRolledBackRecoveryCandidate(suffix); |
| 2859 | const runId = RunId.make(`run:${suffix}:rolled-back`); |
| 2860 | const nodeId = NodeId.make(`node:${suffix}:rolled-back`); |
| 2861 | const now = yield* DateTime.now; |
| 2862 | const addSecret = (id: string, ordinal: number, status: "waiting" | "completed") => |
| 2863 | projectionStore.apply({ |
| 2864 | id: EventId.make(`event:${id}`), |
| 2865 | type: "turn-item.updated", |
| 2866 | threadId, |
| 2867 | runId, |
| 2868 | nodeId, |
| 2869 | driver, |
| 2870 | occurredAt: now, |
| 2871 | payload: { |
| 2872 | id: TurnItemId.make(id), |
| 2873 | threadId, |
| 2874 | runId, |
| 2875 | nodeId, |
| 2876 | providerThreadId: null, |
| 2877 | providerTurnId: null, |
| 2878 | nativeItemRef: null, |
| 2879 | parentItemId: null, |
| 2880 | ordinal, |
| 2881 | status, |
| 2882 | title: null, |
| 2883 | startedAt: now, |
| 2884 | completedAt: null, |
| 2885 | updatedAt: now, |
| 2886 | type: "secret_request", |
| 2887 | label: "Test credential", |
| 2888 | reason: "Test pending input", |
| 2889 | secretStatus: status === "waiting" ? "pending" : "saved", |
| 2890 | }, |
| 2891 | }); |
| 2892 | yield* addSecret("secret:a", 2, "waiting"); |
| 2893 | yield* addSecret("secret:b", 3, "waiting"); |
| 2894 | yield* addSecret("secret:c", 4, "completed"); |
| 2895 | |
| 2896 | for (const status of [ |
| 2897 | "preparing", |
| 2898 | "starting", |
| 2899 | "running", |
| 2900 | "waiting", |
| 2901 | "completed", |
| 2902 | "rolled_back", |
| 2903 | ]) { |
| 2904 | yield* sql` |
| 2905 | UPDATE orchestration_v2_projection_runs |
| 2906 | SET status = ${status}, payload_json = json_set(payload_json, '$.status', ${status}) |
| 2907 | WHERE run_id = ${runId} |
| 2908 | `; |
| 2909 | const shell = yield* projectionStore.getShellSnapshot(); |
| 2910 | const thread = shell.threads.find((candidate) => candidate.id === threadId); |
| 2911 | assert.isDefined(thread); |
| 2912 | assert.equal( |
| 2913 | thread?.pendingRuntimeRequest?.id ?? null, |
| 2914 | status === "completed" || status === "rolled_back" ? null : "secret:b", |
| 2915 | ); |
| 2916 | } |
| 2917 | }), |
| 2918 | ); |
| 2919 | |
| 2853 | 2920 | it.effect("builds shell snapshots without decoding full turn item payloads", () => |
| 2854 | 2921 | Effect.gen(function* () { |
| 2855 | 2922 | const projectionStore = yield* ProjectionStore.ProjectionStoreV2; |
Yash Singh <saiansh2525@gmail.com> 2026-10-06 03:58:31-05:00
GitHub <noreply@github.com> 2026-10-06 03:58:31-05:00