使用 Mockito 在方法调用时拦截对象

mocking java mockito

2022-09-01 11:33:39

我有一个模拟类，使用void方法，例如

public class Mock {
    public void method(String string) {
        // doSomething
    }
}

我不在乎这种方法做什么，但我想发送字符串。

此字符串实际上是 JSON 格式的对象，我正在测试的方法是根据最初发送的字符串修改此对象（假设是相当随机的）。

method(String json) {
    Object obj = unparse(json);
    obj.setRandomValue(random);
    String parsed = parse(obj);
    Mock.method(parsed);
}

我只想看看“randomValue”，以前是null，实际上是在方法调用后用随机设置的。

最好的方法是拦截json，解析它并检查对象。

答案 1

你正在寻找一个参数捕获器：

  Mock mock =  Mockito.mock(Mock.class);
  ArgumentCaptor<String> captor = ArgumentCaptor.forClass(String.class);
  mock.method("input");
  Mockito.verify(mock).method(captor.capture());
  String actualValue = captor.getValue();

答案 2

还有一种方法可以做到这一点，下面是一个示例：

Mock mock =  Mockito.mock(Mock.class);
when(mock.method(any(String.class))).thenAnswer(new Answer<Void>() {

        @Override
        public Void answer(InvocationOnMock invocation) throws Throwable {
            String actualValue = invocation.getArgumentAt(0, String.class);
            Assert.assertEquals("input", actualValue);
            return null;
        }
    });
// Test your method 
mock.method("input");
Mockito.verify(mock, Mockito.times(1)).method("input");