在 Jest 的 class 方法中测试 function 赋值

Question

我有一个 class，我在其中使用策略模式，它看起来像这样：

class Foo {
   constructor() {
      this.doStuff = function() { /* not used really */ }
   }
   create(config) {
      this.type = config.type;

      // assign other variables to this

      this.doStuff = StuffStrategy(config.type);

      this.doStuff();
   }
}

StuffStrategy 是一个 function，它返回根据类型不同使用this上下文的其他函数。

function StuffStrategy(type) {
   switch(type) {
     case A: return StrategyA;
     case B: return StrategyB;
     // ...
   }
}

function StrategyA() {
  if(this.someVarInFoo) {
    return 'a thing'
  } else 
    return 'a different thing' + this.otherVarInFoo
}

我在创建方法中分配了特定的策略 function。 然后我想测试create方法是否调用doStuff 。

describe('how create method works', () => {
    const instance = new Foo();
    const spy = jest.spyOn(instance, 'doStuff');
    instance.create(config); 
    expect(spy).toBeCalled();
});

但是当我在调用 instance.create 之前尝试制作间谍时，它指的是在构造函数中分配的默认方法，它在 create 中被替换。

如果我在调用 instance.create 之后进行监视，那么它不会接听电话。 我在定义 this.doStuff 时尝试添加.bind ：

this.doStuff = StuffStrategy(config.type).bind(this);

但它也不起作用。

我的设置有问题吗？ 我怎样才能使这个测试用例工作？

Answer 1

您必须监视您的 Foo class 的策略方法。因此对于每个config.type ，您检查然后调用了哪个策略方法。

export class Foo {
  constructor(){
    this.doStuff = null;
  }

  create(config){
     this.type = config.type;

     // assign other variables to this

     this.doStuff = StuffStrategy(config.type);

     this.doStuff();
  }

  strategyA(){...}

  strategyB(){...}

  StuffStrategy(configtype) {
    switch (configtype) {
     case "A": return this.strategyA;
     case "B": return this.strategyB;
   }
  }
}

import { Foo } from 'anyPlaceFoo/foo';

describe('Strategy', () => {

  it('should call strategy A', () => {
    const foo = new Foo();

    // here you can spy on every strategy method.
    jest.spyOn(foo, 'strategyA');
    jest.spyOn(foo, 'strategyB');

    foo.create({ type: 'A' });

    // check if the selected one has been called but not the others
    expect(foo.strategyA).toHaveBeenCalled();
    expect(foo.strategyB).not.toHaveBeenCalled();
  })
})