test WIP

Author: pwang347

src/kernels/kernelProvider.web.ts

@@ -80,6 +80,7 @@ export class KernelProvider extends BaseCoreKernelProvider {
             this.workspaceStorage
         ) as IKernel;
         kernel.onRestarted(() => this._onDidRestartKernel.fire(kernel), this, this.disposables);
+        kernel.onPostInitialized(() => this._onDidPostInitializeKernel.fire(kernel), this, this.disposables);
         kernel.onDisposed(() => this._onDidDisposeKernel.fire(kernel), this, this.disposables);
         kernel.onStarted(() => this._onDidStartKernel.fire(kernel), this, this.disposables);
         kernel.onStatusChanged(
@@ -132,6 +133,7 @@ export class ThirdPartyKernelProvider extends BaseThirdPartyKernelProvider {
             this.workspaceStorage
         );
         kernel.onRestarted(() => this._onDidRestartKernel.fire(kernel), this, this.disposables);
+        kernel.onPostInitialized(() => this._onDidPostInitializeKernel.fire(kernel), this, this.disposables);
         kernel.onDisposed(() => this._onDidDisposeKernel.fire(kernel), this, this.disposables);
         kernel.onStarted(() => this._onDidStartKernel.fire(kernel), this, this.disposables);
         kernel.onStatusChanged(

src/kernels/kernelProvider.web.unit.test.ts

@@ -17,7 +17,7 @@ import { ThirdPartyKernelProvider } from './kernelProvider.node';
 import { dispose } from '../platform/common/utils/lifecycle';
 import { noop } from '../test/core';
 
-suite('Jupyter Session', () => {
+suite.only('Jupyter Session', () => {
     suite('Web Kernel Provider', function () {
         let disposables: IDisposable[] = [];
         const asyncDisposables: { dispose: () => Promise<unknown> }[] = [];

src/standalone/api/kernels/api.vscode.common.test.ts

@@ -41,6 +41,7 @@ import { JVSC_EXTENSION_ID } from '../../../platform/common/constants';
 import { escapeStringToEmbedInPythonCode } from '../../../kernels/chat/generator';
 import { notebookCellExecutions } from '../../../platform/notebooks/cellExecutionStateService';
 import { createKernelApiForExtension } from './kernel';
+import { noop } from '../../../test/core';
 
 suiteMandatory('Kernel API Tests @typescript', function () {
     const disposables: IDisposable[] = [];
@@ -203,7 +204,14 @@ suiteMandatory('Kernel API Tests @typescript', function () {
             })
         );
 
-        await realKernel.start();
+        await realKernel.start({
+            disableUI: true,
+            onDidChangeDisableUI: () => ({
+                dispose: noop
+            })
+        });
+        assert.equal(startEventCounter, 0);
+
         const kernel = createKernelApiForExtension(JVSC_EXTENSION_ID_FOR_TESTS, realKernel);
 
         logger.info(`Execute code silently`);
@@ -226,12 +234,16 @@ suiteMandatory('Kernel API Tests @typescript', function () {
         disposables.push(
             kernels.onDidStart(({ kernel }) => {
                 startEventCounter++;
-                kernel.executeCode(`foo = ${startEventCounter}`, source.token);
+                const codeToRun =
+                    startEventCounter === 0 ? `let foo = ${startEventCounter}` : `foo = ${startEventCounter}`;
+                kernel.executeCode(codeToRun, source.token);
             })
         );
         await insertCodeCell('console.log(foo)', { index: 0, language: 'typescript' });
 
         await realKernel.start();
+        assert.equal(startEventCounter, 1);
+
         const cell = notebook.cellAt(0)!;
         const executionOrderSet = createDeferred();
         const eventHandler = notebookCellExecutions.onDidChangeNotebookCellExecutionState((e) => {