test: add tests for REPL custom evals

this commit reintroduces the REPL custom eval tests that have
been introduced in #57691
but reverted in #57793

the tests turned out problematic before because `getReplOutput`,
the function used to return the repl output wasn't taking into
account that input processing and output emitting are asynchronous
operation can can resolve with a small delay

Author: dario-piotrowicz

lib/repl.js

@@ -302,7 +302,6 @@ function REPLServer(prompt,
     options.useColors = shouldColorize(options.output);
   }
 
-  // TODO(devsnek): Add a test case for custom eval functions.
   const preview = options.terminal &&
     (options.preview !== undefined ? !!options.preview : !eval_);
 

test/parallel/test-repl-custom-eval.js

@@ -0,0 +1,164 @@
+'use strict';
+
+require('../common');
+const ArrayStream = require('../common/arraystream');
+const assert = require('assert');
+const { describe, it } = require('node:test');
+
+const repl = require('repl');
+
+// processes some input in a REPL instance and returns a promise that
+// resolves to the produced output (as a string), such promise resolves
+// when the repl output stream has been deemed idle.
+function getReplOutput(input, replOptions, run = true) {
+  return new Promise((resolve) => {
+    const inputStream = new ArrayStream();
+    const outputStream = new ArrayStream();
+
+    const replServer = repl.start({
+      input: inputStream,
+      output: outputStream,
+      ...replOptions,
+    });
+
+    let output = '';
+
+    let resolveOnIdleTimeout;
+    const resetResolveOnIdleTimeout = () => {
+      if (resolveOnIdleTimeout) {
+        clearTimeout(resolveOnIdleTimeout);
+      }
+      resolveOnIdleTimeout = setTimeout(() => {
+        replServer.close();
+        resolve(output);
+      }, 500);
+    };
+    resetResolveOnIdleTimeout();
+
+    outputStream.write = (chunk) => {
+      resetResolveOnIdleTimeout();
+      output += chunk;
+    };
+
+    inputStream.emit('data', input);
+
+    if (run) {
+      inputStream.run(['']);
+    }
+
+    return output;
+  });
+}
+
+describe('repl with custom eval', { concurrency: true }, () => {
+  it('uses the custom eval function as expected', async () => {
+    const output = await getReplOutput('Convert this to upper case', {
+      terminal: true,
+      eval: (code, _ctx, _replRes, cb) => cb(null, code.toUpperCase()),
+    });
+    assert.match(
+      output,
+      /Convert this to upper case\r\n'CONVERT THIS TO UPPER CASE\\n'/
+    );
+  });
+
+  it('surfaces errors as expected', async () => {
+    const output = await getReplOutput('Convert this to upper case', {
+      terminal: true,
+      eval: (_code, _ctx, _replRes, cb) => cb(new Error('Testing Error')),
+    });
+    assert.match(output, /Uncaught Error: Testing Error\n/);
+  });
+
+  it('provides a repl context to the eval callback', async () => {
+    const context = await new Promise((resolve) => {
+      const r = repl.start({
+        eval: (_cmd, context) => resolve(context),
+      });
+      r.context = { foo: 'bar' };
+      r.write('\n.exit\n');
+    });
+    assert.strictEqual(context.foo, 'bar');
+  });
+
+  it('provides the global context to the eval callback', async () => {
+    const context = await new Promise((resolve) => {
+      const r = repl.start({
+        useGlobal: true,
+        eval: (_cmd, context) => resolve(context),
+      });
+      global.foo = 'global_foo';
+      r.write('\n.exit\n');
+    });
+
+    assert.strictEqual(context.foo, 'global_foo');
+    delete global.foo;
+  });
+
+  it('inherits variables from the global context but does not use it afterwords if `useGlobal` is false', async () => {
+    global.bar = 'global_bar';
+    const context = await new Promise((resolve) => {
+      const r = repl.start({
+        useGlobal: false,
+        eval: (_cmd, context) => resolve(context),
+      });
+      global.baz = 'global_baz';
+      r.write('\n.exit\n');
+    });
+
+    assert.strictEqual(context.bar, 'global_bar');
+    assert.notStrictEqual(context.baz, 'global_baz');
+    delete global.bar;
+    delete global.baz;
+  });
+
+  /**
+   * Default preprocessor transforms
+   *  function f() {}  to
+   *  var f = function f() {}
+   * This test ensures that original input is preserved.
+   * Reference: https://github.com/nodejs/node/issues/9743
+   */
+  it('preserves the original input', async () => {
+    const cmd = await new Promise((resolve) => {
+      const r = repl.start({
+        eval: (cmd) => resolve(cmd),
+      });
+      r.write('function f() {}\n.exit\n');
+    });
+    assert.strictEqual(cmd, 'function f() {}\n');
+  });
+
+  it("doesn't show previews by default", async () => {
+    const input = "'Hello custom' + ' eval World!'";
+    const output = await getReplOutput(
+      input,
+      {
+        terminal: true,
+        eval: (code, _ctx, _replRes, cb) => cb(null, eval(code)),
+      },
+      false
+    );
+    assert.strictEqual(output, input);
+    assert.doesNotMatch(output, /Hello custom eval World!/);
+  });
+
+  it('does show previews if `preview` is set to `true`', async () => {
+    const input = "'Hello custom' + ' eval World!'";
+    const output = await getReplOutput(
+      input,
+      {
+        terminal: true,
+        eval: (code, _ctx, _replRes, cb) => cb(null, eval(code)),
+        preview: true,
+      },
+      false
+    );
+
+    const escapedInput = input.replace(/\+/g, '\\+'); // TODO: migrate to `RegExp.escape` when it's available.
+    assert.match(
+      output,
+      new RegExp(`${escapedInput}\n// 'Hello custom eval World!'`)
+    );
+  });
+});