内容纲要

准备

在我们进一步讨论之前,让我们探索几种不同的方法来启用Mockito测试中注释的使用。

方式一 MockitoJUnitRunner

我们拥有的第一个选择是使用MockitoJUnitRunner注释JUnit测试,如以下示例所示:

@RunWith(MockitoJUnitRunner.class)
public class MockitoAnnotationTest {
    ...
}

方式二 MockitoAnnotations.initMocks()

另外,我们也可以通过调用MockitoAnnotations.initMocks()来以编程方式启用Mockito注释:

@Before
public void init() {
    MockitoAnnotations.initMocks(this);
}

方式三 MockitoJUnit.rule()

或者,我们可以使用MockitoJUnit.rule(),如下所示:

public class MockitoInitWithMockitoJUnitRuleUnitTest {

    @Rule
    public MockitoRule initRule = MockitoJUnit.rule();

    ...
}

在这种情况下,我们必须记住将Rule成员设置为public。


Mockito注解介绍

@Mock Annotation

Mockito中使用最广泛的注释是@Mock。我们可以使用@Mock来创建和注入模拟实例,而不必手动调用Mockito.mock。

在下面的示例中–我们将使用手动方式创建模拟ArrayList,而不使用@Mock注释:

@Test
public void whenNotUseMockAnnotation_thenCorrect() {
    List mockList = Mockito.mock(ArrayList.class);

    mockList.add("one");
    Mockito.verify(mockList).add("one");
    assertEquals(0, mockList.size());

    Mockito.when(mockList.size()).thenReturn(100);
    assertEquals(100, mockList.size());
}

现在我们将做同样的事情,但是我们将使用@Mock注释注入模拟:

@Mock
List<String> mockedList;

@Test
public void whenUseMockAnnotation_thenMockIsInjected() {
    mockedList.add("one");
    Mockito.verify(mockedList).add("one");
    assertEquals(0, mockedList.size());

    Mockito.when(mockedList.size()).thenReturn(100);
    assertEquals(100, mockedList.size());
}

请注意,在两个示例中,我们如何与模拟对象进行交互并验证其中的某些交互,只是为了确保模拟行为正确。

@Spy Annotation

现在–让我们看看如何使用@Spy注释监视现有实例。

在下面的示例中,我们使用旧方法创建一个列表的spy,而不使用@spy注释:

@Test
public void whenNotUseSpyAnnotation_thenCorrect() {
    List<String> spyList = Mockito.spy(new ArrayList<String>());

    spyList.add("one");
    spyList.add("two");

    Mockito.verify(spyList).add("one");
    Mockito.verify(spyList).add("two");

    assertEquals(2, spyList.size());

    Mockito.doReturn(100).when(spyList).size();
    assertEquals(100, spyList.size());
}

现在让我们做同样的事情–在列表上监视–但是使用@spy注释:

@Spy
List<String> spiedList = new ArrayList<String>();

@Test
public void whenUseSpyAnnotation_thenSpyIsInjectedCorrectly() {
    spiedList.add("one");
    spiedList.add("two");

    Mockito.verify(spiedList).add("one");
    Mockito.verify(spiedList).add("two");

    assertEquals(2, spiedList.size());

    Mockito.doReturn(100).when(spiedList).size();
    assertEquals(100, spiedList.size());
}

请注意,像以前一样,我们在这里与间谍对象进行交互以确保其行为正确。在此示例中,我们:

  • 使用实际方法spiedList.add()将元素添加到spiedList。
  • 使用Mockito.doReturn()将spiedList.size()方法存根以返回100而不是2。

@Captor Annotation

接下来–让我们看看如何使用@Captor注释创建ArgumentCaptor实例。

在下面的示例中–我们使用旧方法创建ArgumentCaptor,而不使用@Captor注释:

@Test
public void whenNotUseCaptorAnnotation_thenCorrect() {
    List mockList = Mockito.mock(List.class);
    ArgumentCaptor<String> arg = ArgumentCaptor.forClass(String.class);

    mockList.add("one");
    Mockito.verify(mockList).add(arg.capture());

    assertEquals("one", arg.getValue());
}

现在,出于相同的目的,使用@Captor –创建一个ArgumentCaptor实例:

@Mock
List mockedList;

@Captor
ArgumentCaptor argCaptor;

@Test
public void whenUseCaptorAnnotation_thenTheSam() {
    mockedList.add("one");
    Mockito.verify(mockedList).add(argCaptor.capture());

    assertEquals("one", argCaptor.getValue());
}

请注意,当我们取出配置逻辑时,测试如何变得更简单、更可读。

@InjectMocks Annotation

现在-让我们讨论如何使用@InjectMocks注释-将模拟字段自动注入到测试对象中。

在以下示例中,我们使用@InjectMocks将模拟wordMap注入MyDictionary dic:

@Mock
Map<String, String> wordMap;

@InjectMocks
MyDictionary dic = new MyDictionary();

@Test
public void whenUseInjectMocksAnnotation_thenCorrect() {
    Mockito.when(wordMap.get("aWord")).thenReturn("aMeaning");

    assertEquals("aMeaning", dic.getMeaning("aWord"));
}

这是MyDictionary类:

public class MyDictionary {
    Map<String, String> wordMap;

    public MyDictionary() {
        wordMap = new HashMap<String, String>();
    }
    public void add(final String word, final String meaning) {
        wordMap.put(word, meaning);
    }
    public String getMeaning(final String word) {
        return wordMap.get(word);
    }
}

Notes

最后–这是有关Mockito注释的一些说明:

  • Mockito的注释将重复的模拟创建代码减至最少
  • 它们使测试更具可读性
  • @InjectMocks支持同时注入@Spy和@Mock实例

在本快速教程中,我们展示了Mockito库中注释的基础。

注:本文翻译自原文:https://www.baeldung.com/mockito-annotations ,并且在此基础上有改动。

发表评论

电子邮件地址不会被公开。