MCPcopy
hub / github.com/coder/mux / createStreamInfoForTests

Function createStreamInfoForTests

src/node/services/streamManager.test.ts:185–226  ·  view source on GitHub ↗
(
  overrides: Record<string, unknown> = {}
)

Source from the content-addressed store, hash-verified

183}
184
185function 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
228describe("StreamManager - workflow run attachments", () => {
229 test("persists attached workflow run metadata to partial immediately", async () => {

Callers 3

processCleanupStreamFunction · 0.85

Calls 3

createTestLanguageModelFunction · 0.85
resolveMethod · 0.80

Tested by

no test coverage detected