在 Typescript 中使用 jest 进行测试时如何模拟 @azure/storage-blob

如何解决在 Typescript 中使用 jest 进行测试时如何模拟 @azure/storage-blob

我有个方法，他的功能是通过storage.blob()下载一个文件，但是我不能正确的进行mock。 我总是以错误返回结束，因为我已经验证了download()函数没有执行。

这是您下载的存储库的代码

import * as Azure from '@azure/storage-blob';
import AzureManager from '@shared/infra/azure/AzureManager';
import connectionUrl from '@shared/infra/azure/helpers/connectionUrl';
import IDocumentsRepository from '../IDocumentsRepository';
import streamToString from './helpers/streamToString';
import azureConfig from '@config/azure';
export default class AzureDocumentsRepository implements IDocumentsRepository {
  public async getBlob(name: string): Promise<string> {
    const connection = await AzureManager.connect(connectionUrl());

    const blobUrl = Azure.BlobURL.fromContainerURL(connection,name);
    const downloadBlockResponse = await blobUrl.download(Azure.Aborter.none,0);
    return streamToString(
      downloadBlockResponse.readableStreamBody as NodeJS.ReadableStream
    );
  }
}

这是您下载的存储库的代码

import { Response } from 'express';
import { inject,injectable,container } from 'tsyringe';

import Zs from 'zip-stream';

import ILogger from '@shared/infra/logger/interfaces/ILogger';
import IDocumentsRepository from '../repositories/IDocumentsRepository';

import ZipUseCase from './ZipUseCase';

interface IRequest {
  files: string[];
  requestId: string;
  requestingUserId: string;
  response: Response;
}

@injectable()
export default class GetDownloadXmlUseCase {
  constructor(
    @inject('logger') private logger: ILogger,@inject('DocumentsRepository')
    private documentsRepository: IDocumentsRepository
  ) {}

  // eslint-disable-next-line consistent-return
  public async execute({
    files,requestId,requestingUserId,response,}: IRequest): Promise<any> {
    this.logger.log('info','downloading xml nfe',{
      files,});
    const archiver = new Zs();
    archiver.pipe(response);
    const zipUseCase = container.resolve(ZipUseCase);
    for (const file of files) {
      try {
        const xml = await this.documentsRepository.getBlob(
          `nfe/${file.trim()}.xml`
        );
        console.log('XML',xml);
        await zipUseCase.execute({
          archiver,file,xml,});
      } catch (error) {
        if (typeof response.status === 'function') {
          return response.status(500).json({ ...error });
        }
        throw error;
      }
    }
      archiver.finish(); 
  }
}

这里是测试代码

/* eslint-disable global-require */
import 'reflect-Metadata';
import { container } from 'tsyringe';
import { Response } from 'express';
import { createResponse } from 'node-mocks-http';

import AppError from '@shared/errors/AppError';
import faker from '@shared/infra/faker/Faker';

import ILogger from '@shared/infra/logger/interfaces/ILogger';
import blobFactory from '../factories/blobFactory';

import AzureDocumentsRepository from '@modules/documents/repositories/implementations/AzureDocumentsRepository';
import GetDownloadXmlUseCase from './GetDownloadXmlUseCase';
import IDocumentsRepository from '../repositories/IDocumentsRepository';
import FakerLogger from '@shared/infra/logger/fakes/FakeLogger';
import AzureManager from '@shared/infra/azure/AzureManager';
import * as Azure from '@azure/storage-blob';
import {
  ContainerURL,BlobURL,BlockBlobURL,Pipeline,ServiceURL,} from '@azure/storage-blob';

jest.mock('@azure/storage-blob');

jest.mock('zip-stream');

container.registerSingleton<IDocumentsRepository>(
  'DocumentsRepository',AzureDocumentsRepository
);

container.registerInstance<ILogger>('logger',FakerLogger);

const azureRepository = new AzureDocumentsRepository();

const getDownloadXmlUseCase = new GetDownloadXmlUseCase(
  FakerLogger,azureRepository
);

const requestingUserId = faker.guid({ version: 4 });
const requestId = faker.guid({ version: 4 });

describe('GetDownloadXmlUseCase',() => {
  beforeAll(async () => {});
  beforeEach(() => {});
  afterall(async () => {});
  it('should get download xml with nfe',async () => {
    const resp: Response = createResponse();

    const response = await getDownloadXmlUseCase.execute({
      files: ['CTe29180701568962000104570010000119791007119444'],response: resp,});

    expect(response).toBe(true);
  });
});