Mockito 重新存根方法已使用 thenthrow 存根

Question

我遇到了mockito的问题。 我正在开发一个网络应用程序。在我的测试中，用户管理被嘲笑。 在某些情况下，我必须更改 getLoggedInUser() 方法返回的用户。

问题是，我的 getLoggedInUser() 方法也可能引发 AuthenticationException。

因此，当我尝试从无用户切换到某个用户时，调用会

when(userProvider.getLoggedInUser()).thenReturn(user);

引发异常，因为 userProvider.getLoggedInUser() 已被 thenTrow()

存根如何告诉 when 方法不关心异常？

提前致谢 - István

Answer 1

在新的 Mockito 版本中，您可以使用存根连续调用来在第一个调用上抛出异常并在第二个调用上返回一个值。

when(mock.someMethod("some arg"))
    .thenThrow(new RuntimeException())
    .thenReturn("foo");

https://javadoc.io/doc /org.mockito/mockito-core/latest/org/mockito/Mockito.html#10

Answer 2

我对你的问题的第一反应是，听起来你在一次测试中试图做太多事情。

为了便于测试和简单化，每个测试应该只测试一件事。这与单一职责原则相同。我经常发现程序员试图在一次测试中测试多个事物，并因此遇到各种各样的问题。因此，每个单元测试方法都应遵循以下流程：

1. 为测试设置单个场景。
2. 调用正在测试的类来触发正在测试的代码。
3. 验证行为。

因此，就您的情况而言，我预计至少会看到两次测试。一种是 getLoggedInUser() 返回用户，另一种是 getLoggedInUser() 抛出异常。这样，您在尝试模拟模拟中的不同行为时就不会遇到问题。

我想到的第二个想法是不要存根。考虑使用expect代替，因为你可以设置一系列的期望。即第一个调用返回一个用户，第二个调用抛出一个异常，第三个调用返回一个不同的用户，等等。

Answer 3

有什么方法可以告诉when方法不关心异常吗？

要实际回答这个问题：

import static org.hamcrest.Matchers.equalTo;
import static org.hamcrest.Matchers.is;
import static org.junit.Assert.assertThat;
import static org.mockito.ArgumentMatchers.any;
import static org.powermock.api.mockito.PowerMockito.mock;
import static org.powermock.api.mockito.PowerMockito.when;

import org.junit.Test;
import org.mockito.Mockito;

import java.util.ArrayList;

public class MyTest {

    @Test
    public void testA() {

        // setup
        ArrayList<Object> list = mock(ObjectArrayList.class);
        when(list.indexOf(any())).thenReturn(6);
        when(list.indexOf(any())).thenReturn(12);

        // execute
        int index = list.indexOf(new Object());

        // verify
        assertThat(index, is(equalTo(12)));
    }

    @Test
    public void testB() {

        // setup
        ArrayList<Object> list = mock(ObjectArrayList.class);
        when(list.add(any())).thenThrow(new AssertionError("can't get rid of me!"));
        when(list.add(any())).thenReturn(true);

        // execute
        list.add(new Object());
    }

    @Test
    public void testC() {

        // setup
        ArrayList<Object> list = mock(ObjectArrayList.class);
        when(list.add(any())).thenThrow(new AssertionError("can't get rid of me!"));
        Mockito.reset(list);
        when(list.add(any())).thenReturn(true);

        // execute
        list.add(new Object());
    }

    /**
     * Exists to work around the fact that mocking an ArrayList<Object>
     * requires a cast, which causes "unchecked" warnings, that can only be suppressed...
     */
    class ObjectArrayList extends ArrayList<Object> {

    }
}

TestB 由于无法摆脱断言而失败。 TestC 展示了如何使用 reset 方法重置模拟并删除其上的 thenThrow 命令。

请注意，在我拥有的一些更复杂的示例中，重置似乎并不总是有效。我怀疑这可能是因为他们使用的是 PowerMockito.mock 而不是 Mockito.mock？

Answer 4

使用 Mockito.reset() 重置任何特定的模拟，例如 Mockito.reset(mock1, mock2)

检查更多详细信息：https://stackoverflow.com/a/68126634/12085680

例如：

    @Test
    void test() {   
    when(mock.someMethod(any())).thenThrow(SomeException.class);
    // Using a service that uses the mock inside for example
    assertThrows(SomeException.class, () -> service.someMethodThatUsesTheMock("test"));
    Mockito.reset(mock); // reset it so you can reuse that mock on another test
    }