2

可能重复：
在使用 Mockito 进行方法调用时拦截对象

我有一个可以从一种格式映射到另一种格式的类。由于这是遗留代码，我不敢重写它，它基本上是一组插件，所以如果我更改一个，我可能不得不更改所有其他。它的开发没有考虑到测试。所以这是我的问题。

interface Mapper {
    void handle(ClassA classA);
    void handle(ClassB classB);
}

public interface Publisher {
    public void publish(MappedClass mappedClass);
}

class MyMapper implements Mapper {

    private Publisher publisher;
    public void setPublisher(final Publisher publisher) {
        this.publisher = publisher;
    }

    public handle(ClassA classA) {
        final MappedClass mappedClass = // Map from ClassA to MappedClass
        publisher.publish(mappedClass);
    }

    public handle(ClassB classB) {
        final MappedClass mappedClass = // Map from ClassB to MappedClass
        publisher.publish(mappedClass);
    }
}

好的。因此，根据哪个类被“处理”MappedClass将发布不同的值，这是我要验证的值（测试）。问题是我将得到一个测试，我首先必须编写代码来测试该publish方法是否被调用，

private boolean wasCalled;

@Test
public void testClassAMapped() {
    wasCalled = false;
    final MyMapper myMapper = new MyMapper();
    myMapper.setPublisher(new Publisher() {
        public void publish(final MappedClass mappedClass) {
            wasCalled = true;
           // Code for verifying the fields in mappedClass
        });
    }
    final ClassA classA = // Create classA
    myMapper.handle(classA);
    assertTrue(wasCalled);
}

所以首先我们创建我们的模拟Publisher，它首先将状态设置为wasCalled，true所以我们知道这个方法曾经被调用过（这个例子被简化了，所以代码中实际上有一个调度程序......遗留代码所以我不想改变它)，其次我想验证MappedClass具有正确的字段值。我想知道的是，是否有人知道更好的测试方法？对于我的许多测试， wasCalled, 和wasCalled检查或多或少成为样板代码，但由于我不想添加那么多混乱（自己的黑客、测试基类等），我想知道是否有办法做这是在 Mockito 还是 EasyMock？

4

2 回答 2

6

使用Mockito ArgumentCaptor

@Test
public void test(){
  Publisher publisher = Mockito.mock(Publisher.class);
  myMapper.setPublisher(publisher);
  ArgumentCaptor<MappedClass> captor = ArgumentCaptor.forClass(MappedClass.class);

  ....
  myMapper.handle(...);
  ...

  verify(publisher).publish(captor.capture());
  MappedClass passedValue = captor.getValue();
  // assert stuff here
}
于 2012-09-10T13:31:35.983 回答
5

我不确定我是否完全理解这个问题，但看起来你正在寻找Mockito.verify(publisher).publish(Matchers.isA(MappedClass.class));

为此，您必须模拟Publisher通过
Publisher publisher = Mockito.mock(Publisher.class)
，然后将其交给MyMapper.

如果您需要断言 的状态MappedClass，请使用ArgumentCaptor. 有关示例，请参见此答案。

Mockito -API-doc有许多其他示例。

于 2012-09-10T13:17:52.667 回答