Prevent exit before stdout is drained (#4983)

* Fix error handling in CLI tests

* Ensure process.stdout is closed before exiting

Author: lukastaegert

cli/cli.ts

@@ -21,6 +21,8 @@ if (command.help || (process.argv.length <= 2 && process.stdin.isTTY)) {
 	} catch {
 		// do nothing
 	}
-
-	run(command).then(() => process.exit(0));
+	run(command).then(() => {
+		process.stdout.on('finish', () => process.exit(0));
+		process.stdout.end();
+	});
 }

test/cli/index.js

@@ -58,88 +58,76 @@ async function runTest(config, command) {
 				killSignal: 'SIGKILL'
 			},
 			async (error, code, stderr) => {
-				if (config.after) {
-					await config.after(error, code, stderr);
-				}
-				if (error && !error.killed) {
-					if (config.error) {
-						if (!config.error(error)) {
-							return resolve();
+				try {
+					if (config.after) {
+						await config.after(error, code, stderr);
+					}
+					if (error && !error.killed) {
+						if (config.error) {
+							if (!config.error(error)) {
+								return resolve();
+							}
+						} else {
+							return reject(error);
 						}
-					} else {
-						return reject(error);
 					}
-				}
-				if (childProcess.signalCode === 'SIGKILL') {
-					return reject(new Error('Test aborted due to timeout.'));
-				}
+					if (childProcess.signalCode === 'SIGKILL') {
+						return reject(new Error('Test aborted due to timeout.'));
+					}
 
-				if ('stderr' in config) {
-					const shouldContinue = config.stderr(stderr);
-					if (!shouldContinue) return resolve();
-				} else if (stderr) {
-					console.error(stderr);
-				}
+					if ('stderr' in config) {
+						const shouldContinue = config.stderr(stderr);
+						if (!shouldContinue) return resolve();
+					} else if (stderr) {
+						console.error(stderr);
+					}
 
-				let unintendedError;
+					let unintendedError;
 
-				if (config.execute) {
-					try {
-						const function_ = new Function('require', 'module', 'exports', 'assert', code);
-						const module = {
-							exports: {}
-						};
-						function_(require, module, module.exports, assert);
+					if (config.execute) {
+						try {
+							const function_ = new Function('require', 'module', 'exports', 'assert', code);
+							const module = {
+								exports: {}
+							};
+							function_(require, module, module.exports, assert);
 
-						if (config.error) {
-							unintendedError = new Error('Expected an error while executing output');
-						}
+							if (config.error) {
+								unintendedError = new Error('Expected an error while executing output');
+							}
 
-						if (config.exports) {
-							await config.exports(module.exports);
+							if (config.exports) {
+								config.exports(module.exports);
+							}
+						} catch (error) {
+							if (config.error) {
+								config.error(error);
+							} else {
+								unintendedError = error;
+							}
 						}
-					} catch (error) {
-						if (config.error) {
-							config.error(error);
-						} else {
-							unintendedError = error;
-						}
-					}
 
-					if (config.show || unintendedError) {
-						console.log(code + '\n\n\n');
-					}
+						if (config.show || unintendedError) {
+							console.log(code + '\n\n\n');
+						}
 
-					if (config.solo) console.groupEnd();
+						if (config.solo) console.groupEnd();
 
-					return unintendedError ? reject(unintendedError) : resolve();
-				}
-				if (config.result) {
-					try {
+						return unintendedError ? reject(unintendedError) : resolve();
+					}
+					if (config.result) {
 						config.result(code);
 						return resolve();
-					} catch (error) {
-						return reject(error);
 					}
-				}
-				if (config.test) {
-					try {
+					if (config.test) {
 						config.test();
 						return resolve();
-					} catch (error) {
-						return reject(error);
 					}
-				}
-				if (existsSync('_expected') && statSync('_expected').isDirectory()) {
-					try {
+					if (existsSync('_expected') && statSync('_expected').isDirectory()) {
 						assertDirectoriesAreEqual('_actual', '_expected');
 						return resolve();
-					} catch (error) {
-						return reject(error);
 					}
-				}
-				const expected = readFileSync('_expected.js', 'utf8');
-				try {
+					const expected = readFileSync('_expected.js', 'utf8');
 					assert.equal(normaliseOutput(code), normaliseOutput(expected));
 					return resolve();
 				} catch (error) {