如何为Node.js中使用readline的函数编写测试并实现自动输入

你可以通过两种常见方案实现readline测试的自动化输入：

方案1：Mock readline 模块（单元测试推荐） #

该方案无需修改原有业务代码，通过模拟readline的返回结果直接测试逻辑，执行速度快，隔离性好。

1. 先安装mock工具依赖：
   npm install sinon --save-dev
2. 调整测试文件input-test.js代码如下：

const chai = require('chai');
chai.should();
const sinon = require('sinon');
const readline = require('readline');
const input = require('../index.js');

it('test for inputNum', async function () {
  // 构造readline实例的模拟对象
  const mockRl = {
    // 调用question时直接触发回调，传入模拟的用户输入值'1'
    question: sinon.stub().callsArgWith(1, '1'),
    close: sinon.stub()
  };
  // 替换readline.createInterface方法，返回我们构造的模拟实例
  const createInterfaceStub = sinon.stub(readline, 'createInterface').returns(mockRl);

  try {
    const res = await input();
    res.should.be.equal('1');
    // 可选：验证方法调用符合预期
    mockRl.question.calledOnce.should.be.true;
    mockRl.close.calledOnce.should.be.true;
  } finally {
    // 恢复被替换的方法，避免影响其他测试用例
    createInterfaceStub.restore();
  }
});

方案2：替换process.stdin输入流（集成测试适用） #

如果需要贴近真实运行场景，不mock底层依赖，可以手动替换标准输入流写入模拟数据：
修改input-test.js代码如下：

const chai = require('chai');
chai.should();
const { PassThrough } = require('stream');
const input = require('../index.js');

it('test for inputNum', async function () {
  // 暂存原始stdin对象，测试结束后恢复
  const originalStdin = process.stdin;
  // 创建可写的模拟流
  const mockStdin = new PassThrough();
  // 替换全局process.stdin
  Object.defineProperty(process, 'stdin', {
    value: mockStdin
  });

  try {
    // 写入模拟输入，末尾加\n模拟用户按下回车提交
    mockStdin.write('1\n');
    const res = await input();
    res.should.be.equal('1');
  } finally {
    // 恢复原始stdin
    Object.defineProperty(process, 'stdin', {
      value: originalStdin
    });
  }
});

方案选择建议 #

• 纯逻辑验证的单元测试优先选方案1，不需要处理流的异步逻辑，稳定性更高
• 需要验证完整输入流程的集成测试可以选方案2，更贴近真实运行表现

内容的提问来源于stack exchange，提问作者Sanket Shevkar