web-dev-qa-db-ja.com

Jestを使用して特定の引数を持つ関数をモックできますか?

Jestで関数をモックしたいのですが、特定の引数で呼び出された場合にのみ、たとえば:

function sum(x, y) {
  return x + y;
}

// mock sum(1, 1) to return 4
sum(1, 1) // returns 4 (mocked)
sum(1, 2) // returns 3 (not mocked)

RubyのRSpecライブラリには同様の機能が実装されています。

class Math
  def self.sum(x, y)
    return x + y
  end
end

allow(Math).to receive(:sum).with(1, 1).and_return(4)
Math.sum(1, 1) # returns 4 (mocked)
Math.sum(1, 2) # returns 3 (not mocked)

私のテストで達成しようとしているのは、より良い分離です。たとえば、sumに依存する関数をテストしたいとしましょう。

function sum2(x) {
  return sum(x, 2);
}

// I don't want to depend on the sum implementation in my tests, 
// so I would like to mock sum(1, 2) to be "anything I want", 
// and so be able to test:

expect(sum2(1)).toBe("anything I want");

// If this test passes, I've the guarantee that sum2(x) is returning
// sum(x, 2), but I don't have to know what sum(x, 2) should return

私は次のようなことをすることでこれを実装する方法があることを知っています:

sum = jest.fn(function (x, y) {
  if (x === 1 && y === 2) {
    return "anything I want";
  } else {
    return sum(x, y);
  }
});

expect(sum2(1)).toBe("anything I want");

しかし、それを単純化するための糖の機能があればいいでしょう。

それは合理的に聞こえますか? Jestにはすでにこの機能がありますか?

ご意見をいただきありがとうございます。

26
Nícolas Iensen

私の同僚が最近書いたこのライブラリを見つけました:jest-when

import { when } from 'jest-when';

const fn = jest.fn();
when(fn).calledWith(1).mockReturnValue('yay!');

const result = fn(1);
expect(result).toEqual('yay!');

ライブラリは次のとおりです。 https://github.com/timkindberg/jest-when

19
STeve Shary

いいえ、Jestでこれを行う方法はまだありません。これには sinons stubs を使用できます。ドキュメントから:

stub.withArgs(arg1 [、arg2、...]);

指定された引数に対してのみメソッドをスタブします。これは、同じ呼び出しでスパイにアクセスできるアサーションで表現力を高めるのに役立ちます。また、異なる引数に応じて異なる動作をすることができるスタブを作成すると便利です。

"test should stub method differently based on arguments": function () {
    var callback = sinon.stub();
    callback.withArgs(42).returns(1);
    callback.withArgs(1).throws("TypeError");

    callback(); // No return value, no exception
    callback(42); // Returns 1
    callback(1); // Throws TypeError
}
9

次を使用できます。

const mockSum = jest.fn();

mockSum.mockImplementation((x, y) => {
  // Return whatever you want based on x and y...
});
6
Daniel Muñoz

これは役立つかもしれません...

似たようなものがあり、スタブ化/モック化された呼び出しから異なる戻り結果を必要とする異なるパラメーターで同じメソッドが呼び出されました。私は、モックされたサービスを呼び出したときに関数のリストで変数を使用しました。キューの先頭から関数を取り出し、関数を実行します。テストする実行順序の知識が必要であり、引数による応答の変更を実際に処理しませんが、jestの制限を回避することができました。

var mockedQueue = [];
mockedQueue.Push(() => {return 'A';})
mockedQueue.Push(() => {return 'B';})

service.invoke = jest.fn(()=>{
    serviceFunctionToCall = mockedQueue.shift();
    return serviceFunctionToCall();
})
2
Adam