WIP

Author: lerouxb

packages/compass-e2e-tests/helpers/selectors.ts

@@ -1430,3 +1430,11 @@ export const GlobalWrites = {
   SampleFindingDocuments: '[data-testid="sample-finding-documents"]',
   SampleInsertingDocuments: '[data-testid="sample-inserting-documents"]',
 };
+
+export const AutoUpdateToast = '[data-testid="toast-compass-update"]';
+export const AutoUpdateRestartButton =
+  '[data-testid="auto-update-restart-button"]';
+export const AutoUpdateDownloadLink =
+  '[data-testid="auto-update-download-link"]';
+export const AutoUpdateReleaseNotesLink =
+  '[data-testid="auto-update-release-notes-link"]';

packages/compass-e2e-tests/installers/types.ts

@@ -5,6 +5,7 @@ export type Package = {
   packageFilepath: string;
   // TODO: once we can download the most recent release
   //releaseFilepath: string;
+  updatable: boolean;
   installer: Installer;
 };
 

packages/compass-e2e-tests/smoke-test.ts

@@ -8,16 +8,16 @@ import { hideBin } from 'yargs/helpers';
 import https from 'https';
 import { pick } from 'lodash';
 import { handler as writeBuildInfo } from 'hadron-build/commands/info';
-import type { InstalledAppInfo, Package, Installer } from './installers/types';
+import type { Package, Installer } from './installers/types';
 import { installMacDMG } from './installers/mac-dmg';
-import { execute } from './installers/helpers';
 import {
   assertBuildInfoIsOSX,
   assertBuildInfoIsRHEL,
   assertBuildInfoIsUbuntu,
   assertBuildInfoIsWindows,
   assertCommonBuildInfo,
 } from './helpers/buildinfo';
+import { testAutoUpdateFrom } from './smoketests/auto-update-from';
 
 const argv = yargs(hideBin(process.argv))
   .scriptName('smoke-tests')
@@ -202,12 +202,13 @@ async function run() {
   }
 
   if (match) {
-    const pkg = {
+    const pkg: Package = {
       // we need appName because it is the name of the executable inside the
       // package, regardless of what the package filename is named or where it
       // gets installed
       appName: buildInfo.productName,
       packageFilepath: path.join(compassDir, 'dist', match.filename),
+      updatable: match.updatable,
       // TODO: releaseFilepath once we download the most recent released version too
       installer: match.installer,
     };
@@ -234,13 +235,18 @@ async function run() {
     // TODO: installing either the packaged file or the released file is better
     // done as part of tests so we can also clean up and install one after the
     // other, but that's for a separate PR.
+    /*
     console.log('installing', pkg.packageFilepath);
     const installedInfo = await pkg.installer({
       appName: pkg.appName,
       filepath: pkg.packageFilepath,
     });
     console.log('testing', installedInfo.appPath);
     await testInstalledApp(pkg, installedInfo);
+    */
+    await testAutoUpdateFrom(pkg);
+    // TODO:
+    //await testAutoUpdateTo(pkg);
   } else {
     throw new Error(`${context.package} not implemented`);
   }
@@ -278,6 +284,7 @@ async function downloadFile(url: string, targetFile: string): Promise<void> {
   });
 }
 
+/*
 function testInstalledApp(
   pkg: Package,
   appInfo: InstalledAppInfo
@@ -302,6 +309,7 @@ function testInstalledApp(
     }
   );
 }
+*/
 
 run()
   .then(function () {

packages/compass-e2e-tests/smoketests/auto-update-from.ts

@@ -0,0 +1,67 @@
+import { expect } from 'chai';
+import {
+  init,
+  cleanup,
+  //screenshotIfFailed,
+  Selectors,
+} from '../helpers/compass';
+
+import type { Package } from '../installers/types';
+
+export async function testAutoUpdateFrom(pkg: Package) {
+  // install the app
+  console.log(`installing ${pkg.packageFilepath}`);
+  const { appPath, uninstall } = await pkg.installer({
+    appName: pkg.appName,
+    filepath: pkg.packageFilepath,
+  });
+
+  console.log(appPath);
+
+  process.env.COMPASS_APP_NAME = pkg.appName;
+  process.env.COMPASS_APP_PATH = appPath;
+
+  try {
+    // TODO: start the autoupdate server
+    try {
+      // run the app and wait for it to auto-update
+      const compass = await init('auto-update from', { firstRun: true });
+      const { browser } = compass;
+
+      await browser.$(Selectors.AutoUpdateToast).waitForDisplayed();
+
+      if (pkg.updatable) {
+        await browser.$(Selectors.AutoUpdateRestartButton).waitForDisplayed();
+      } else {
+        // When auto-update is not supported the toast contains a link to down
+        const linkElement = browser.$(Selectors.AutoUpdateDownloadLink);
+        await browser.$(linkElement).waitForDisplayed();
+        expect(await linkElement.getAttribute('href')).to.equal(
+          'https://www.mongodb.com/try/download/compass'
+        );
+      }
+
+      await cleanup(compass);
+
+      if (pkg.updatable) {
+        // run the app again and check that the version changed
+        const compass = await init('auto-update from restart', {
+          firstRun: false,
+        });
+        const { browser } = compass;
+        await browser.$(Selectors.AutoUpdateToast).waitForDisplayed();
+        await browser
+          .$(Selectors.AutoUpdateReleaseNotesLink)
+          .waitForDisplayed();
+      }
+    } finally {
+      // TODO: stop the autoupdate server
+    }
+  } finally {
+    delete process.env.COMPASS_APP_NAME;
+    delete process.env.COMPASS_APP_PATH;
+
+    // remove the app
+    await uninstall();
+  }
+}

packages/compass/src/app/components/update-toasts.tsx

@@ -53,6 +53,7 @@ const RestartCompassToastContent = ({
       <button
         className={cx(buttonStyles, darkmode && buttonDarkStyles)}
         onClick={onUpdateClicked}
+        data-testid="auto-update-restart-button"
       >
         Restart
       </button>
@@ -83,6 +84,7 @@ export function onAutoupdateExternally({
           Compass features.
         </Body>
         <Link
+          data-testid="auto-update-download-link"
           as="a"
           target="_blank"
           href={'https://www.mongodb.com/try/download/compass'}
@@ -137,6 +139,7 @@ export function onAutoupdateInstalled({ newVersion }: { newVersion: string }) {
     title: `Compass ${newVersion} installed successfully`,
     description: (
       <Link
+        data-testid="auto-update-release-notes-link"
         as="a"
         target="_blank"
         href={`https://github.com/mongodb-js/compass/releases/tag/v${newVersion}`}