(
overrides: Record<string, unknown> = {}
)
| 183 | } |
| 184 | |
| 185 | function createStreamInfoForTests( |
| 186 | overrides: Record<string, unknown> = {} |
| 187 | ): Record<string, unknown> { |
| 188 | const now = Date.now(); |
| 189 | const model = overrides.model ?? TEST_STREAM_MODEL_ID; |
| 190 | return { |
| 191 | state: "streaming", |
| 192 | streamResult: createStreamResultForTests( |
| 193 | (async function* emptyStream() { |
| 194 | await Promise.resolve(); |
| 195 | yield* []; |
| 196 | })() |
| 197 | ), |
| 198 | abortController: new AbortController(), |
| 199 | messageId: "test-message", |
| 200 | token: "test-token", |
| 201 | startTime: now, |
| 202 | lastPartTimestamp: now, |
| 203 | toolCompletionTimestamps: new Map<string, number>(), |
| 204 | pendingWorkflowRunAttachments: new Map<string, unknown>(), |
| 205 | model, |
| 206 | metadataModel: overrides.metadataModel ?? model, |
| 207 | historySequence: 1, |
| 208 | request: { model: createTestLanguageModel(), messages: [], providerOptions: undefined }, |
| 209 | toolModelUsages: [], |
| 210 | parts: [], |
| 211 | lastPartialWriteTime: 0, |
| 212 | partialWriteTimer: undefined, |
| 213 | partialWritePromise: undefined, |
| 214 | processingPromise: Promise.resolve(), |
| 215 | softInterrupt: { pending: false as const }, |
| 216 | runtimeTempDir: "", |
| 217 | runtime: LOCAL_TEST_RUNTIME, |
| 218 | cumulativeUsage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 }, |
| 219 | cumulativeProviderMetadata: undefined, |
| 220 | didRetryPreviousResponseIdAtStep: false, |
| 221 | receivedTerminalEvent: false, |
| 222 | currentStepStartIndex: 0, |
| 223 | stepTracker: {}, |
| 224 | ...overrides, |
| 225 | }; |
| 226 | } |
| 227 | |
| 228 | describe("StreamManager - workflow run attachments", () => { |
| 229 | test("persists attached workflow run metadata to partial immediately", async () => { |
no test coverage detected