测试根据输入调用内部方法的公共方法

Question

所以我们这里有这个方法，系统的其余部分可以根据输入调用底层方法。

public SomeReturnObj doSomethingWithInputs(List<Input> inputs) {
  for(Input input : inputs) {
    if(input.getName().equals("A") {
        handleAInput(input);
    }
    else if(input.getName().equals("B") {
        handleBInput(input);
    }
    else { ... }
  }

  // ...
}

为了获得良好的代码覆盖率，我想测试一下，如果我将两个 Inputs 名称为 A 和三个名称为 B 的列表放在一起，则相应的内部方法会被调用两次或三次，分别。

所以我尝试了以下方法：

@Test
public void separatingInputsByName() {
    Input entry1 = mock(Input .class);
    Input entry2 = mock(Input .class);
    Input entry3 = mock(Input .class);

    doReturn("A").when(entry1).getName();
    doReturn("A").when(entry2).getName();
    doReturn("B").when(entry3).getName();

    ClassUnderTest sut = mock(ClassUnderTest .class);

    sut.doSomethingWithInputs(Arrays.asList(entry1, entry2, entry3));

    verify(sut).handleAInput(entry1);
    verify(sut).handleAInput(entry2);
    verify(sut).handleBInput(entry3);
}

很遗憾，这并没有导致正确调用内部方法，可能是因为被测类是模拟的，所以方法实现不同。

如何正确测试/验证这样的方法？

Answer 1

你应该使用spy()，而不是mock()。

当您使用mock() 时，所有方法都被“覆盖”，以便采取默认操作而不是调用真实方法； spy() 只会注册方法调用。

因此：

ClassUnderTest sut = spy(new ClassUnderTest(...));

sut.doSomethingWithInputs(Arrays.asList(entry1, entry2, entry3));

verify(sut).handleAInput(entry1);
verify(sut).handleAInput(entry2);
verify(sut).handleBInput(entry3);
verifyNoMoreInteractions(sut); // if necessary

此外，您还可以：

when(entry1.getName()).thenReturn("A");

就我个人而言，我觉得它更容易阅读，但这当然是个人喜好问题。

另外，您可以在您的情况下使用InOrder：

final InOrder inOrder = inOrder(sut);

inOrder.verify(sut).handleAInput(entry1);
// etc