Java - 使用Mockito进行测试模拟

Question

我对Mockito并不熟悉，但我正在尝试使用模拟来测试模拟的行为，这是类：

package simulator;

import java.util.Map;

import org.apache.commons.lang3.Validate;
import simulator.enums.Team;
import simulator.fixtures.Fixture;

public class SimulateBasketballMatchResult implements Simulation<Team> {

    private final Fixture fixture;

    public SimulateBasketballMatchResult(Fixture fixture) {

        Validate.notNull(fixture, "fixture cannot be null");

        this.fixture = fixture;
    }

    @Override
    public Team simulate(Map<Team, Double> outcomeProbabilityMap) {

        Validate.notNull(outcomeProbabilityMap, "outcomeProbabilityMap cannot be null");

        final Team homeTeam = fixture.getHomeTeam();
        final Team awayTeam = fixture.getAwayTeam();

        double random = randomDoubleGenerator();

        double homeWinProbability = outcomeProbabilityMap.get(homeTeam);

        return random < homeWinProbability ? homeTeam : awayTeam;

    }

    public Double randomDoubleGenerator() {
        return Math.random();
    }

}

以下是测试类：

@RunWith(MockitoJUnitRunner.class)
public class SimulateBasketballMatchResultTest {

    @Rule
    public ExpectedException expectedException = ExpectedException.none();

    private static final Map<Team, Double> MATCH_RESULT_PROBABILITY_MAP = new HashMap<>();

    private static final Fixture FIXTURE = new Fixture(GOLDEN_STATE_WARRIORS, HOUSTON_ROCKETS, REGULAR_SEASON);

    static {
        MATCH_RESULT_PROBABILITY_MAP.put(FIXTURE.getHomeTeam(), 0.7d);
        MATCH_RESULT_PROBABILITY_MAP.put(FIXTURE.getAwayTeam(), 0.3d);
    }

    @Mock
    private SimulateBasketballMatchResult simulateBasketballMatchResult;

    @Test
    public void shouldReturnGoldenStateWarriorsAsWinner() {

        when(simulateBasketballMatchResult.randomDoubleGenerator()).thenReturn(0.5d);

        assertThat(simulateBasketballMatchResult.simulate(MATCH_RESULT_PROBABILITY_MAP), is(GOLDEN_STATE_WARRIORS));

    }

}

我想声明GOLDEN_STATE_WARRIORS在概率范围介于0和GOLDEN_STATE_WARRIORS之间时返回 - 但是我的测试代码出现了一个null的断言错误。

java.lang.AssertionError: 
Expected: is <GOLDEN_STATE_WARRIORS>
     but: was null
Expected :is <GOLDEN_STATE_WARRIORS>

Answer 1

simulateBasketballMatchResult是一个模拟对象，因此默认情况下，它将为其所有方法返回null （当然，它具有非原始返回值）。

你应该窥探它，而不是嘲笑那个对象：

@Spy
private SimulateBasketballMatchResult simulateBasketballMatchResult = 
    new SimulateBasketballMatchResult(Fixture);

Answer 2

试试这个：

@Mock
private Fixture fixture;

private SimulateBasketballMatchResult simulator;

@Before
public void setUp() {
    simulator = spy(new SimulateBasketballMatchResult(fixture));
    doCallRealMethod().when(simulator).simulate();
}

@Test
public void shouldReturnGoldenStateWarriorsAsWinner() {
    doReturn(0.5).when(simulator).randomDoubleGenerator();
    when(fixture.getHomeTeam()).thenReturn(GOLDEN_STATE_WARRIORS);
    when(fixture.getAwayTeam()).thenReturn(HOUSTON_ROCKETS);

    assertThat(simulator.simulate(MATCH_RESULT_PROBABILITY_MAP), is(GOLDEN_STATE_WARRIORS));
}

Mockito.spy和@Spy允许你模拟一个真实对象的一些方法，但是Mockito.mock和@Mock模拟整个对象。

在一个的Mockito 假装是一个正常的模拟（允许你存根调用;也就是说，返回的特定值了方法调用）。

mockito中的间谍是部分模拟（对象的一部分将被模拟，部分将使用真实的方法调用）。

阅读更多