stream locking

Author: YDarma

index.js

@@ -5,6 +5,7 @@ var createDefaultStream = require('./lib/default_stream');
 var Test = require('./lib/test');
 var createResult = require('./lib/results');
 var through = require('@ljharb/through');
+var EventEmitter = require('events').EventEmitter;
 
 var canEmitExit = typeof process !== 'undefined' && process
 	&& typeof process.on === 'function' && process.browser !== true;
@@ -53,6 +54,10 @@ module.exports = (function () {
 		return harness.createStream(options);
 	};
 
+	lazyLoad.async = function () {
+		return getHarness().async.apply(this, arguments);
+	};
+
 	lazyLoad.onFinish = function () {
 		return getHarness().onFinish.apply(this, arguments);
 	};
@@ -142,10 +147,15 @@ function createExitHarness(conf, wait) {
 		stream.on('end', function () { ended = true; });
 	}
 
+	run();
+
 	if (wait) {
-		harness.run = run;
-	} else {
-		run();
+		var waiter = new EventEmitter();
+		waiter.run = function () {};
+		harness._results.push(waiter);
+		harness.run = function () {
+			waiter.emit('end');
+		};
 	}
 
 	if (config.exit === false) { return harness; }

test/wait-run-object-stream.js

@@ -0,0 +1,33 @@
+'use strict';
+
+var tap = require('tap');
+var tape = require('../');
+
+tap.test('Create stream then import async tests', function (t) {
+    t.plan(3);
+
+    var actualTests = function () {
+        tape('This one should pass', function (t1) {
+            t1.pass('This one should pass');
+            t1.end();
+        });
+    };
+
+    var simulateAsyncEsmImport = function () {
+        return new Promise(function (resolve) {
+            setTimeout(function () {
+                actualTests();
+                resolve();
+            });
+        });
+    };
+
+    tape.createStream({ objectMode: true }).on('data', function (res) {
+        t.pass(res.type);
+    });
+
+    tape.wait();
+    simulateAsyncEsmImport().then(function () {
+        tape.run();
+    });
+});