2

我正在使用 Android SDK 和 junit4 + Mockito 进行单元测试。假设我的应用中有这样的课程：

public class Container{

  @NonNull private Set<String> values = new HashSet<>();

  public void addValue(String value) {
    values.add(value);
  }

  @NonNull
  public Set<String> getValues() {
    return values;
  }
}

而且我还有一个使用 Mockito 的单元测试，如下所示：

public class ContainerTest {

  private Container container;

  @Before
  public void before() {
    container = mock(Container.class);
  }

  @Test
  public void shouldAddValue() {
    container.add("test_value");
    assertTrue(container.getValues.contains("test_value"));
  }

}

这个测试实际上在 "container.add("test_value");" 行失败了 因为mock(Container.class)创建了一个类，其中values字段实际上设置为 null，所以addValue()方法中的values.add( value)会抛出 NPE。我可以在addValue()中添加一个空检查来解决这个问题，但这似乎很荒谬，因为值已经被声明为非空。

有什么方法可以让 Mockito 尊重 @NonNull 注释并正确初始化该字段？

4

1 回答 1

2

我认为您没有以正确的方式使用 Mockito，因为您必须定义 Mockito 行为。

例如，你应该有这样的东西：

@Test
public void shouldAddValue() {
    Set<String> mySet = new HashSet<String>();
    mySet.put("test_value");

    // Mock container getValues() method to return mySet
    when(container.getValues()).thenReturn(mySet); // do import static for Mockito.when

    assertTrue(container.getValues().contains("test_value"));
}

当您模拟响应时，Mockito 工作得很好，但是您想要的是让 Mockito 为您初始化类，这显然不是 Mockito 的目标。

因此，如果你想测试你的 Container 对象，那么你不必模拟 Container 本身，你可以有这样的东西：

public class ContainerTest {

    private Container container;

    @Before
    public void before() {
        container = new Container(); // Initialize container
    }

    @Test
    public void shouldAddValue() {
        container.addValue("test_value");
        assertTrue(container.getValues().contains("test_value"));
    }

}
于 2015-11-09T20:00:37.377 回答