Function runDrainPass

packages/pi-plugin/src/context-handler.test.ts:2590–2624

(args: {
			db: ReturnType<typeof createTestDb>;
			sessionId: string;
			appendCompaction?: (...args: unknown[]) => string | undefined;
		})

		}

		async function runDrainPass(args: {
			db: ReturnType<typeof createTestDb>;
			sessionId: string;
			appendCompaction?: (...args: unknown[]) => string | undefined;
		}): Promise<void> {
			const fake = createFakePi();
			registerPiContextHandler(fake.pi as never, {
				db: args.db,
				ctxReduceEnabled: true,
				injection: { injectionBudgetTokens: 10_000 },
			});
			const handler = fake.handlers.get("context") as (
				event: { messages: never[] },
				ctx: never,
			) => Promise<{ messages: never[] }>;
			const messages = [
				userMessage("first", 1),
				assistantMessage("second", 2),
				userMessage("third", 3),
			] as never[];
			const ctx = fakeContext(
				args.sessionId,
				process.cwd(),
				["entry-1", "entry-2", "entry-3"],
				messages as never,
			) as never as {
				sessionManager: {
					appendCompaction?: (...args: unknown[]) => string | undefined;
				};
			};
			if (args.appendCompaction) {
				ctx.sessionManager.appendCompaction = args.appendCompaction;
			}
			await handler({ messages }, ctx as never);
		}

		it("drains a deferred Pi marker only on a materializing pass", async () => {
			const db = createTestDb();