chore(ci): run tests against enterprise server versions

package.json

@@ -122,7 +122,7 @@
     "lerna": "^3.10.7",
     "mocha": "^7.1.2",
     "mongodb": "github:mongodb/node-mongodb-native#62717a1d",
-    "mongodb-download-url": "^0.6.1",
+    "mongodb-download-url": "^0.6.3",
     "mongodb-js-precommit": "^2.0.0",
     "node-codesign": "^0.3.2",
     "node-fetch": "^2.6.1",

packages/node-runtime-worker-thread/package.json

@@ -18,9 +18,9 @@
   },
   "scripts": {
     "pretest": "npm run webpack-build-dev -- --no-stats --no-devtool",
-    "test": "cross-env TS_NODE_PROJECT=./tsconfig.test.json mocha --timeout 15000 --colors -r ts-node/register \"./src/**/*.spec.ts\"",
+    "test": "cross-env TS_NODE_PROJECT=./tsconfig.test.json mocha -r \"../../scripts/import-expansions.js\" --timeout 15000 --colors -r ts-node/register \"./src/**/*.spec.ts\"",
     "pretest-ci": "npm run webpack-build -- --no-stats --no-devtool",
-    "test-ci": "cross-env TS_NODE_PROJECT=./tsconfig.test.json mocha --timeout 15000 -r ts-node/register \"./src/**/*.spec.ts\"",
+    "test-ci": "cross-env TS_NODE_PROJECT=./tsconfig.test.json mocha -r \"../../scripts/import-expansions.js\" --timeout 15000 -r ts-node/register \"./src/**/*.spec.ts\"",
     "lint": "eslint \"./src/**/*.{js,ts,tsx}\"",
     "check": "npm run lint",
     "webpack-build": "webpack --mode production",

testing/download-mongodb.ts

@@ -57,27 +57,38 @@ type FullJSON = {
 // On Linux, getting the exact download variant to pick is hard, so we still
 // use mongodb-download-url for that. For macOS, either would probably be fine.
 // TODO: upstream all of this into mongodb-download-url :(
-async function lookupDownloadUrl(versionInfo: VersionInfo): Promise<string> {
-  const knownDistroRegex = /^(?<name>rhel80|rhel7[0-9]|debian10)/;
+async function lookupDownloadUrl(versionInfo: VersionInfo, enterprise: boolean): Promise<string> {
+  const knownDistroRegex = /^(?<name>rhel80|rhel7[0-9]|debian10|ubuntu(?:\d+))/;
   const { version } = versionInfo;
   const distroId = process.env.DISTRO_ID || '';
   if ((process.platform === 'win32' && semver.lt(version, '4.4.0')) ||
       (process.platform === 'linux' && semver.lt(version, '4.2.0')) ||
       (process.platform !== 'win32' && !knownDistroRegex.test(distroId))) {
-    return (await promisify(getDownloadURL)({ version })).url;
+    let { url } = (await promisify(getDownloadURL)({ version, enterprise }));
+    if (semver.gte(version, '4.2.0')) {
+      url = url.replace('mongodb-osx', 'mongodb-macos');
+    }
+    if (semver.lt(version, '4.2.0') && distroId) {
+      url = url.replace('enterprise-linux_64', `enterprise-${distroId.split('-')[0]}`);
+    }
+    return url;
   }
 
   let downloadInfo: DownloadInfo;
   if (process.platform === 'win32') {
     downloadInfo = versionInfo.downloads
       .find((downloadInfo: DownloadInfo) =>
-        downloadInfo.target === 'windows' && downloadInfo.edition === 'base') as DownloadInfo;
+        downloadInfo.target === 'windows' &&
+        downloadInfo.edition === (enterprise ? 'enterprise' : 'base') &&
+        downloadInfo.arch === 'x86_64') as DownloadInfo;
   } else {
     let distro = distroId.match(knownDistroRegex)!.groups!.name;
     if (distro.match(/rhel7[0-9]/)) distro = 'rhel70';
     downloadInfo = versionInfo.downloads
       .find((downloadInfo: DownloadInfo) =>
-        downloadInfo.target === distro && downloadInfo.edition === 'targeted') as DownloadInfo;
+        downloadInfo.target === distro &&
+        downloadInfo.edition === (enterprise ? 'enterprise' : 'targeted') &&
+        downloadInfo.arch === 'x86_64') as DownloadInfo;
   }
   return downloadInfo.archive.url;
 }
@@ -116,7 +127,7 @@ export async function downloadMongoDb(targetVersionSemverSpecifier = '*'): Promi
     .filter((info: VersionInfo) => semver.satisfies(info.version, targetVersionSemverSpecifier))
     .sort((a: VersionInfo, b: VersionInfo) => semver.rcompare(a.version, b.version));
   const versionInfo: VersionInfo = productionVersions[0];
-  return await doDownload(versionInfo.version, () => lookupDownloadUrl(versionInfo));
+  return await doDownload(versionInfo.version, () => lookupDownloadUrl(versionInfo, true));
 }
 
 const downloadPromises: Record<string, Promise<string>> = {};