Class TestSessionCoordinator

src/shared/test/coordinator.ts:14–336

/** Handles results from a test debug session and provides them to the test model. */
export class TestSessionCoordinator implements IAmDisposable {
	private disposables: IAmDisposable[] = [];

	/**
	 * A link between a suite path and the debug session ID that owns it, so we can ensure
	 * it is correctly ended when the debug session ends, even if we don't get the correct
	 * end events.
	 */
	private owningDebugSessions: Record<string, string | undefined> = {};

	/** For a given debug session, lookups by IDs to get back to the suite. */
	private debugSessionLookups: Record<string, {
		cwd: string | undefined,
		suiteForID: Record<string, SuiteData | undefined>,
		suiteForTestID: Record<string, SuiteData | undefined>,
	} | undefined> = {};

	/**
	 * A link between a suite path and a visitor for visiting its latest outline data.
	 * This data is refreshed when a test suite starts running.
	 */
	private suiteOutlineVisitors: Record<string, TestOutlineVisitor | undefined> = {};

	/**
	 * For each debug session ID, stores a mapping of phantom (empty) groups and their parent IDs so we can
	 * jump over them.
	 */
	private phantomGroupParents: Record<string, Record<number, number | null | undefined>> = {};

	constructor(private readonly logger: Logger, private readonly data: TestModel, private readonly fileTracker: { getOutlineFor(uri: URI): Outline | undefined }) { }

	public handleDebugSessionStart(debugSessionID: string, dartCodeDebugSessionID: string | undefined, cwd: string | undefined) {
		dartCodeDebugSessionID ??= `untagged-session-${debugSessionID}`;

		let lookup = this.debugSessionLookups[dartCodeDebugSessionID];
		if (!lookup)
			lookup = this.debugSessionLookups[dartCodeDebugSessionID] = { cwd, suiteForID: {}, suiteForTestID: {} };

		if (cwd)
			lookup.cwd = cwd;
	}

	public handleDebugSessionCustomEvent(debugSessionID: string, dartCodeDebugSessionID: string | undefined, event: string, body?: any) {
		dartCodeDebugSessionID ??= `untagged-session-${debugSessionID}`;
		if (event === "dart.testNotification") {
			void this.handleNotification(debugSessionID, dartCodeDebugSessionID, body as Notification).catch((e) => this.logger.error(e));
		}
	}

	public handleDebugSessionEnd(debugSessionID: string, dartCodeDebugSessionID: string | undefined) {
		// Get the suite paths that have us as the owning debug session.
		const suitePaths = Object.keys(this.owningDebugSessions).filter((suitePath) => {
			const owningSessionID = this.owningDebugSessions[suitePath];
			return owningSessionID === debugSessionID;
		});

		// End them all and remove from the lookup.
		for (const suitePath of suitePaths) {