0

我有一个 TypeScript 文件，它导出两个对象（由Typegoose创建）。

// my-document.ts
export class MyDocument extends Typegoose {
  // ...
}

export const MyDocumentModel = new MyDocument().getModelForClass(MyDocument, ...)

我将脚本导入到被测脚本中。

// script-under-test.ts
import { MyDocumentModel } from './my-document';

export async function createDocument(doc: any) {
  return await MyDocumentModel.create(doc);
}

现在我试图模拟 call MyDocumentModel.create(doc)。

// script-under-test.test.ts
import { MyDocumentModel } from './my-document';
import { createDocument } from './script-under-test.ts';

jest.mock('./my-document');

describe('creation', () => {
  const MyDocumentModelMock = MyDocumentModel as unknown as jest.Mock;

  it('should create a new document', async () => {
    const payload = { foo: 'bar' };
    const document = {} as Document;
    const createMock = jest.fn();
    createMock.mockReturnValueOnce(document);
    MyDocumentModel.mockImplementation(() => ({ create: createMock }));

    const result = await createDocument(payload);

    expect(createMock).toHaveBeenCalledWith(payload);
    expect(result).toStrictEqual(document);
  });
});

但我收到以下错误 - Number of calls: 0。

如何模拟导入对象的函数？

4

1 回答 1

1

您应该实现jest.mock(moduleName, factory, options)的工厂功能。

例如

my-document.ts：

// simulate the Typegoose class
class Typegoose {
  public getModelForClass(cls) {
    return cls.toString();
  }
}

export class MyDocument extends Typegoose {}

export const MyDocumentModel = new MyDocument().getModelForClass(MyDocument);

script-under-test.ts：

import { MyDocumentModel } from './my-document';

export async function createDocument(doc: any) {
  return await MyDocumentModel.create(doc);
}

script-under-test.test.ts：

import { MyDocumentModel } from './my-document';
import { createDocument } from './script-under-test';

jest.mock('./my-document', () => {
  const mMyDocumentModel = { create: jest.fn() };
  return { MyDocumentModel: mMyDocumentModel };
});

describe('creation', () => {
  it('should create a new document', async () => {
    const payload = { foo: 'bar' };
    MyDocumentModel.create.mockResolvedValueOnce(document);
    const result = await createDocument(payload);
    expect(MyDocumentModel.create).toHaveBeenCalledWith(payload);
    expect(result).toStrictEqual(document);
  });
});

覆盖率 100% 的单元测试结果：

 PASS  stackoverflow/61388982/ script-under-test.test.ts (10.103s)
  creation
    ✓ should create a new document (5ms)

----------------------|---------|----------|---------|---------|-------------------
File                  | % Stmts | % Branch | % Funcs | % Lines | Uncovered Line #s 
----------------------|---------|----------|---------|---------|-------------------
All files             |     100 |      100 |     100 |     100 |                   
 script-under-test.ts |     100 |      100 |     100 |     100 |                   
----------------------|---------|----------|---------|---------|-------------------
Test Suites: 1 passed, 1 total
Tests:       1 passed, 1 total
Snapshots:   0 total
Time:        11.535s
于 2020-04-28T03:45:01.170 回答