如何设置Mock有一个默认的行为，并可以在一些测试中覆盖它

Question

我想模拟一个依赖项，并在大多数测试用例中返回一个默认值，因为大多数测试用例都不应该关心返回的值，但是有些特定的情况下，我想测试，比如依赖项返回一些奇怪的值，或者只是抛出。所以我用这种方式来建模。大多数情况下，它应该返回一个好的有效值。

测试设置，默认情况下为所有测试类返回20L。

Dependency dependency = Mockito.mock(Dependency.class);
when(dependency.returnSomeVal()).thenReturn(20L);

在特定的测试用例类中，我想重写如下行为：

when(dependency.returnSomeVal()).thenThrow(); //failure cases
when(dependency.returnSomeVal()).thenReturn(Weird_Val); //failure cases

但是我没有找到一个好的解决方案来覆盖现有的行为？任何想法？

Answer 1

最后，我用自己的这种模式模拟了一个类中提供配置的一组方法。

在一个方法中，我为一个模拟对象设置了一组存根，这些存根为每个测试提供了正确的配置。之后，在每个测试中，只重写其中一个存根以提供不同的配置和测试不同的错误情况是非常方便的。

我认为Hari Menon的回答是正确的，但不知何故它违背了问题中解释的目的。如果重置模拟，则需要再次添加所有存根，这使得此模式非常混乱（与使用重置相比，最好不要使用任何重写，在这种情况下，代码将更加简单）。

添加到问题中的评论确实提供了一个间接的答案，说明了如何实现这一点，以及它为什么会起作用，但我花了一点时间才让它起作用。

尽管有一条评论，我还是通过在@Before fixture when（）中使用，使一切正常。然后返回（）并用doReturn（）覆盖混凝土桩。当（）

例子：

public class WorkerTest {

    private ConfigProvider mockedConfigProvider = mock(ConfigProvider.class);

    @Before
    public void setup() {
    // Setup stubs with a correct config
        when(mockedConfigProvider.getValue("property1")).thenReturn("value1");
        when(mockedConfigProvider.getValue("property2")).thenReturn("value2");
        when(mockedConfigProvider.getValue("property3")).thenReturn("value3");
        when(mockedConfigProvider.getValue("property4")).thenReturn("value4");
    }

    @Test
    public void test_GoodConfig(){
        // The config object gets injected in the test worker
        Worker testWorker = new Worker(mockedConfigProvider);
        // testWorker.execute() returns true if everything went well
        assertTrue(testWorker.execute());
    }

    @Test
    public void test_BadConfigProp1(){
        // Test now with a broken 'property1', overriding that stub.
        doReturn(null).when(mockedConfigProvider).getValue("property1");
        Worker testWorker = new Worker(mockedConfigProvider);
        // testWorker.execute() returns false if there is a problem.
        assertFalse(testWorker.execute());
    }

    @Test
    public void test_BadConfigProp2(){
    // This test needs to only override the result of property2
    doReturn("crazy result").when(mockedConfigProvider).getValue("property2");
    ...

}

Answer 2

您可以重置模拟和添加行为。在测试中，你要做什么

Mockito.reset(dependency);
when(dependency.returnSomeVal()).thenThrow(); //failure cases
when(dependency.returnSomeVal()).thenReturn(Weird_Val); //failure cases

重置将删除这个类上的所有模拟行为。如果您想只重写一些方法，那么您必须从头开始创建模拟。