在单元测试控制器 NestJs 中上传模拟文件

Question

我在 nest js 控制器中有一个方法，它带有两个参数，如下所示：

@UseGuards(JwtAuthGuard, RolesGuard)
  @Roles(MemberType.OWNER)
  @Post('')
  @UseInterceptors(
    FilesInterceptor('images', 4, {
      storage: fileStorageProductImage,
      fileFilter: fileFilter,
      limits: fileSizeLimit,
    }),
  )
  @UseInterceptors(new MulterInterceptor('images'))
  public async addProduct(
    @Req()
    req: Request,
    @UploadedFiles() files: Express.Multer.File[],
  ) {
  
  //code
  //
  }

这是我的单元测试

describe('Unit testing path 1', () => {
      const MemberServiceProvider = {
        provide: MemberService,
        useClass: MemberServiceStub,
      };
      beforeEach(async () => {
        const module: TestingModule = await Test.createTestingModule({
          controllers: [ProductsController],
          providers: [MemberServiceProvider],
        }).compile();
        controller = module.get<ProductsController>(ProductsController);
      });
      //
      const result = controller.addProduct(..., ...);
    });

我目前不知道要将什么传递给 addProduct 方法，因为我不知道如何模拟请求对象和 Express.Multer.File[]。提前致谢

Answer 1

对于仍在寻找答案的任何人。我只是想知道如何解决这个问题。首先，我们需要定义一个方法，该方法可用于通过流从本地机器读取文件。

const fileToBuffer = (filename) => {
  const readStream = fs.createReadStream(filename);
  const chunks = [];
  return new Promise((resolve, reject) => {
    // Handle any errors while reading
    readStream.on('error', (err) => {
      // handle error

      // File could not be read
      reject(err);
    });

    // Listen for data
    readStream.on('data', (chunk) => {
      chunks.push(chunk);
    });

    // File is done being read
    readStream.on('close', () => {
      // Create a buffer of the image from the stream
      resolve(Buffer.concat(chunks));
    });
  });
};

然后，我们在单元测试中调用该方法

const imageBuffer = (await fileToBuffer(
          __dirname + 'path-to-file',
        )) as Buffer;

在我们的控制器中，上传的文件具有 Express.Multer.File[] 类型，因此我们必须创建与上述类型相同的变量。

const imageFiles: Express.Multer.File[] = [];
const myReadableStreamBuffer = new streamBuffers.ReadableStreamBuffer({
      frequency: 10, // in milliseconds.
      chunkSize: 2048, // in bytes.
});
myReadableStreamBuffer.put(imageBuffer as Buffer);
imageFiles.push({
    buffer: imageBuffer,
    fieldname: 'fieldname-defined-in-@UseInterceptors-decorator',
    originalname: 'original-filename',
    encoding: '7bit',
    mimetype: 'file-mimetyp',
    destination: 'destination-path',
    filename: 'file-name',
    path: 'file-path',
    size: 955578,
    stream: myReadableStreamBuffer,
});

最后，我们可以使用新创建的 imagesFiles 变量作为参数调用我们的 addProduct 方法。