使用Jest模拟Node.js流管道

const { Readable, Writable, Transform } = require('stream');

// 创建一个可读流
class MyReadable extends Readable {
  _read() {
    // 从底层数据源读取数据，并使用push方法将数据推送到流中
    // 这里可以模拟不同的数据输入
    this.push('data1');
    this.push('data2');
    this.push(null); // 数据读取完毕，推送null表示结束
  }
}

// 创建一个可写流
class MyWritable extends Writable {
  _write(chunk, encoding, callback) {
    // 处理流中的数据
    console.log(chunk.toString());
    callback();
  }
}

// 创建一个转换流
class MyTransform extends Transform {
  _transform(chunk, encoding, callback) {
    // 对流中的数据进行转换处理
    const transformedData = chunk.toString().toUpperCase();
    this.push(transformedData);
    callback();
  }
}

// 创建测试用例
test('test stream pipeline', () => {
  const readable = new MyReadable();
  const writable = new MyWritable();
  const transform = new MyTransform();

  // 使用Jest提供的模拟功能，模拟流的输入和输出
  const mockWrite = jest.spyOn(writable, '_write');
  const mockTransform = jest.spyOn(transform, '_transform');

  // 将流对象连接起来，形成流管道
  readable.pipe(transform).pipe(writable);

  // 断言模拟的写入和转换方法是否被调用
  expect(mockWrite).toHaveBeenCalledTimes(2);
  expect(mockTransform).toHaveBeenCalledTimes(2);
});