Могу ли я издеваться над функциями с конкретными аргументами, используя 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?

Спасибо за ваши отзывы.

Ответ 1

Я нашел эту библиотеку, которую недавно написал мой коллега: 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

Ответ 2

Нет, пока нет возможности сделать это в Jest. Вы могли бы использовать заглушки sinons для этого. из документов:

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
}

Ответ 3

Это может помочь...

У меня было что-то подобное, в результате чего у меня был тот же метод, который вызывается с разными параметрами, требующими другого возвращенного результата от обрезанного/издевающегося вызова. Я использовал переменную со списком функций, когда я звонил в издеваемую службу, я беру функцию с верхней части очереди и выполняю функцию. Это требует знания порядка выполнения, который вы тестируете, и на самом деле не справляется с изменением ответа по аргументу, но позволило мне обойти ограничение в шутке.

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

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

Ответ 4

Ты можешь использовать:

const mockSum = jest.fn();

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