feat(chat): Prompt to connect when a disconnected user tries to run code from participant VSCODE-618 (#872)

Author: gagik

src/editors/playgroundController.ts

@@ -583,10 +583,13 @@ export default class PlaygroundController {
       .get('confirmRunCopilotCode');
 
     if (!this._connectionController.isCurrentlyConnected()) {
-      // TODO(VSCODE-618): Prompt user to connect when clicked.
-      void vscode.window.showErrorMessage(connectBeforeRunningMessage);
+      const successfullyConnected =
+        await this._connectionController.changeActiveConnection();
 
-      return false;
+      if (!successfullyConnected) {
+        void vscode.window.showErrorMessage(connectBeforeRunningMessage);
+        return false;
+      }
     }
 
     if (shouldConfirmRunCopilotCode === true) {

src/test/suite/editors/playgroundController.test.ts

@@ -270,6 +270,60 @@ suite('Playground Controller Test Suite', function () {
           expectedMessage
         );
       });
+
+      suite('running code from the participant', function () {
+        beforeEach(function () {
+          sinon
+            .stub(testPlaygroundController, '_evaluateWithCancelModal')
+            .resolves({ result: '123' } as any);
+          sinon.stub(testPlaygroundController, '_openInResultPane').resolves();
+
+          showInformationMessageStub.resolves('Yes');
+        });
+
+        afterEach(() => sinon.restore());
+
+        test('prompts to connect to a database and succeeds with selection', async () => {
+          const changeActiveConnectionStub = sinon.stub(
+            testPlaygroundController._connectionController,
+            'changeActiveConnection'
+          );
+          // Mocks the user selecting a connection.
+          changeActiveConnectionStub.resolves(true);
+
+          const result = await testPlaygroundController.evaluateParticipantCode(
+            'console.log("test");'
+          );
+
+          expect(showErrorMessageStub.notCalled).is.true;
+
+          expect(changeActiveConnectionStub.calledOnce).is.true;
+
+          expect(result).is.true;
+        });
+
+        test('prompts to connect to a database and errors if not selected', async () => {
+          const changeActiveConnectionStub = sinon.stub(
+            testPlaygroundController._connectionController,
+            'changeActiveConnection'
+          );
+          // Mocks the user selecting a connection.
+          changeActiveConnectionStub.resolves(false);
+
+          const result = await testPlaygroundController.evaluateParticipantCode(
+            'console.log("test");'
+          );
+
+          const expectedMessage =
+            'Please connect to a database before running a playground.';
+          await testPlaygroundController.runAllOrSelectedPlaygroundBlocks();
+          expect(showErrorMessageStub.firstCall.args[0]).to.be.equal(
+            expectedMessage
+          );
+
+          expect(result).is.false;
+        });
+      });
     });
 
     suite('user is connected', () => {
@@ -431,6 +485,35 @@ suite('Playground Controller Test Suite', function () {
           expect(result).to.be.false;
         });
       });
+
+      suite('running code from the participant', function () {
+        beforeEach(function () {
+          sinon
+            .stub(testPlaygroundController, '_evaluateWithCancelModal')
+            .resolves({ result: '123' } as any);
+          sinon.stub(testPlaygroundController, '_openInResultPane').resolves();
+
+          showInformationMessageStub.resolves('Yes');
+        });
+
+        afterEach(() => sinon.restore());
+
+        test('does not prompt to connect to the database', async () => {
+          const changeActiveConnectionStub = sinon.stub(
+            testPlaygroundController._connectionController,
+            'changeActiveConnection'
+          );
+          const result = await testPlaygroundController.evaluateParticipantCode(
+            'console.log("test");'
+          );
+
+          expect(showErrorMessageStub.notCalled).is.true;
+
+          expect(changeActiveConnectionStub.notCalled).is.true;
+
+          expect(result).is.true;
+        });
+      });
     });
   });
 });