Function fakeContext

packages/pi-plugin/src/test-utils.test.ts:127–161

(
	sessionId = "ses-test",
	cwd = process.cwd(),
	entryIds: string[] = ["entry-1"],
	/**
	 * Optional message references used as `entry.message` on each fake
	 * SessionEntry. Pi's runtime keeps `entry.message === sourceAgentMessage`
	 * by reference (session-manager.js:580), and Magic Context's
	 * `collectMessageEntryIdsByRef` relies on that reference identity to
	 * map event.messages → entry ids. Tests that want boundary-id
	 * resolution to work should pass the same message objects they put
	 * into `event.messages`. Without explicit messages, we fall back to
	 * synthesized userMessage instances — which intentionally do NOT
	 * match anything the test passes via `event.messages`, exercising
	 * the unmapped-slot fallback path.
	 */
	messages?: PiMessage[],
)

}

export function fakeContext(
	sessionId = "ses-test",
	cwd = process.cwd(),
	entryIds: string[] = ["entry-1"],
	/**
	 * Optional message references used as `entry.message` on each fake
	 * SessionEntry. Pi's runtime keeps `entry.message === sourceAgentMessage`
	 * by reference (session-manager.js:580), and Magic Context's
	 * `collectMessageEntryIdsByRef` relies on that reference identity to
	 * map event.messages → entry ids. Tests that want boundary-id
	 * resolution to work should pass the same message objects they put
	 * into `event.messages`. Without explicit messages, we fall back to
	 * synthesized userMessage instances — which intentionally do NOT
	 * match anything the test passes via `event.messages`, exercising
	 * the unmapped-slot fallback path.
	 */
	messages?: PiMessage[],
) {
	return {
		cwd,
		hasUI: true,
		signal: new AbortController().signal,
		ui: { notify: () => undefined },
		sessionManager: {
			getSessionId: () => sessionId,
			getBranch: () =>
				entryIds.map((id, index) => ({
					type: "message",
					id,
					message: messages?.[index] ?? userMessage("", index + 1),
				})),
		},
		getContextUsage: () => ({ tokens: 0, percent: 0, contextWindow: 100_000 }),
	};
}