Allow users to pass files on the CLI

cli.mjs

@@ -2,8 +2,14 @@
 import run from './runner.mjs'
 import concise from './report-concise.mjs'
 import verbose from './report-verbose.mjs'
+import crawl from './crawler.mjs'
 
-run('./test/', process.env.CI ? verbose : concise)
+const eventualFiles = process.argv.length > 2
+  ? Promise.resolve(process.argv.slice(2))
+  : crawl('./test/')
+
+eventualFiles
+  .then(files => run(files, process.env.CI ? verbose : concise))
   .then(({ passed, failed }) => failed ? process.exit(1) : process.exit(0))
   .catch(err => {
     process.stderr.write(err.toString())

crawler.mjs

@@ -0,0 +1,10 @@
+import fs from 'fs'
+import util from 'util'
+import path from 'path'
+
+const readdir = util.promisify(fs.readdir)
+
+export default function crawl (dir) {
+  return readdir(dir)
+    .then(files => files.map(file => path.resolve(dir, file)))
+}

example.mjs

@@ -1,5 +1,14 @@
 import run from './runner.mjs'
+import crawl from './crawler.mjs'
 import concise from './report-concise.mjs'
 import verbose from './report-verbose.mjs'
 
-run('./test/fixtures/', concise).then(() => run('./test/fixtures/', verbose))
+const fixtures = crawl('./test/fixtures/')
+
+fixtures
+  .then(fixtures => {
+    return run(fixtures, concise)
+  })
+  .then(fixtures => {
+    return run(fixtures, verbose)
+  })

package.json

@@ -9,6 +9,7 @@
   "files": [
     "batch-promise.mjs",
     "cli.mjs",
+    "crawler.mjs",
     "oletus.mjs",
     "report-concise.mjs",
     "report-verbose.mjs",

runner.mjs

@@ -1,36 +1,27 @@
-import fs from 'fs'
 import childProcess from 'child_process'
 import perfHooks from 'perf_hooks'
 import os from 'os'
 import batchPromise from './batch-promise.mjs'
 
-export default function run (testDirectory, report = () => {}) {
+export default function run (files, report = () => {}) {
   const timestamp = perfHooks.performance.now()
+  let passed = 0
+  let failed = 0
 
-  return new Promise((resolve, reject) => {
-    fs.readdir(testDirectory, (err, files) => {
-      if (err) reject(err)
+  return batchPromise(files, os.cpus().length, file => {
+    return new Promise(resolve => {
+      const forked = childProcess.fork(file)
 
-      let passed = 0
-      let failed = 0
-
-      batchPromise(files, os.cpus().length, file => {
-        return new Promise(resolve => {
-          const forked = childProcess.fork(`${testDirectory}${file}`)
-
-          forked.on('message', ({ didPass, title, location, message }) => {
-            didPass ? passed++ : failed++
-            report({ didPass, title, location, message, timestamp, file, passed, failed, isComplete: false })
-          })
-
-          forked.on('close', () => resolve())
-        })
+      forked.on('message', ({ didPass, title, location, message }) => {
+        didPass ? passed++ : failed++
+        report({ didPass, title, location, message, timestamp, file, passed, failed, isComplete: false })
       })
-        .then(() => {
-          report({ timestamp, passed, failed, isComplete: true })
-          resolve({ passed, failed })
-        })
-        .catch(err => reject(err))
+
+      forked.on('close', () => resolve())
     })
   })
+    .then(() => {
+      report({ timestamp, passed, failed, isComplete: true })
+      return { passed, failed }
+    })
 }

test/run.mjs

@@ -1,12 +1,14 @@
 import run from '../runner.mjs'
+import crawl from '../crawler.mjs'
 import assert from 'assert'
 
 function fail (e) {
   console.error(e.toString())
   process.exit(1)
 }
 
-run('./test/fixtures/')
+crawl('./test/fixtures/')
+  .then(run)
   .then(({ passed, failed }) => {
     assert.strict.equal(passed, 6)
     assert.strict.equal(failed, 4)