Имитация функций импортированного объекта в TypeScript - PullRequest
Новая
       67

Имитация функций импортированного объекта в TypeScript

0 голосов
/ 23 апреля 2020

У меня есть файл TypeScript, который экспортирует два объекта (созданные с помощью Тип goose). 

// 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);
}

Теперь я попытался смоделировать вызов 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.

Как я могу издеваться над функцией импортированный объект?

1 Ответ

1 голос
/ 28 апреля 2020

Вы должны реализовать заводскую функцию 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
...