1 files changed, 161 insertions, 0 deletions
diff --git a/node_modules/ava/profile.js b/node_modules/ava/profile.js
new file mode 100644
index 000000000..530543fcb
--- /dev/null
+++ b/node_modules/ava/profile.js
@@ -0,0 +1,161 @@
+'use strict';
+
+// Iron-node does not work with forked processes
+// This cli command will run a single file in the current process.
+// Intended to be used with iron-node for profiling purposes.
+
+const path = require('path');
+const EventEmitter = require('events');
+const meow = require('meow');
+const Promise = require('bluebird');
+const pkgConf = require('pkg-conf');
+const findCacheDir = require('find-cache-dir');
+const uniqueTempDir = require('unique-temp-dir');
+const arrify = require('arrify');
+const resolveCwd = require('resolve-cwd');
+const babelConfigHelper = require('./lib/babel-config');
+const CachingPrecompiler = require('./lib/caching-precompiler');
+const globals = require('./lib/globals');
+
+function resolveModules(modules) {
+	return arrify(modules).map(name => {
+		const modulePath = resolveCwd(name);
+
+		if (modulePath === null) {
+			throw new Error(`Could not resolve required module '${name}'`);
+		}
+
+		return modulePath;
+	});
+}
+
+// Chrome gets upset when the `this` value is non-null for these functions
+globals.setTimeout = setTimeout.bind(null);
+globals.clearTimeout = clearTimeout.bind(null);
+
+Promise.longStackTraces();
+
+const conf = pkgConf.sync('ava', {
+	defaults: {
+		babel: 'default'
+	}
+});
+
+// Define a minimal set of options from the main CLI
+const cli = meow(`
+	Usage
+	  $ iron-node node_modules/ava/profile.js <test-file>
+
+	Options
+	  --fail-fast   Stop after first test failure
+	  --serial, -s  Run tests serially
+
+`, {
+	string: [
+		'_'
+	],
+	boolean: [
+		'fail-fast',
+		'verbose',
+		'serial',
+		'tap'
+	],
+	default: conf,
+	alias: {
+		s: 'serial'
+	}
+});
+
+if (cli.input.length !== 1) {
+	throw new Error('Specify a test file');
+}
+
+const file = path.resolve(cli.input[0]);
+const cacheDir = findCacheDir({
+	name: 'ava',
+	files: [file]
+}) || uniqueTempDir();
+
+babelConfigHelper.build(process.cwd(), cacheDir, conf.babel, true)
+	.then(result => {
+		const precompiler = new CachingPrecompiler({
+			path: cacheDir,
+			getBabelOptions: result.getOptions,
+			babelCacheKeys: result.cacheKeys
+		});
+
+		const precompiled = {};
+		precompiled[file] = precompiler.precompileFile(file);
+
+		const opts = {
+			file,
+			failFast: cli.flags.failFast,
+			serial: cli.flags.serial,
+			tty: false,
+			cacheDir,
+			precompiled,
+			require: resolveModules(conf.require)
+		};
+
+		const events = new EventEmitter();
+		let uncaughtExceptionCount = 0;
+
+		// Mock the behavior of a parent process
+		process.channel = {ref() {}, unref() {}};
+		process.send = data => {
+			if (data && data.ava) {
+				const name = data.name.replace(/^ava-/, '');
+
+				if (events.listeners(name).length > 0) {
+					events.emit(name, data.data);
+				} else {
+					console.log('UNHANDLED AVA EVENT:', name, data.data);
+				}
+
+				return;
+			}
+
+			console.log('NON AVA EVENT:', data);
+		};
+
+		events.on('test', data => {
+			console.log('TEST:', data.title, data.error);
+		});
+
+		events.on('results', data => {
+			if (console.profileEnd) {
+				console.profileEnd();
+			}
+
+			console.log('RESULTS:', data.stats);
+
+			if (process.exit) {
+				process.exit(data.stats.failCount + uncaughtExceptionCount); // eslint-disable-line unicorn/no-process-exit
+			}
+		});
+
+		events.on('stats', () => {
+			setImmediate(() => {
+				process.emit('ava-run', {});
+			});
+		});
+
+		events.on('uncaughtException', data => {
+			uncaughtExceptionCount++;
+			let stack = data && data.exception && data.exception.stack;
+			stack = stack || data;
+			console.log(stack);
+		});
+
+		// `test-worker` will read process.argv[2] for options
+		process.argv[2] = JSON.stringify(opts);
+		process.argv.length = 3;
+
+		if (console.profile) {
+			console.profile('AVA test-worker process');
+		}
+
+		setImmediate(() => {
+			require('./lib/test-worker'); // eslint-disable-line import/no-unassigned-import
+		});
+	});