模拟函数

Mock functions are also known as “spies”, because they let you spy on the behavior of a function that is called indirectly by some other code, rather than only testing the output. You can create a mock function with jest.fn(). If no implementation is given, the mock function will return undefined when invoked. 你可以通过 jest.fn() 创建 mock 函数。 如果没有提供实现,调用模拟函数将返回 undefined

模拟函数 - 图1info

The TypeScript examples from this page will only work as documented if you import jest from '@jest/globals':

  1. import {jest} from '@jest/globals';

方法


参考

mockFn.getMockName()

Returns the mock name string set by calling mockFn.mockName(value).

mockFn.mock.calls

An array containing the call arguments of all calls that have been made to this mock function. Each item in the array is an array of arguments that were passed during the call. Each item in the array is an array of arguments that were passed during the call.

For example: A mock function f that has been called twice, with the arguments f('arg1', 'arg2'), and then with the arguments f('arg3', 'arg4'), would have a mock.calls array that looks like this:

  1. [
  2. ['arg1', 'arg2'],
  3. ['arg3', 'arg4'],
  4. ];

mockFn.mock.results

An array containing the results of all calls that have been made to this mock function. An array containing the results of all calls that have been made to this mock function. Each entry in this array is an object containing a type property, and a value property. type will be one of the following: type will be one of the following:

  • 'return' - Indicates that the call completed by returning normally.
  • 'throw' - Indicates that the call completed by throwing a value.
  • 'incomplete' - Indicates that the call has not yet completed. 'incomplete' - Indicates that the call has not yet completed. This occurs if you test the result from within the mock function itself, or from within a function that was called by the mock.

The value property contains the value that was thrown or returned. The value property contains the value that was thrown or returned. value is undefined when type === 'incomplete'.

For example: A mock function f that has been called three times, returning 'result1', throwing an error, and then returning 'result2', would have a mock.results array that looks like this:

  1. [
  2. {
  3. type: 'return',
  4. value: 'result1',
  5. },
  6. {
  7. type: 'throw',
  8. value: {
  9. /* Error instance */
  10. },
  11. },
  12. {
  13. type: 'return',
  14. value: 'result2',
  15. },
  16. ];

mockFn.mock.instances

An array that contains all the object instances that have been instantiated from this mock function using new.

For example: A mock function that has been instantiated twice would have the following mock.instances array:

  1. const mockFn = jest.fn();
  2. const a = new mockFn();
  3. const b = new mockFn();
  4. mockFn.mock.instances[0] === a; // true
  5. mockFn.mock.instances[1] === b; // true

mockFn.mock.contexts

一个包含mock函数所有调用上下文的数组。

A context is the this value that a function receives when called. The context can be set using Function.prototype.bind, Function.prototype.call or Function.prototype.apply.

例如:

  1. const mockFn = jest.fn();
  2. const boundMockFn = mockFn.bind(thisContext0);
  3. boundMockFn('a', 'b');
  4. mockFn.call(thisContext1, 'a', 'b');
  5. mockFn.apply(thisContext2, ['a', 'b']);
  6. mockFn.mock.contexts[0] === thisContext0; // true
  7. mockFn.mock.contexts[1] === thisContext1; // true
  8. mockFn.mock.contexts[2] === thisContext2; // true

mockFn.mock.lastCall

An array containing the call arguments of the last call that was made to this mock function. If the function was not called, it will return undefined.

For example: A mock function f that has been called twice, with the arguments f('arg1', 'arg2'), and then with the arguments f('arg3', 'arg4'), would have a mock.lastCall array that looks like this:

  1. ['arg3', 'arg4'];

mockFn.mockClear()

清空所有存储在mockFn.mock.calls, mockFn.mock.instances, mockFn.mock.contexts的信息和mockFn.mock.results 数组。 Often this is useful when you want to clean up a mocks usage data between two assertions.

请注意, mockFn.mockClear()会替换掉 mockFn.mock,而不只是重置其属性的值! You should, therefore, avoid assigning mockFn.mock to other variables, temporary or not, to make sure you don’t access stale data.

The clearMocks configuration option is available to clear mocks automatically before each tests.

mockFn.mockReset()

Does everything that mockFn.mockClear() does, and also removes any mocked return values or implementations.

This is useful when you want to completely reset a mock back to its initial state. (Note that resetting a spy will result in a function with no return value). (注意重置一个 spy 将导致一个没有返回值的函数)。

The mockReset configuration option is available to reset mocks automatically before each test.

mockFn.mockRestore()

Does everything that mockFn.mockReset() does, and also restores the original (non-mocked) implementation.

This is useful when you want to mock functions in certain test cases and restore the original implementation in others.

Beware that mockFn.mockRestore() only works when the mock was created with jest.spyOn(). Thus you have to take care of restoration yourself when manually assigning jest.fn().

The restoreMocks configuration option is available to restore mocks automatically before each test.

mockFn.mockImplementation(fn)

Accepts a function that should be used as the implementation of the mock. The mock itself will still record all calls that go into and instances that come from itself – the only difference is that the implementation will also be executed when the mock is called.

模拟函数 - 图2tip

jest.fn(implementation) is a shorthand for jest.fn().mockImplementation(implementation).

  • JavaScript
  • TypeScript
  1. const mockFn = jest.fn(scalar => 42 + scalar);
  2. mockFn(0); // 42
  3. mockFn(1); // 43
  4. mockFn.mockImplementation(scalar => 36 + scalar);
  5. mockFn(2); // 38
  6. mockFn(3); // 39
  1. const mockFn = jest.fn((scalar: number) => 42 + scalar);
  2. mockFn(0); // 42
  3. mockFn(1); // 43
  4. mockFn.mockImplementation(scalar => 36 + scalar);
  5. mockFn(2); // 38
  6. mockFn(3); // 39

.mockImplementation() 也可以用于模拟类构造器:

  • JavaScript
  • TypeScript

SomeClass.js

  1. module.exports = class SomeClass {
  2. method(a, b) {}
  3. };

SomeClass.test.js

  1. const SomeClass = require('./SomeClass');
  2. jest.mock('./SomeClass'); // this happens automatically with automocking
  3. const mockMethod = jest.fn();
  4. SomeClass.mockImplementation(() => {
  5. return {
  6. method: mockMethod,
  7. };
  8. });
  9. const some = new SomeClass();
  10. some.method('a', 'b');
  11. console.log('Calls to method: ', mockMethod.mock.calls);

SomeClass.ts

  1. export class SomeClass {
  2. method(a: string, b: string): void {}
  3. }

SomeClass.test.ts

  1. import {SomeClass} from './SomeClass';
  2. jest.mock('./SomeClass'); // this happens automatically with automocking
  3. const mockMethod = jest.fn<(a: string, b: string) => void>();
  4. SomeClass.mockImplementation(() => {
  5. return {
  6. method: mockMethod,
  7. };
  8. });
  9. const some = new SomeClass();
  10. some.method('a', 'b');
  11. console.log('Calls to method: ', mockMethod.mock.calls);

mockFn.mockImplementationOnce(fn)

Accepts a function that will be used as an implementation of the mock for one call to the mocked function. Can be chained so that multiple function calls produce different results. Can be chained so that multiple function calls produce different results.

  • JavaScript
  • TypeScript
  1. const mockFn = jest
  2. .fn()
  3. .mockImplementationOnce(cb => cb(null, true))
  4. .mockImplementationOnce(cb => cb(null, false));
  5. mockFn((err, val) => console.log(val)); // true
  6. mockFn((err, val) => console.log(val)); // false
  1. const mockFn = jest
  2. .fn<(cb: (a: null, b: boolean) => void) => void>()
  3. .mockImplementationOnce(cb => cb(null, true))
  4. .mockImplementationOnce(cb => cb(null, false));
  5. mockFn((err, val) => console.log(val)); // true
  6. mockFn((err, val) => console.log(val)); // false

When the mocked function runs out of implementations defined with .mockImplementationOnce(), it will execute the default implementation set with jest.fn(() => defaultValue) or .mockImplementation(() => defaultValue) if they were called:

  1. const mockFn = jest
  2. .fn(() => 'default')
  3. .mockImplementationOnce(() => 'first call')
  4. .mockImplementationOnce(() => 'second call');
  5. mockFn(); // 'first call'
  6. mockFn(); // 'second call'
  7. mockFn(); // 'default'
  8. mockFn(); // 'default'

mockFn.mockName(name)

用一个字符串来代替 'jest.fn()' 在测试结果的输出,用于表示正在引用哪个mock函数。

例如:

  1. const mockFn = jest.fn().mockName('mockedFunction');
  2. // mockFn();
  3. expect(mockFn).toHaveBeenCalled();

Will result in this error:

  1. expect(mockedFunction).toHaveBeenCalled()
  2. Expected mock function "mockedFunction" to have been called, but it was not called.

mockFn.mockReturnThis()

Syntactic sugar function for:

  1. jest.fn(function () {
  2. return this;
  3. });

mockFn.mockReturnValue(value)

Accepts a value that will be returned whenever the mock function is called.

  • JavaScript
  • TypeScript
  1. const mock = jest.fn();
  2. mock.mockReturnValue(42);
  3. mock(); // 42
  4. mock.mockReturnValue(43);
  5. mock(); // 43
  1. const mock = jest.fn<() => number>();
  2. mock.mockReturnValue(42);
  3. mock(); // 42
  4. mock.mockReturnValue(43);
  5. mock(); // 43

mockFn.mockReturnValueOnce(value)

Accepts a value that will be returned for one call to the mock function. Accepts a value that will be returned for one call to the mock function. Can be chained so that successive calls to the mock function return different values. When there are no more mockReturnValueOnce values to use, calls will return a value specified by mockReturnValue. When there are no more mockReturnValueOnce values to use, calls will return a value specified by mockReturnValue.

  • JavaScript
  • TypeScript
  1. const mockFn = jest
  2. .fn()
  3. .mockReturnValue('default')
  4. .mockReturnValueOnce('first call')
  5. .mockReturnValueOnce('second call');
  6. mockFn(); // 'first call'
  7. mockFn(); // 'second call'
  8. mockFn(); // 'default'
  9. mockFn(); // 'default'
  1. const mockFn = jest
  2. .fn<() => string>()
  3. .mockReturnValue('default')
  4. .mockReturnValueOnce('first call')
  5. .mockReturnValueOnce('second call');
  6. mockFn(); // 'first call'
  7. mockFn(); // 'second call'
  8. mockFn(); // 'default'
  9. mockFn(); // 'default'

mockFn.mockResolvedValue(value)

Syntactic sugar function for:

  1. jest.fn().mockImplementation(() => Promise.resolve(value));

Useful to mock async functions in async tests:

  • JavaScript
  • TypeScript
  1. test('async test', async () => {
  2. const asyncMock = jest.fn().mockResolvedValue(43);
  3. await asyncMock(); // 43
  4. });
  1. test('async test', async () => {
  2. const asyncMock = jest.fn<() => Promise<number>>().mockResolvedValue(43);
  3. await asyncMock(); // 43
  4. });

mockFn.mockResolvedValueOnce(value)

Syntactic sugar function for:

  1. jest.fn().mockImplementationOnce(() => Promise.resolve(value));

Useful to resolve different values over multiple async calls:

  • JavaScript
  • TypeScript
  1. test('async test', async () => {
  2. const asyncMock = jest
  3. .fn()
  4. .mockResolvedValue('default')
  5. .mockResolvedValueOnce('first call')
  6. .mockResolvedValueOnce('second call');
  7. await asyncMock(); // 'first call'
  8. await asyncMock(); // 'second call'
  9. await asyncMock(); // 'default'
  10. await asyncMock(); // 'default'
  11. });
  1. test('async test', async () => {
  2. const asyncMock = jest
  3. .fn<() => Promise<string>>()
  4. .mockResolvedValue('default')
  5. .mockResolvedValueOnce('first call')
  6. .mockResolvedValueOnce('second call');
  7. await asyncMock(); // 'first call'
  8. await asyncMock(); // 'second call'
  9. await asyncMock(); // 'default'
  10. await asyncMock(); // 'default'
  11. });

mockFn.mockRejectedValue(value)

Syntactic sugar function for:

  1. jest.fn().mockImplementation(() => Promise.reject(value));

Useful to create async mock functions that will always reject:

  • JavaScript
  • TypeScript
  1. test('async test', async () => {
  2. const asyncMock = jest
  3. .fn()
  4. .mockRejectedValue(new Error('Async error message'));
  5. await asyncMock(); // throws 'Async error message'
  6. });
  1. test('async test', async () => {
  2. const asyncMock = jest
  3. .fn<() => Promise<never>>()
  4. .mockRejectedValue(new Error('Async error message'));
  5. await asyncMock(); // throws 'Async error message'
  6. });

mockFn.mockRejectedValueOnce(value)

Syntactic sugar function for:

  1. jest.fn().mockImplementationOnce(() => Promise.reject(value));

Useful together with .mockResolvedValueOnce() or to reject with different exceptions over multiple async calls:

  • JavaScript
  • TypeScript
  1. test('async test', async () => {
  2. const asyncMock = jest
  3. .fn()
  4. .mockResolvedValueOnce('first call')
  5. .mockRejectedValueOnce(new Error('Async error message'));
  6. await asyncMock(); // 'first call'
  7. await asyncMock(); // throws 'Async error message'
  8. });
  1. test('async test', async () => {
  2. const asyncMock = jest
  3. .fn<() => Promise<string>>()
  4. .mockResolvedValueOnce('first call')
  5. .mockRejectedValueOnce(new Error('Async error message'));
  6. await asyncMock(); // 'first call'
  7. await asyncMock(); // throws 'Async error message'
  8. });

TypeScript Usage

模拟函数 - 图3tip

请参阅 [入门](GettingStarted. md#using-typescript) 指南,了解如何使用 TypeScript 设置Jest。

jest.fn(implementation?)

Correct mock typings will be inferred if implementation is passed to jest.fn(). There are many use cases where the implementation is omitted. To ensure type safety you may pass a generic type argument (also see the examples above for more reference):

  1. import {expect, jest, test} from '@jest/globals';
  2. import type add from './add';
  3. import calculate from './calc';
  4. test('calculate calls add', () => {
  5. // Create a new mock that can be used in place of `add`.
  6. const mockAdd = jest.fn<typeof add>();
  7. // `.mockImplementation()` now can infer that `a` and `b` are `number`
  8. // and that the returned value is a `number`.
  9. mockAdd.mockImplementation((a, b) => {
  10. // Yes, this mock is still adding two numbers but imagine this
  11. // was a complex function we are mocking.
  12. return a + b;
  13. });
  14. // `mockAdd` is properly typed and therefore accepted by anything
  15. // requiring `add`.
  16. calculate(mockAdd, 1, 2);
  17. expect(mockAdd).toBeCalledTimes(1);
  18. expect(mockAdd).toBeCalledWith(1, 2);
  19. });