Function waitForCodeLenses

src/test/datascience/helpers.ts:272–306

(document: vscode.Uri, command: string)

}

export async function waitForCodeLenses(document: vscode.Uri, command: string) {
    // First make sure the editor has focus
    const selection = new vscode.Range(new vscode.Position(0, 0), new vscode.Position(0, 0));
    let editor = vscode.window.visibleTextEditors.find((e) => arePathsSame(e.document.uri.fsPath, document.fsPath));
    if (editor) {
        await vscode.window
            .showTextDocument(editor.document, { selection, viewColumn: editor.viewColumn })
            .then((e) => {
                e.revealRange(selection, vscode.TextEditorRevealType.InCenter);
            });
    }

    let codeLenses: vscode.CodeLens[] = [];
    // Wait for the code lens to appear
    await waitForCondition(
        async () => {
            const textDocument = await vscode.workspace.openTextDocument(document);
            await vscode.window.showTextDocument(textDocument, undefined, false);
            codeLenses = (await vscode.commands.executeCommand(
                'vscode.executeCodeLensProvider',
                document
            )) as vscode.CodeLens[];
            return (
                codeLenses &&
                codeLenses.length > 0 &&
                codeLenses.find((c) => c.command?.command === command) != undefined
            );
        },
        defaultNotebookTestTimeout,
        `Code lens with command ${command} not found`
    );

    logger.ci(`Found code lenses with command ${command}`);
    return codeLenses;
}

export async function verifySelectedControllerIsRemoteForRemoteTests(notebook?: vscode.NotebookDocument) {
    if (!IS_REMOTE_NATIVE_TEST() || !isWeb()) {