Add a command to use master version of elm-explorations/test

lib/Install.js

@@ -1,8 +1,13 @@
 // @flow
 
+const {exec} = require('child_process');
 const spawn = require('cross-spawn');
+const {unzip} = require('zlib');
 const fs = require('fs');
+const os = require('os');
+const https = require('https');
 const path = require('path');
+const chalk = require('chalk');
 const ElmJson = require('./ElmJson');
 const Project = require('./Project');
 
@@ -117,6 +122,122 @@ function install(
   return 'SuccessfullyInstalled';
 }
 
+async function downloadFileNative(url, filePath) {
+  const usedCommand = `require("https").get(${JSON.stringify(url)})`;
+  const errorPrefix = `${usedCommand}\nThe above call errored: `;
+
+  return new Promise((resolve, reject) => {
+    const file = fs.createWriteStream(filePath);
+    let fileInfo = null;
+
+    const request = https.get(url, response => {
+      if (response.statusCode !== 200) {
+        fs.unlink(filePath, () => {
+          reject(new Error(`Failed to get '${url}' (${response.statusCode})`));
+        });
+        return;
+      }
+
+      fileInfo = {
+        mime: response.headers['content-type'],
+        size: parseInt(response.headers['content-length'], 10),
+      };
+
+      response.pipe(file);
+    });
+
+    // The destination stream is ended by the time it's called
+    file.on('finish', () => resolve(fileInfo));
+
+    request.on('error', err => {
+      fs.unlink(filePath, () => reject(err));
+    });
+
+    file.on('error', err => {
+      fs.unlink(filePath, () => reject(err));
+    });
+
+    request.end();
+  });
+}
+
+async function installUnstableTestMaster(projectRootDir) {
+  console.log(chalk.yellow('Using the master version of elm-explorations/test'));
+  console.log(chalk.yellow('Note you might need to rm -rf ~/.elm and/or ./elm-stuff afterwards.'));
+
+  const pkg = 'elm-explorations/test';
+  const version = '1.2.2';
+  const pkgWithDash = pkg.replace('/','-');
+
+  const tempPath = path.join(
+    projectRootDir,
+    'elm-stuff',
+    'generated-code',
+    'elm-community',
+    'elm-test'
+  );
+  const zipballUrl = `https://codeload.github.com/${pkg}/zip/refs/heads/master`;
+  const zipballFilename = `${pkgWithDash}.zip`;
+  const zipballPath = path.join(tempPath, zipballFilename);
+  const homeDir = os.homedir();
+
+  // based on info in https://package.elm-lang.org/packages/elm/project-metadata-utils/latest/
+  const elmHome = process.env.ELM_HOME ?? (
+    process.platform === 'win32'
+      ? path.join(process.env.APPDATA, 'elm')
+      : path.join(homeDir, '.elm')
+  );
+  const packagePath = path.join(
+    elmHome,
+    '0.19.1',
+    'packages',
+    pkg,
+    version
+  );
+
+  fs.rmSync(tempPath,    { recursive: true, force: true });
+  fs.rmSync(packagePath, { recursive: true, force: true });
+  fs.mkdirSync(tempPath,    { recursive: true });
+  fs.mkdirSync(packagePath, { recursive: true });
+  await downloadFileNative(zipballUrl, zipballPath);
+  const unzipResult = spawn.sync(
+    'unzip',
+    [
+      '-o', // overwrite
+      zipballFilename, // file to unzip
+      '-d', tempPath // directory where to extract files
+    ],
+    {
+      cwd: tempPath,
+    }
+  );
+  if (unzipResult.status !== 0) {
+    const tarResult = spawn.sync(
+      'tar',
+      [
+        'zxf', // eXtract Zipped File
+        zipballFilename, // file to unzip
+        '-C', tempPath // directory where to extract files
+      ],
+      {
+        cwd: tempPath,
+      }
+    );
+    if (tarResult.status !== 0) {
+      throw new Error("Failed to unzip the elm-explorations/test repo zipfile");
+    }
+  }
+  fs.renameSync(
+    path.join(
+      tempPath,
+      'test-master'
+    ),
+    packagePath
+  );
+  fs.rmSync(zipballPath, { recursive: true, force: true });
+}
+
 module.exports = {
   install,
+  installUnstableTestMaster,
 };

lib/RunTests.js

@@ -10,6 +10,7 @@ const DependencyProvider = require('./DependencyProvider.js');
 const ElmJson = require('./ElmJson');
 const FindTests = require('./FindTests');
 const Generate = require('./Generate');
+const Install = require('./Install');
 const Project = require('./Project');
 const Report = require('./Report');
 const Supervisor = require('./Supervisor');
@@ -117,7 +118,7 @@ function watcherEventMessage(queue /*: typeof Queue */) /*: string */ {
   return `${filePaths.length} files ${events.join('/')}`;
 }
 
-function runTests(
+async function runTests(
   dependencyProvider /*: DependencyProvider */,
   projectRootDir /*: string */,
   pathToElmBinary /*: string */,
@@ -128,11 +129,13 @@ function runTests(
     report,
     seed,
     fuzz,
+    unstableUseTestMaster,
   } /*: {
     watch: boolean,
     report: typeof Report.Report,
     seed: number,
     fuzz: number,
+    unstableUseTestMaster: boolean,
   } */
 ) /*: Promise<number> */ {
   let watcher = undefined;
@@ -249,6 +252,10 @@ function runTests(
     }
   }
 
+  if (unstableUseTestMaster) {
+    await Install.installUnstableTestMaster(projectRootDir);
+  }
+
   if (watch) {
     progressLogger.clearConsole();
 

lib/elm-test.js

@@ -142,6 +142,10 @@ function main() {
       '--color',
       'Force colored console output (setting FORCE_COLOR to anything but 0 also works)'
     )
+    .option(
+      '--unstable-use-test-master',
+      'Use the bleeding-edge version of the elm-explorations/test library'
+    )
     .option(
       '--compiler <path>',
       'Use a custom path to an Elm executable (default: elm)',
@@ -241,12 +245,12 @@ function main() {
     // commander would interpret the `tests` in `elm-test tests src` as a
     // command and only run tests in `src/`, ignoring all files in `tests/`.
     .command('__elmTestCommand__ [globs...]', { hidden: true, isDefault: true })
-    .action((testFileGlobs) => {
+    .action(async (testFileGlobs) => {
       const options = program.opts();
       const pathToElmBinary = getPathToElmBinary(options.compiler);
       const projectRootDir = getProjectRootDir('tests');
       const processes = Math.max(1, os.cpus().length);
-      RunTests.runTests(
+      await RunTests.runTests(
         dependencyProvider,
         projectRootDir,
         pathToElmBinary,