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

import { Test } from '@nestjs/testing';
import { FilesController } from './files.controller';
import { FilesService } from './files.service';

describe('FilesController', () => {
  let filesController: FilesController;
  let filesService: FilesService;

  beforeEach(async () => {
    const moduleRef = await Test.createTestingModule({
      controllers: [FilesController],
      providers: [FilesService],
    }).compile();

    filesService = moduleRef.get<FilesService>(FilesService);
    filesController = moduleRef.get<FilesController>(FilesController);
  });

  describe('uploadFile', () => {
    it('should upload a file', async () => {
      const file = {
        originalname: 'test.txt',
        buffer: Buffer.from('Hello World'),
      };

      jest.spyOn(filesService, 'upload').mockImplementation(async () => {
        return { success: true, message: 'File uploaded successfully' };
      });

      const result = await filesController.uploadFile(file);

      expect(result).toEqual({ success: true, message: 'File uploaded successfully' });
      expect(filesService.upload).toHaveBeenCalledWith(file);
    });
  });
});