java - 模拟一个类将空值分配给@NonNull 字段(Android)

Question

我正在使用 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 的类，所以 values.add(value) 在 < strong>addValue() 方法抛出 NPE。我可以在 addValue() 中添加一个 null 检查来解决这个问题，但这似乎很荒谬，因为值已经被声明为非 null。

有没有办法让 Mockito 尊重@NonNull 注释并正确初始化字段？

Answer 1

我认为您没有以正确的方式使用 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"));
    }

}