如何使用 jest 模擬在回調函數中發回的結果

Question

有沒有辦法模擬storage對象，以便我可以在執行storage.get時控制回調函數中傳回的err和v ？

const mod = (() => {
  let value;

  return {
    init: (storage) => {
      storage.get('key', (err, v) => {
        value = v;
      });
    },
    get: () => value
  };
})();

由於我關心storage.get副作用，我不能簡單地執行mockImplementation來覆蓋該函數。 那是對的嗎？

Answer 1

創建一個mod.init storage對象並將其傳遞給mod.init方法。

例如

index.ts ：

const mod = (() => {
  let value;

  return {
    init: (storage) => {
      storage.get('key', (err, v) => {
        value = v;
      });
    },
    get: () => value,
  };
})();

export { mod };

index.test.ts ：

import { mod } from './';

describe('63640360', () => {
  it('should pass', () => {
    const mStorage = {
      get: jest.fn().mockImplementationOnce((key, callback) => {
        callback(null, 'teresa teng');
      }),
    };
    mod.init(mStorage);
    const actual = mod.get();
    expect(actual).toBe('teresa teng');
    expect(mStorage.get).toBeCalledWith('key', expect.any(Function));
  });
});

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

 PASS  src/stackoverflow/63640360/index.test.ts
  63640360
    ✓ should pass (5ms)

----------|----------|----------|----------|----------|-------------------|
File      |  % Stmts | % Branch |  % Funcs |  % Lines | Uncovered Line #s |
----------|----------|----------|----------|----------|-------------------|
All files |      100 |      100 |      100 |      100 |                   |
 index.ts |      100 |      100 |      100 |      100 |                   |
----------|----------|----------|----------|----------|-------------------|
Test Suites: 1 passed, 1 total
Tests:       1 passed, 1 total
Snapshots:   0 total
Time:        4.079s, estimated 9s