Typescript + Jest：如何部分模拟类实例

Question

考虑两个类A 和B，如下所示：

class A {
    private b: B;

    public constructor(b: B){
        this.b=b;
    }

    public doSomething(){
        this.b.myMethod();
    }
}

class B {
    public myMethod(){...}
    public someOtherMethod(){...}
}

我想在模拟B.myMethod() 的行为的同时测试类A

目前我们这样做：

const bMock: Partial<B> = {
    myMethod: jest.fn(<some mock here>),
}

const sut = new A(bMock as any);

sut.doSomething();

expect(bMock.myMethod).toBeCalled();

我们想要实现的是类似的结果，但不必通过 as any 模拟，也不必自己模拟所有方法。检查模拟类型对我们来说非常重要，否则我们将无法通过此测试捕获模拟依赖项中的重大更改。

我们也已经查看了sinon，但在某些情况下，我们不希望调用模拟依赖项的构造函数，因此在创建对象后存根不是一种选择。对整个类进行存根会导致与上述类似的问题。

Answer 1

编辑：现在我们使用jest-mock-extended 来获得更好的笑话兼容性和更多功能

我使用Substitute找到了一个不错的解决方案。

唯一的问题是缺少对自述文件中提到的 null/undefined 的检查。不过还是比使用as any 好。

import Substitute, { SubstituteOf } from '@fluffy-spoon/substitute';

class A {
  private b: B;

  public constructor(b: B) {
    this.b = b;
  }

  public doSomething() {
    return this.b.myMethod();
  }
}

class B {
  public myMethod() {
    return 'realMethod';
  }
  public someOtherMethod() {
    return 'realSomeOtherMethod';
  }
}

let bMock: SubstituteOf<B>;
beforeEach(() => {
  bMock = Substitute.for<B>();
});

test('empty mock', () => {
  const sut = new A(bMock);
  console.log(sut.doSomething()); // Output: '[Function]'
});

test('mock myMethod', () => {
  bMock.myMethod().returns('You got mocked!');
  const sut = new A(bMock);
  console.log(sut.doSomething()); // Output:  'You got mocked!'
});

test('does not compile', () => {
  bMock.myMethod().returns(1337); // Will show compilation error due to incompatible type (string vs. number)
  const sut = new A(bMock);
  console.log(sut.doSomething());
});

test('missing null checks', () => {
  bMock.myMethod().returns(); // Will not complain
  const sut = new A(bMock);
  console.log(sut.doSomething()); // Output 'undefined'
});

Answer 2

创建一个泛型怎么样？ （您可以像在您的示例中一样使用 Partial，但您必须在调用它之前检查您是否提供了“sut.doSomething”实现。

class B {
  public myMethod(){}
  public someOtherMethod(){}
}

class A<T extends B> {
  private b: T;

  public constructor(b: T){
      this.b=b;
  }

  public doSomething(){
      this.b.myMethod();
  }
}

const bMock = {myMethod: jest.fn(), someOtherMethod: jest.fn()}
const sut = new A(bMock)
sut.doSomething()
expect(bMock.myMethod).toBeCalled();