Misc

microsoft · Mar 17, 2022 · 0a06222 · 0a06222
1 parent bb64bcb
commit 0a06222
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/test/datascience/jupyter/kernels/installationPrompts.vscode.test.ts b/src/test/datascience/jupyter/kernels/installationPrompts.vscode.test.ts
@@ -440,16 +440,15 @@ suite('DataScience Install IPyKernel (slow) (install)', function () {
         assert.equal(cell.outputs.length, 0);
 
         // Insert another cell so we can test run all
-        await insertCodeCell('print("foo")');
+        const cell2 = await insertCodeCell('print("foo")');
 
         // The prompt should be displayed when we run a cell.
         const runPromise = runAllCellsInActiveNotebook();
         await waitForCondition(async () => prompt.displayed.then(() => true), delayForUITest, 'Prompt not displayed');
 
         // Now the run should finish
         await runPromise;
-        cell = vscodeNotebook.activeNotebookEditor?.document.cellAt(0)!;
-        await waitForExecutionCompletedSuccessfully(cell);
+        await Promise.all([waitForExecutionCompletedSuccessfully(cell), waitForExecutionCompletedSuccessfully(cell2)]);
     });
 
     test('Ensure ipykernel install prompt is NOT displayed when auto start is enabled & ipykernel is missing (VSCode Notebook)', async function () {