Test for stop all

Author: znorman-harris

apps/client-e2e/src/tests/notebooks/_notebook-runner.ts

@@ -10,6 +10,7 @@ import { NotebookFormats_1_0_0 } from './notebook-formats-1.0.0';
 import { NotebookFormats_2_0_0 } from './notebook-formats-2.0.0';
 import { RunNotebookReset } from './notebook-reset';
 import { RunNotebookStop } from './notebook-stop';
+import { RunNotebookStopAll } from './notebook-stop-all';
 import { NotebookToProCodeAllCells } from './notebook-to-pro-code-all-cells';
 import { NotebookToProCodeAllCells2 } from './notebook-to-pro-code-all-cells-2';
 import { NotebookToProCodeOnlyCode } from './notebook-to-pro-code-only-code';
@@ -223,6 +224,13 @@ NOTEBOOK_RUNNER.addTest({
   critical: true,
 });
 
+// stop all notebooks
+NOTEBOOK_RUNNER.addTest({
+  name: 'Stop all does the right thing',
+  fn: RunNotebookStopAll,
+  critical: true,
+});
+
 // stop at the end to make sure the process exits
 NOTEBOOK_RUNNER.addTest({
   name: 'Stop does the right thing',

apps/client-e2e/src/tests/notebooks/notebook-stop-all.ts

@@ -0,0 +1,38 @@
+import { GetExtensionPath, IDL_COMMANDS, Sleep } from '@idl/shared';
+import { OpenNotebookInVSCode, VSCODE_COMMANDS } from '@idl/vscode/shared';
+import expect from 'expect';
+import * as vscode from 'vscode';
+
+import { RunnerFunction } from '../runner.interface';
+
+/**
+ * Stop all notebooks
+ */
+export const RunNotebookStopAll: RunnerFunction = async (init) => {
+  /**
+   * Get the file we are going to open
+   */
+  const file = GetExtensionPath(
+    'idl/test/client-e2e/notebooks/stop-notebook.idlnb'
+  );
+
+  /**
+   * Open the notebook
+   */
+  const nb = await OpenNotebookInVSCode(file);
+
+  // trigger a run to start IDL
+  await vscode.commands.executeCommand(VSCODE_COMMANDS.NOTEBOOK_RUN_ALL);
+
+  // make sure launched
+  expect(init.notebooks.controller.isStarted(nb)).toBeTruthy();
+
+  // stop all notebooks
+  await vscode.commands.executeCommand(IDL_COMMANDS.NOTEBOOKS.STOP_ALL_KERNELS);
+
+  // short pause
+  await Sleep(500);
+
+  // make sure stopped
+  expect(init.notebooks.controller.isStarted(nb)).toBeFalsy();
+};