add tests

src/connectionController.ts

@@ -76,7 +76,8 @@ function isOIDCAuth(connectionString: string): boolean {
   return authMechanismString === 'MONGODB-OIDC';
 }
 
-function getNotifyDeviceFlowForConnectionAttempt(
+// Exported for testing.
+export function getNotifyDeviceFlowForConnectionAttempt(
   connectionOptions: ConnectionOptions
 ) {
   const isOIDCConnectionAttempt = isOIDCAuth(

src/test/suite/connectionController.test.ts

@@ -5,9 +5,11 @@ import * as vscode from 'vscode';
 import { afterEach, beforeEach } from 'mocha';
 import assert from 'assert';
 import * as mongodbDataService from 'mongodb-data-service';
+import ConnectionString from 'mongodb-connection-string-url';
 
 import ConnectionController, {
   DataServiceEventTypes,
+  getNotifyDeviceFlowForConnectionAttempt,
 } from '../../connectionController';
 import formatError from '../../utils/formatError';
 import { StorageController, StorageVariables } from '../../storage';
@@ -53,10 +55,14 @@ suite('Connection Controller Test Suite', function () {
     telemetryService: testTelemetryService,
   });
   let showErrorMessageStub: SinonStub;
+  let showInformationMessageStub: SinonStub;
   const sandbox = sinon.createSandbox();
 
   beforeEach(() => {
-    sandbox.stub(vscode.window, 'showInformationMessage');
+    showInformationMessageStub = sandbox.stub(
+      vscode.window,
+      'showInformationMessage'
+    );
     sandbox.stub(testTelemetryService, 'trackNewConnection');
     showErrorMessageStub = sandbox.stub(vscode.window, 'showErrorMessage');
   });
@@ -462,6 +468,45 @@ suite('Connection Controller Test Suite', function () {
     assert.strictEqual(JSON.stringify(workspaceStoreConnections), objectString);
   });
 
+  test('getNotifyDeviceFlowForConnectionAttempt returns a function that shows a message with the url when oidc is set', function () {
+    const expectedUndefinedDeviceFlow = getNotifyDeviceFlowForConnectionAttempt(
+      {
+        connectionString: TEST_DATABASE_URI,
+      }
+    );
+
+    assert.strictEqual(expectedUndefinedDeviceFlow, undefined);
+
+    const oidcConnectionString = new ConnectionString(TEST_DATABASE_URI);
+    oidcConnectionString.searchParams.set('authMechanism', 'MONGODB-OIDC');
+
+    const expectedFunction = getNotifyDeviceFlowForConnectionAttempt({
+      connectionString: oidcConnectionString.toString(),
+    });
+    assert.notStrictEqual(expectedFunction, undefined);
+    assert.strictEqual(showInformationMessageStub.called, false);
+
+    (
+      expectedFunction as (deviceFlowInformation: {
+        verificationUrl: string;
+        userCode: string;
+      }) => void
+    )({
+      verificationUrl: 'test123',
+      userCode: 'testabc',
+    });
+
+    assert.strictEqual(showInformationMessageStub.called, true);
+    assert.strictEqual(
+      showInformationMessageStub.firstCall.args[0].includes('test123'),
+      true
+    );
+    assert.strictEqual(
+      showInformationMessageStub.firstCall.args[0].includes('testabc'),
+      true
+    );
+  });
+
   test('when a connection is removed it is also removed from workspace store', async () => {
     await testConnectionController.loadSavedConnections();
     await vscode.workspace

src/test/suite/views/webviewController.test.ts

@@ -1,7 +1,7 @@
 import sinon from 'sinon';
 import * as vscode from 'vscode';
 import assert from 'assert';
-import { beforeEach, afterEach } from 'mocha';
+import { before, after, beforeEach, afterEach } from 'mocha';
 import fs from 'fs';
 import path from 'path';
 
@@ -127,6 +127,66 @@ suite('Webview Test Suite', () => {
     );
   });
 
+  test('web view content sets the oidc device auth id globally', () => {
+    const fakeWebview: any = {
+      asWebviewUri: (jsUri) => {
+        return jsUri;
+      },
+    };
+
+    const extensionPath = mdbTestExtension.extensionContextStub.extensionPath;
+    const htmlString = getWebviewContent({
+      extensionPath,
+      telemetryUserId: 'test',
+      webview: fakeWebview,
+    });
+
+    assert(
+      htmlString.includes(
+        ">window['VSCODE_EXTENSION_OIDC_DEVICE_AUTH_ID'] = false;"
+      )
+    );
+  });
+
+  suite('when oidc device auth flow setting is enabled', function () {
+    let originalDeviceAuthFlow;
+    before(async function () {
+      originalDeviceAuthFlow = vscode.workspace.getConfiguration(
+        'mdb.showOIDCDeviceAuthFlow'
+      );
+
+      await vscode.workspace
+        .getConfiguration('mdb')
+        .update('showOIDCDeviceAuthFlow', true);
+    });
+    after(async function () {
+      await vscode.workspace
+        .getConfiguration('mdb')
+        .update('showOIDCDeviceAuthFlow', originalDeviceAuthFlow);
+    });
+
+    test('web view content sets the oidc device auth id globally', () => {
+      const fakeWebview: any = {
+        asWebviewUri: (jsUri) => {
+          return jsUri;
+        },
+      };
+
+      const extensionPath = mdbTestExtension.extensionContextStub.extensionPath;
+      const htmlString = getWebviewContent({
+        extensionPath,
+        telemetryUserId: 'test',
+        webview: fakeWebview,
+      });
+
+      assert(
+        htmlString.includes(
+          ">window['VSCODE_EXTENSION_OIDC_DEVICE_AUTH_ID'] = true;"
+        )
+      );
+    });
+  });
+
   test('web view listens for a connect message and adds the connection', (done) => {
     const extensionContextStub = new ExtensionContextStub();
     const testStorageController = new StorageController(extensionContextStub);