typescript - 开玩笑 mock rxjs 的 lastValueFrom

Question

我有一个使用 rxjs 的 lastValueFrom 方法的 Controller ，我想模拟该方法，这样实际的方法就不会被调用。我的 Controller 代码是:

async doSomething(request) {
  ...
  const data = await lastValueFrom(
    ...
  );
  ...
}

为了测试，我尝试了多种方式，即:

import rxjs from 'rxjs';

jest.spyOn(rxjs, 'lastValueFrom').mockReturnValueOnce(
  new Promise((resolve, reject) => {
    resolve(true);
  }),
);
// This gives me error: Cannot spyOn on a primitive value; undefined given

import { lastValueFrom } from 'rxjs';

lastValueFrom = jest.fn().mockReturnValueOnce(
  new Promise((resolve, reject) => {
    resolve(true),
);
// This gives me error: Cannot assign to 'lastValueFrom' because it is an import.

import rxjs from 'rxjs';

rxjs.lastValueFrom = jest.fn().mockReturnValue(
  new Promise((resolve, reject) => {
    resolve(true);
  }),
);
// This gives me error: Cannot set property 'lastValueFrom' of undefined

Answer 1

如果您使用的是 TypeScript，恐怕唯一的方法是模拟导入 'rxjs'。

jest.mock('rxjs', () => {
  const original = jest.requireActual('rxjs');

  return {
    ...original,
    lastValueFrom: () =>
      new Promise((resolve, reject) => {
        resolve(true);
      }),
  };
});

import { of, lastValueFrom } from 'rxjs';

test('test', () => {
  lastValueFrom(of(1)).then((val) => {
    expect(val).toStrictEqual(true);
  });
});

现场演示:https://stackblitz.com/edit/node-6vjbxb?file=index.spec.ts

只用 ES* 似乎你可以做到:How to spy on a default exported function with Jest?