如何使用 jest 和 typscript 模擬模塊變量

Question

我有一個地圖定義了一些這樣的轉換函數

export const transformFuncMap: { [key: string]: (args: TransformFunctionArgs) => Promise<any> } = {
    [TransformationType.UNZIP]: unzipArchiveAndUploadToS3,
    // tslint:disable-next-line: no-empty
    [TransformationType.NOOP]: async () => {},
};

后來在同一個模塊中，我有一個句柄函數，它根據我提供的一些 seq 調用這些函數。

function handle(funcKeys: TransformationType[]) {
   for(const funcKey of funcKey) {
       await transformFuncMap[funcKey](); 
   }
}

當單元測試處理。 我只關心一些函數是按照我提供的順序調用的。 我不想運行函數實現。

無論如何，是否可以開玩笑地用這樣的東西來模擬transformFuncMap 。

export const transformFuncMap: { [key: string]: (args: TransformFunctionArgs) => Promise<any> } = {
    [TransformationType.UNZIP]:jest.fn(),
    // tslint:disable-next-line: no-empty
    [TransformationType.NOOP]: jest.fn(),
};

我希望能夠做到這一點，而無需在參數中使用某種 Java 風格的依賴注入。

Answer 1

您可以使用jest.fn()替換transformFuncMap的原始方法/函數。

index.ts ：

const unzipArchiveAndUploadToS3 = async () => null;

type TransformFunctionArgs = any;
export enum TransformationType {
  UNZIP = 'UNZIP',
  NOOP = 'NOOP',
}

export const transformFuncMap: { [key: string]: (args: TransformFunctionArgs) => Promise<any> } = {
  [TransformationType.UNZIP]: unzipArchiveAndUploadToS3,
  // tslint:disable-next-line: no-empty
  [TransformationType.NOOP]: async () => {},
};

export async function handle(funcKeys: TransformationType[]) {
  for (const funcKey of funcKeys) {
    const args = {};
    await transformFuncMap[funcKey](args);
  }
}

index.spec.ts ：

import { handle, transformFuncMap, TransformationType } from './';

describe('59383743', () => {
  it('should pass', async () => {
    transformFuncMap[TransformationType.UNZIP] = jest.fn();
    transformFuncMap[TransformationType.NOOP] = jest.fn();
    const funcKeys: TransformationType[] = [TransformationType.NOOP, TransformationType.UNZIP];
    await handle(funcKeys);
    expect(transformFuncMap[TransformationType.UNZIP]).toBeCalledWith({});
    expect(transformFuncMap[TransformationType.NOOP]).toBeCalledWith({});
  });
});

帶有覆蓋率報告的單元測試結果：

 PASS  src/stackoverflow/59383743/index.spec.ts
  59383743
    ✓ should pass (7ms)

----------|----------|----------|----------|----------|-------------------|
File      |  % Stmts | % Branch |  % Funcs |  % Lines | Uncovered Line #s |
----------|----------|----------|----------|----------|-------------------|
All files |    84.62 |      100 |       50 |       90 |                   |
 index.ts |    84.62 |      100 |       50 |       90 |                12 |
----------|----------|----------|----------|----------|-------------------|
Test Suites: 1 passed, 1 total
Tests:       1 passed, 1 total
Snapshots:   0 total
Time:        4.626s, estimated 10s

源代碼： https : //github.com/mrdulin/jest-codelab/tree/master/src/stackoverflow/59383743