Function startBedrockMockServer

apps/vscode-e2e/src/bedrock-mock-server.ts:91–152

()

}

export async function startBedrockMockServer(): Promise<BedrockMockServer> {
	// HTTP/2 cleartext (h2c) — matches what @aws-sdk/client-bedrock-runtime uses by default.
	const server = http2.createServer()
	let lastRequestHeaders: http2.IncomingHttpHeaders | undefined
	const sessions = new Set<http2.ServerHttp2Session>()

	server.on("session", (session) => {
		sessions.add(session)
		session.on("close", () => sessions.delete(session))
	})

	server.on("stream", (stream, headers) => {
		const path = headers[":path"] as string
		const method = headers[":method"] as string

		if (!path?.includes("converse-stream") || method !== "POST") {
			stream.respond({ ":status": 404 })
			stream.end(JSON.stringify({ error: { message: "Not found", type: "not_found" } }))
			return
		}

		lastRequestHeaders = headers

		// Drain the request body before responding (AWS SDK sends the full request before reading).
		stream.resume()
		stream.on("end", () => {
			stream.respond({
				":status": 200,
				"content-type": "application/vnd.amazon.eventstream",
			})
			const frames = buildToolCallFrames(
				"attempt_completion",
				"tooluse_bedrock_mock_001",
				JSON.stringify({ result: "4" }),
			)
			for (const frame of frames) {
				stream.write(frame)
			}
			stream.end()
		})
	})

	await new Promise<void>((resolve) => server.listen(0, "127.0.0.1", resolve))
	const addr = server.address() as net.AddressInfo
	const url = `http://127.0.0.1:${addr.port}`

	return {
		url,
		get lastRequestHeaders() {
			return lastRequestHeaders
		},
		close: () => {
			// Destroy all open HTTP/2 sessions first so server.close() resolves immediately
			// instead of waiting for the extension's retry loop to exhaust itself.
			for (const session of sessions) {
				session.destroy()
			}
			sessions.clear()