download and test with the latest release

Author: lerouxb

packages/compass-smoke-tests/src/build-info.ts

@@ -8,6 +8,8 @@ import { type PackageKind } from './packages';
 import { type SmokeTestsContext } from './context';
 import { pick } from 'lodash';
 
+const SUPPORTED_CHANNELS = ['dev', 'beta', 'stable'] as const;
+
 function assertObjectHasKeys(
   obj: unknown,
   name: string,
@@ -25,13 +27,21 @@ function assertObjectHasKeys(
 
 // subsets of the hadron-build info result
 
-export const commonKeys = ['productName'] as const;
-export type CommonBuildInfo = Record<typeof commonKeys[number], string>;
+export const commonKeys = ['productName', 'channel'] as const;
+export type CommonBuildInfo = Record<typeof commonKeys[number], string> & {
+  channel: 'dev' | 'beta' | 'stable';
+};
 
 export function assertCommonBuildInfo(
   buildInfo: unknown
 ): asserts buildInfo is CommonBuildInfo {
   assertObjectHasKeys(buildInfo, 'buildInfo', commonKeys);
+  assert(
+    SUPPORTED_CHANNELS.includes((buildInfo as any).channel),
+    `Expected ${(buildInfo as any).channel} to be in ${SUPPORTED_CHANNELS.join(
+      ','
+    )}`
+  );
 }
 
 export const windowsFilenameKeys = [

packages/compass-smoke-tests/src/cli.ts

@@ -145,6 +145,60 @@ async function getTestSubject(
   }
 }
 
+type Arch = 'arm64' | 'x64';
+
+type PlatformShortName =
+  | 'darwin-arm64'
+  | 'darwin-x64'
+  | 'windows'
+  | 'linux_deb'
+  | 'linux_rpm';
+
+function getPlatformShortName(
+  arch: Arch,
+  kind: PackageKind
+): PlatformShortName {
+  if (arch === 'arm64') {
+    if (kind === 'osx_dmg' || kind === 'osx_zip') {
+      return 'darwin-arm64';
+    }
+  }
+  if (arch === 'x64') {
+    if (kind === 'osx_dmg' || kind === 'osx_zip') {
+      return 'darwin-x64';
+    }
+    if (
+      kind === 'windows_setup' ||
+      kind === 'windows_msi' ||
+      kind === 'windows_zip'
+    ) {
+      return 'windows';
+    }
+    if (kind === 'linux_deb' || kind === 'linux_tar') {
+      return 'linux_deb';
+    }
+    if (kind === 'linux_rpm' || kind === 'rhel_tar') {
+      return 'linux_rpm';
+    }
+  }
+
+  throw new Error(`Unsupported arch/kind combo: ${arch}/${kind}`);
+}
+
+async function getLatestRelease(
+  channel: 'dev' | 'beta' | 'stable',
+  arch: Arch,
+  kind: PackageKind,
+  forceDownload?: boolean
+): Promise<string> {
+  const shortName = getPlatformShortName(arch, kind);
+
+  return await downloadFile({
+    url: `http://compass.mongodb.com/api/v2/download/latest/compass/${channel}/${shortName}`,
+    clearCache: forceDownload,
+  });
+}
+
 function getInstaller(kind: PackageKind) {
   if (kind === 'osx_dmg') {
     return installMacDMG;
@@ -179,6 +233,8 @@ async function run() {
   const install = getInstaller(kind);
 
   try {
+    console.log('downgrade from this package to latest release');
+
     const appName = buildInfo.productName;
 
     const { appPath, uninstall } = install({
@@ -196,6 +252,33 @@ async function run() {
     console.log('Cleaning up sandbox');
     fs.rmSync(context.sandboxPath, { recursive: true });
   }
+
+  console.log('update from latest release to this package');
+  const releasepath = await getLatestRelease(
+    buildInfo.channel,
+    context.arch,
+    kind,
+    context.forceDownload
+  );
+
+  try {
+    const appName = buildInfo.productName;
+
+    const { appPath, uninstall } = install({
+      appName,
+      filepath: releasepath,
+      destinationPath: context.sandboxPath,
+    });
+
+    try {
+      runTest({ appName, appPath });
+    } finally {
+      await uninstall();
+    }
+  } finally {
+    console.log('Cleaning up sandbox');
+    fs.rmSync(context.sandboxPath, { recursive: true });
+  }
 }
 
 type RunTestOptions = {

packages/compass-smoke-tests/src/downloads.ts

@@ -7,7 +7,7 @@ import { ensureDownloadsDirectory } from './directories';
 
 type DownloadFileOptions = {
   url: string;
-  targetFilename: string;
+  targetFilename?: string;
   clearCache?: boolean;
 };
 
@@ -18,6 +18,14 @@ export async function downloadFile({
 }: DownloadFileOptions): Promise<string> {
   const response = await fetch(url);
 
+  if (!targetFilename) {
+    // if no filename was specified, work out the filename based on the final
+    // URL the way a browser would
+    targetFilename = path.basename(new URL(response.url).pathname);
+  }
+
+  assert(targetFilename, 'Expected a filename');
+
   const etag = response.headers.get('etag');
   assert(etag, 'Expected an ETag header');
   const cleanEtag = etag.match(/[0-9a-fA-F]/g)?.join('');
@@ -50,5 +58,7 @@ export async function downloadFile({
     fs.createWriteStream(outputPath)
   );
 
+  console.log(outputPath);
+
   return outputPath;
 }