Add tests

Author: DonJayamanne

src/test/datascience/interactiveWindow.vscode.common.test.ts

@@ -104,6 +104,15 @@ suite(`Interactive window execution`, async function () {
         assert.equal(actualSource, source, `Executed cell has unexpected source code`);
         await waitForExecutionCompletedSuccessfully(secondCell!);
         await waitForTextOutput(secondCell!, '42');
+
+        // If we're connected to a remote kernel,
+        // then ensure the kernel points to a live connection.
+        if (IS_REMOTE_NATIVE_TEST()) {
+            assert.strictEqual(
+                controllerSelection.getSelected(notebookDocument!)?.connection.kind,
+                'connectToLiveRemoteKernel'
+            );
+        }
     });
     test('__file__ exists even after restarting a kernel', async function () {
         // Ensure we click `Yes` when prompted to restart the kernel.

src/test/datascience/notebook/remoteNotebookEditor.vscode.common.test.ts

@@ -35,7 +35,8 @@ import { setIntellisenseTimeout } from '../../../standalone/intellisense/pythonK
 import {
     IControllerDefaultService,
     IControllerLoader,
-    IControllerRegistration
+    IControllerRegistration,
+    IControllerSelection
 } from '../../../notebooks/controllers/types';
 
 /* eslint-disable @typescript-eslint/no-explicit-any, no-invalid-this */
@@ -157,6 +158,13 @@ suite('DataScience - VSCode Notebook - Remote Execution', function () {
         await insertCodeCell('print("123412341234")', { index: 0 });
         const cell = vscodeNotebook.activeNotebookEditor?.notebook.cellAt(0)!;
         await Promise.all([runCell(cell), waitForTextOutput(cell, '123412341234')]);
+
+        // Ensure the kernel points to a live connection.
+        const controllerSelection = api.serviceManager.get<IControllerSelection>(IControllerSelection);
+        assert.strictEqual(
+            controllerSelection.getSelected(vscodeNotebook.activeNotebookEditor!.notebook!)?.connection.kind,
+            'connectToLiveRemoteKernel'
+        );
     });
 
     test('Remote kernels support completions', async function () {