Question

我有一个Mockito测试看起来有点像这样（当然简化）：

@RunWith(MockitoJUnitRunner.class)
public class BlahTest {
    private static final int VERSION = 41;
    private static final int PAGE_SIZE = 4096;

    @Mock private FileChannel channel;

    @Test
    public void shouldWriteStandardHeader() throws Exception {
        final Blah blah = new Blah(channel, VERSION, PAGE_SIZE);
        blah.create();

        verify(channel).write(littleEndianByteBufferContaining(Blah.MAGIC_NUMBER,
                                                               VERSION,
                                                               PAGE_SIZE));
    }

    private ByteBuffer littleEndianByteBufferContaining(final int... ints) {
        return argThat(byteBufferMatcher(ints));
    }

    private Matcher<ByteBuffer> byteBufferMatcher(final int... ints) {
        return new TypeSafeMatcher<ByteBuffer>() {
            @Override
            public void describeTo(final Description description) {
                description.appendText("a little-endian byte buffer containing integers ").
                            appendValueList("", ",", "", ints);
            }

            @Override
            protected boolean matchesSafely(final ByteBuffer buffer) {
                if (buffer.order() != ByteOrder.LITTLE_ENDIAN) {
                    return false;
                }

                for (final int i : ints) {
                    if (buffer.getInt() != i) {
                        return false;
                    }
                }

                return true;
            }
        };
    }
}

本质上，此测试试图断言在调用Blah.create()时，它会将包含特定数据的ByteBuffer写入FileChannel。

当我运行此测试时，匹配器会被调用两次。这导致BufferUnderflowException。

现在，我可以通过让匹配器在matchesSafely调用开始时存储缓冲区位置并将位置移回到最后位置（在finally块中）来解决这个问题，但似乎对我来说，我的匹配器不应该被叫两次。

任何人都可以对此有所了解吗？

编辑＃1：

可能值得注意的是缓冲区在传递到通道之前被翻转，因此位置为0，限制设置为写入的数据量。

我已经调试了测试，匹配器肯定会被调用两次。

我可以通过在matchesSafely()开头标记缓冲区并在结束时重置它来使测试通过，因此第二次通过匹配器读取相同的数据。这也证实了匹配器被调用两次，否则它仍会失败。

编辑＃2：

所以看起来这是Mockito框架的预期行为。回想起来，我的匹配器有点差，因为它修改了全局状态。我已经修改了匹配器以记录起始位置并在matchesSafely()方法的末尾寻找它。无论如何，这可能是个好主意，因为它可以节省修改全局状态。出于同样的原因，我不使用mark()和reset()。

Answer 1

我不认为您的匹配器被调用两次，您只需要rewind缓冲区，然后再阅读它：

protected boolean matchesSafely(final ByteBuffer buffer) {
    if (buffer.order() != ByteOrder.LITTLE_ENDIAN) {
        return false;
    }
    buffer.rewind();
    ...
}

<强>更新

所以，看起来确实会被调用两次。它最终都发生在verify方法中。如果你看看Mockito来源，有Times.verify方法实际上验证了两件事：

如果方法缺少任何调用
该方法的调用次数正是所需的

Mockito拥有channel模拟对象上所有方法的实际调用列表。要验证哪些调用是正确的，它将匹配与匹配器的每次调用。它实际上做了两次。请查看来源以获得完整的想法。

我不确定这是不是一个bug，你应该问Mockito开发者。我建议每次修复问题时rewind方法中的matchesSafely缓冲区都不会受到影响 - 这不应该损害正确性。

Answer 2

最好使用ArgumentCaptor来验证参数，以避免两次调用自定义匹配器。

ArgumentCaptor<ByteBuffer> captor = ArgumentCaptor.forClass(ByteBuffer.class);
verify(channel).write(captor.capture());
assertThat(captor.getValue().order(), equalTo(ByteOrder.LITTLE_ENDIAN));

为什么Mockito会两次打电话给我的匹配器？

2 个答案: