java - 模拟 ColumnDefinitions.Definition 确实返回模拟，但在测试代码中表现得像 null

Question

我有以下代码准备模拟以使用 Cassandra 测试我的服务(我需要模拟 com.datastax.driver.core.ColumnDefinitions.Definition):

@RunWith(PowerMockRunner.class)
public class TestMyClass{
private MyClass target;
  @Before
  public void setUp() throws Exception {
    ColumnDefinitions mockColumnDefinitions=Mockito.mock(ColumnDefinitions.class);
    Mockito.when(mockRow.getColumnDefinitions()).thenReturn(mockColumnDefinitions);
    target= new MyClass();
    Definition mockDef = Mockito.mock(Definition.class);
    List<Definition> defList = new ArrayList<Definition>();
    defList.add(mockDef);
    Iterator mockIterator = Mockito.mock(Iterator.class);
    Mockito.when(mockColumnDefinitions.iterator()).thenReturn(mockIterator);
    Mockito.when(mockIterator.hasNext()).thenReturn(true, false);
    Mockito.when(mockIterator.next()).thenReturn(mockDef);
    Mockito.when(mockDef.getName()).thenReturn(NAME);
  }
 @Test
 public void testMyMethod() throws Exception {
    target.MyMethod();
 }
}

这个地方的测试执行很顺利，我在不同的地方都有这种类型的代码，所以它应该可以工作。在我正在测试的服务中，我有以下代码:

ColumnDefinitions colDef = row.getColumnDefinitions();
Iterator<Definition> defIterator = colDef.iterator();
while (defIterator.hasNext()) {
    Definition def = defIterator.next();
    String columnName = def.getName();
}

当我调试这段代码时，我发现 colDef 和 defIterator 都被成功模拟了。我在调试变量区域看到类似的东西:

Mock for Iterator, hashCode: 430126690

但是在 defIterator.next() 调用之后，我看到虽然 def 是一个对象而不是 null，但它不像 Iterator< 那样显示哈希码，相反我看到了这个:

com.sun.jdi.InvocationException occurred invoking method.

在调用这个字符串之后:

String columnName = def.getName();

如果 def 为 null，我会立即得到 NullPointerException。我究竟做错了什么？谢谢。

编辑 1 ______________________________________________________________________

我也尝试使用 PowerMockito 来代替相同的方法，结果是一样的。

编辑 2 __________________________________________________________________________

我添加了整个测试方法代码。

Answer 1

创建这个问题已经有一段时间了。几天前我遇到了同样的问题，我已经通过以下方式解决了它(我希望我提出的解决方案对以后的人有所帮助):

首先，我想澄清一下 ColumnDefinition.Definition 类是一个 public static 嵌套类，它有四个 private final 字段和只有一个构造函数:Definition (String keyspace, String table, String name and DataType type)(更多细节请引用ColumnDefinitions.Definition javadoc和ColumnDefinitions source code)。因此 Mockito 无法模拟这个嵌套类也不Powermock因为它的 final 字段。

解决方案:

我必须使用反射创建一个真实的对象，而不是 ColumnDefinition.Definition 类的模拟对象，因此您可以按如下方式初始化 mockDef 对象:

Constructor<Definition> constructor = (Constructor<Definition>) Definition.class.getDeclaredConstructors()[0]; // as Definition only has one constructor, 0 will be passed as index
constructor.setAccessible(true);
Definition mockDef = constructor.newInstance("keyspace", "table", "name", null);

替换您代码段中的这行代码:

Definition mockDef = Mockito.mock(Definition.class);

那么执行这行代码就不会再抛出NullPointerException了:

String columnName = def.getName();