Function waitForLastCellToComplete

src/test/datascience/helpers.ts:246–270

(
    interactiveWindow: InteractiveWindow,
    numberOfCells: number = -1,
    errorsOkay?: boolean
)

}

export async function waitForLastCellToComplete(
    interactiveWindow: InteractiveWindow,
    numberOfCells: number = -1,
    errorsOkay?: boolean
) {
    const notebookDocument = await waitForInteractiveWindow(interactiveWindow);
    let codeCell: vscode.NotebookCell | undefined;
    let codeCells: vscode.NotebookCell[] = [];
    await waitForCondition(
        async () => {
            codeCells = notebookDocument?.getCells().filter((c) => c.kind === vscode.NotebookCellKind.Code);
            codeCell = codeCells && codeCells.length ? codeCells[codeCells.length - 1] : undefined;
            return codeCell && (numberOfCells === -1 || numberOfCells === codeCells!.length) ? true : false;
        },
        defaultNotebookTestTimeout,
        `No code cell found in interactive window notebook document`
    );
    if (errorsOkay) {
        await waitForCellExecutionToComplete(codeCell!);
    } else {
        await waitForExecutionCompletedSuccessfully(codeCell!);
    }
    logger.ci(`finished waiting for last cell to complete of ${codeCells.length} cells`);
    return codeCell!;
}

export async function waitForCodeLenses(document: vscode.Uri, command: string) {
    // First make sure the editor has focus